From e41f287e9e44bd9f8cc3386e83912607223274af Mon Sep 17 00:00:00 2001 From: drebs Date: Tue, 5 Mar 2013 19:31:15 -0300 Subject: Refactor Soledad shared db to another file. --- __init__.py | 107 ++------------------------------------------------------ server.py | 1 + shared_db.py | 111 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 115 insertions(+), 104 deletions(-) create mode 100644 shared_db.py diff --git a/__init__.py b/__init__.py index 78cf27ef..97130cde 100644 --- a/__init__.py +++ b/__init__.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- """ Soledad - Synchronization Of Locally Encrypted Data Among Devices. @@ -18,17 +19,13 @@ try: import simplejson as json except ImportError: import json # noqa -from u1db import errors -from u1db.remote import ( - http_client, - http_database, -) from leap.soledad.backends import sqlcipher from leap.soledad.util import GPGWrapper from leap.soledad.backends.leap_backend import ( LeapDocument, DocumentNotEncrypted, ) +from leap.soledad.shared_db import SoledadSharedDatabase class KeyDoesNotExist(Exception): @@ -438,102 +435,4 @@ class Soledad(object): return self._db.sync(url, creds=None, autocreate=True) -#----------------------------------------------------------------------------- -# Soledad shared database -#----------------------------------------------------------------------------- - -class NoTokenForAuth(Exception): - """ - No token was found for token-based authentication. - """ - - -class Unauthorized(Exception): - """ - User does not have authorization to perform task. - """ - - -class SoledadSharedDatabase(http_database.HTTPDatabase): - """ - This is a shared HTTP database that holds users' encrypted keys. - """ - # TODO: prevent client from messing with the shared DB. - # TODO: define and document API. - - @staticmethod - def open_database(url, create, token=None, soledad=None): - db = SoledadSharedDatabase(url, token=token, soledad=soledad) - db.open(create) - return db - - def delete_database(url): - raise Unauthorized("Can't delete shared database.") - - def __init__(self, url, document_factory=None, creds=None, token=None, - soledad=None): - self._set_token(token) - self._soledad = soledad - super(SoledadSharedDatabase, self).__init__(url, document_factory, - creds) - - def _set_token(self, token): - self._token = token - - def _get_token(self): - return self._token - - token = property(_get_token, _set_token, - doc='Token for token-based authentication.') - - def _request(self, method, url_parts, params=None, body=None, - content_type=None, auth=True): - """ - Perform token-based http request. - """ - if auth: - if not self.token: - raise NoTokenForAuth() - if not params: - params = {} - params['auth_token'] = self.token - return super(SoledadSharedDatabase, self)._request( - method, url_parts, - params, - body, - content_type) - - def _request_json(self, method, url_parts, params=None, body=None, - content_type=None, auth=True): - """ - Perform token-based http request. - """ - res, headers = self._request(method, url_parts, - params=params, body=body, - content_type=content_type, auth=auth) - return json.loads(res), headers - - def get_doc_unauth(self, doc_id): - """ - Modified method to allow for unauth request. - """ - try: - res, headers = self._request( - 'GET', ['doc', doc_id], {"include_deleted": False}, - auth=False) - except errors.DocumentDoesNotExist: - return None - except errors.HTTPError, e: - if (e.status == http_database.DOCUMENT_DELETED_STATUS and - 'x-u1db-rev' in e.headers): - res = None - headers = e.headers - else: - raise - doc_rev = headers['x-u1db-rev'] - has_conflicts = json.loads(headers['x-u1db-has-conflicts']) - doc = self._factory(doc_id, doc_rev, res) - doc.has_conflicts = has_conflicts - return doc - -__all__ = ['util'] +__all__ = ['util', 'server', 'shared_db'] diff --git a/server.py b/server.py index dbaf6a13..a6e520df 100644 --- a/server.py +++ b/server.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- """ A U1DB server that stores data using couchdb. diff --git a/shared_db.py b/shared_db.py new file mode 100644 index 00000000..7a1eae5d --- /dev/null +++ b/shared_db.py @@ -0,0 +1,111 @@ +# -*- coding: utf-8 -*- +""" +Created on Tue Mar 5 18:46:38 2013 + +@author: drebs +""" + +try: + import simplejson as json +except ImportError: + import json # noqa + +from u1db import errors +from u1db.remote import http_database + + +#----------------------------------------------------------------------------- +# Soledad shared database +#----------------------------------------------------------------------------- + +class NoTokenForAuth(Exception): + """ + No token was found for token-based authentication. + """ + + +class Unauthorized(Exception): + """ + User does not have authorization to perform task. + """ + + +class SoledadSharedDatabase(http_database.HTTPDatabase): + """ + This is a shared HTTP database that holds users' encrypted keys. + """ + # TODO: prevent client from messing with the shared DB. + # TODO: define and document API. + + @staticmethod + def open_database(url, create, token=None, soledad=None): + """ + Open a Soledad shared database. + """ + db = SoledadSharedDatabase(url, token=token, soledad=soledad) + db.open(create) + return db + + @staticmethod + def delete_database(url): + """ + Dummy method that prevents from deleting shared database. + """ + raise Unauthorized("Can't delete shared database.") + + def __init__(self, url, document_factory=None, creds=None, token=None, + soledad=None): + self._token = token + self._soledad = soledad + super(SoledadSharedDatabase, self).__init__(url, document_factory, + creds) + + def _request(self, method, url_parts, params=None, body=None, + content_type=None, auth=True): + """ + Perform token-based http request. + """ + if auth: + if not self._token: + raise NoTokenForAuth() + if not params: + params = {} + params['auth_token'] = self._token + return super(SoledadSharedDatabase, self)._request( + method, url_parts, + params, + body, + content_type) + + def _request_json(self, method, url_parts, params=None, body=None, + content_type=None, auth=True): + """ + Perform token-based http request. + """ + res, headers = self._request(method, url_parts, + params=params, body=body, + content_type=content_type, auth=auth) + return json.loads(res), headers + + def get_doc_unauth(self, doc_id): + """ + Modified method to allow for unauth request. + """ + try: + res, headers = self._request( + 'GET', ['doc', doc_id], {"include_deleted": False}, + auth=False) + except errors.DocumentDoesNotExist: + return None + except errors.HTTPError, e: + if (e.status == http_database.DOCUMENT_DELETED_STATUS and + 'x-u1db-rev' in e.headers): + res = None + headers = e.headers + else: + raise + doc_rev = headers['x-u1db-rev'] + has_conflicts = json.loads(headers['x-u1db-has-conflicts']) + doc = self._factory(doc_id, doc_rev, res) + doc.has_conflicts = has_conflicts + return doc -- cgit v1.2.3