1 files changed, 150 insertions, 0 deletions
diff --git a/src/couch/utils.py b/src/couch/utils.py
new file mode 100644
index 0000000..1cd21d8
--- /dev/null
+++ b/src/couch/utils.py
@@ -0,0 +1,150 @@
+# -*- coding: utf-8 -*-
+# Based on py-couchdb (https://github.com/histrio/py-couchdb)
+
+
+import json
+import sys
+
+
+if sys.version_info[0] == 3:
+    from urllib.parse import unquote as _unquote
+    from urllib.parse import urlunsplit, urlsplit
+
+    string_type = str
+    bytes_type = bytes
+
+    from functools import reduce
+
+else:
+    from urllib import unquote as _unquote
+    from urlparse import urlunsplit, urlsplit
+
+    string_type = unicode  # noqa: F821
+    bytes_type = str
+
+URLSPLITTER = '/'
+
+
+json_encoder = json.JSONEncoder()
+
+
+def extract_credentials(url):
+    """
+    Extract authentication (user name and password) credentials from the
+    given URL.
+
+    >>> extract_credentials('http://localhost:5984/_config/')
+    ('http://localhost:5984/_config/', None)
+    >>> extract_credentials('http://joe:secret@localhost:5984/_config/')
+    ('http://localhost:5984/_config/', ('joe', 'secret'))
+    >>> extract_credentials('http://joe%40example.com:secret@'
+    ...                      'localhost:5984/_config/')
+    ('http://localhost:5984/_config/', ('joe@example.com', 'secret'))
+    """
+    parts = urlsplit(url)
+    netloc = parts[1]
+    if '@' in netloc:
+        creds, netloc = netloc.split('@')
+        credentials = tuple(_unquote(i) for i in creds.split(':'))
+        parts = list(parts)
+        parts[1] = netloc
+    else:
+        credentials = None
+    return urlunsplit(parts), credentials
+
+
+def _join(head, tail):
+    parts = [head.rstrip(URLSPLITTER), tail.lstrip(URLSPLITTER)]
+    return URLSPLITTER.join(parts)
+
+
+def urljoin(base, *path):
+    """
+    Assemble a uri based on a base, any number of path segments, and query
+    string parameters.
+
+    >>> urljoin('http://example.org', '_all_dbs')
+    'http://example.org/_all_dbs'
+
+    A trailing slash on the uri base is handled gracefully:
+
+    >>> urljoin('http://example.org/', '_all_dbs')
+    'http://example.org/_all_dbs'
+
+    And multiple positional arguments become path parts:
+
+    >>> urljoin('http://example.org/', 'foo', 'bar')
+    'http://example.org/foo/bar'
+
+    >>> urljoin('http://example.org/', 'foo/bar')
+    'http://example.org/foo/bar'
+
+    >>> urljoin('http://example.org/', 'foo', '/bar/')
+    'http://example.org/foo/bar/'
+
+    >>> urljoin('http://example.com', 'org.couchdb.user:username')
+    'http://example.com/org.couchdb.user:username'
+    """
+    return reduce(_join, path, base)
+
+
+def as_json(response):
+    if "application/json" in response.headers['content-type']:
+        response_src = response.content.decode('utf-8')
+        if response.content != b'':
+            return json.loads(response_src)
+        else:
+            return response_src
+    return None
+
+
+def _path_from_name(name, type):
+    """
+    Expand a 'design/foo' style name to its full path as a list of
+    segments.
+
+    >>> _path_from_name("_design/test", '_view')
+    ['_design', 'test']
+    >>> _path_from_name("design/test", '_view')
+    ['_design', 'design', '_view', 'test']
+    """
+    if name.startswith('_'):
+        return name.split('/')
+    design, name = name.split('/', 1)
+    return ['_design', design, type, name]
+
+
+def encode_view_options(options):
+    """
+    Encode any items in the options dict that are sent as a JSON string to a
+    view/list function.
+
+    >>> opts = {'key': 'foo', "notkey":"bar"}
+    >>> res = encode_view_options(opts)
+    >>> res["key"], res["notkey"]
+    ('"foo"', 'bar')
+
+    >>> opts = {'startkey': 'foo', "endkey":"bar"}
+    >>> res = encode_view_options(opts)
+    >>> res['startkey'], res['endkey']
+    ('"foo"', '"bar"')
+    """
+    retval = {}
+
+    for name, value in options.items():
+        if name in ('key', 'startkey', 'endkey'):
+            value = json_encoder.encode(value)
+        retval[name] = value
+    return retval
+
+
+def force_bytes(data, encoding="utf-8"):
+    if isinstance(data, string_type):
+        data = data.encode(encoding)
+    return data
+
+
+def force_text(data, encoding="utf-8"):
+    if isinstance(data, bytes_type):
+        data = data.decode(encoding)
+    return data