Clean requests interface

Refactorize on build request to support specific requests cleaner

*Also support to verbose on requests library

6 files changed, 92 insertions, 192 deletions

diff --git a/pygithub3/requests/__init__.py b/pygithub3/requests/__init__.py
index 725ea12..9fdde27 100644
--- a/pygithub3/requests/__init__.py
+++ b/pygithub3/requests/__init__.py
@@ -8,60 +8,77 @@ try:
 except ImportError:
     import json
 
-ABS_IMPORT_PREFIX = 'pygithub3.requests'
+from pygithub3.exceptions import DoesNotExists, UriInvalid, ValidationError
+from pygithub3.resources.base import Raw
 
+ABS_IMPORT_PREFIX = 'pygithub3.requests'
 
-class RequestNotFound(Exception):
-    pass
 
+class Body(object):
 
-class RequestUriInvalid(Exception):
-    pass
+    def __init__(self, content, schema):
+        self.content = content
+        self.schema = schema
 
+    def dumps(self):
+        if not self.content:
+            return None
+        return json.dumps(self.parse())
 
-class RequestValidationError(Exception):
-    pass
+    def parse(self):
+        if not self.schema:
+            return self.content
+        if not hasattr(self.content, 'items'):
+            raise ValidationError("'%s' needs a content dictionary"
+                                   % self.__class__.__name__)
+        return {key: self.content[key] for key in self.schema
+                if key in self.content}
 
 
 class Request(object):
     """ """
 
+    uri = ''
+    resource = Raw
+    body_schema = ()
+
     def __init__(self, args):
         """ """
+        self.body = args.pop('body', None)
         self.args = args
-        self.validate()
-        self.uri = self.set_uri()
+        self.clean()
 
-    def validate(self):
-        raise NotImplementedError
+    def clean(self):
+        self.uri = self.clean_uri() or self.uri
+        self.body = Body(self.clean_body(), self.body_schema)
 
-    def set_uri(self):
-        raise NotImplementedError
+    def clean_body(self):
+        return self.body
 
-    def get_data(self):
-        raise NotImplementedError
+    def clean_uri(self):
+        return None
 
-    def get_uri(self):
-        return str(self.uri).strip('/')
-
-    def get_resource(self):
-        return getattr(self, 'resource', '')
+    @property
+    def resource(self):
+        return self.resource
 
     def __getattr__(self, name):
         return self.args.get(name)
 
     def __str__(self):
-        return self.get_uri()
+        return self.populate_uri()
+
+    def populate_uri(self):
+        try:
+            populated_uri = self.uri.format(**self.args)
+        except KeyError:
+            raise ValidationError(
+                "'%s' request wasn't be able to populate the uri '%s' with "
+                "'%s' args" % (self.__class__.__name__, self.uri, self.args))
+        return str(populated_uri).strip('/')
 
-    def _parse_simple_dict(self, to_parse):
-        if not hasattr(to_parse, 'items'):
-            raise RequestValidationError("'%s' needs a data dictionary"
-                                         % self.__class__.__name__)
-        update_params = {
-                valid_key: to_parse[valid_key]
-                for valid_key in self.valid
-                if valid_key in to_parse}
-        return update_params
+    def get_body(self):
+        return self.body.dumps()
 
 
 class Factory(object):
@@ -69,29 +86,19 @@ class Factory(object):
 
     import_pattern = re.compile(r'^(\w+\.)+\w+$')
 
-    def __init__(self):
-        """ """
-        self.args = {}
-
-    def config_with(self, **kwargs):
-        self.args = kwargs
-
-    def clear_config(self):
-        self.args = {}
-
     def __validate(func):
         """ """
 
-        def wrapper(self, request_uri):
+        def wrapper(self, request_uri, **kwargs):
             if not Factory.import_pattern.match(request_uri):
-                raise RequestUriInvalid("'%s' isn't valid form" % request_uri)
-            return func(self, request_uri.lower())
+                raise UriInvalid("'%s' isn't valid form" % request_uri)
+            return func(self, request_uri.lower(), **kwargs)
         return wrapper
 
     def __dispatch(func):
         """ """
 
-        def wrapper(self, request_uri):
+        def wrapper(self, request_uri, **kwargs):
             module_chunk, s, request_chunk = request_uri.rpartition('.')
             try:
                 #  TODO: CamelCase and under_score support, now only Class Name
@@ -99,19 +106,18 @@ class Factory(object):
                                         % (ABS_IMPORT_PREFIX, module_chunk))
                 request = getattr(module, request_chunk.capitalize())
             except ImportError:
-                raise RequestNotFound("'%s' module does not exists"
+                raise DoesNotExists("'%s' module does not exists"
                                        % module_chunk)
             except AttributeError:
-                raise RequestNotFound(
+                raise DoesNotExists(
                     "'%s' request doesn't exists into '%s' module"
                     % (request_chunk.capitalize(), module_chunk))
-            return func(self, request)
+            return func(self, request, **kwargs)
         return wrapper
 
     @__validate
     @__dispatch
-    def __call__(self, request=''):
-        request = request(self.args)
-        self.clear_config()
+    def __call__(self, request='', **kwargs):
+        request = request(kwargs)
         assert isinstance(request, Request)
         return request
diff --git a/pygithub3/requests/users/__init__.py b/pygithub3/requests/users/__init__.py
index 93ab7b6..11e6a39 100644
--- a/pygithub3/requests/users/__init__.py
+++ b/pygithub3/requests/users/__init__.py
@@ -1,4 +1,4 @@
 # -*- encoding: utf-8 -*-
 
-from pygithub3.requests import Request, json, RequestValidationError
+from pygithub3.requests import Request, ValidationError
 from user import *
diff --git a/pygithub3/requests/users/emails.py b/pygithub3/requests/users/emails.py
index f7fb284..7622fcc 100644
--- a/pygithub3/requests/users/emails.py
+++ b/pygithub3/requests/users/emails.py
@@ -1,44 +1,28 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from . import Request, json
-from pygithub3.resources.base import Raw
+import re
 
+from . import Request
 
-class List(Request):
-
-    resource = Raw
 
-    def validate(self):
-        pass
+class List(Request):
 
-    def set_uri(self):
-        return 'user/emails'
+    uri = 'users/emails'
 
 
 class Add(Request):
 
-    resource = Raw
+    uri = 'user/emails'
 
-    def validate(self):
-        pass
+    def clean_body(self):
 
-    def get_data(self):
-        return json.dumps(self.emails)
+        def is_email(email):
+            return re.match(r'.*', email)  # TODO: email regex ;)
 
-    def set_uri(self):
-        return 'user/emails'
+        return filter(is_email, self.body)
 
 
 class Delete(Request):
 
-    resource = Raw
-
-    def validate(self):
-        pass
-
-    def get_data(self):
-        return json.dumps(self.emails)
-
-    def set_uri(self):
-        return 'user/emails'
+    uri = 'user/emails'
diff --git a/pygithub3/requests/users/followers.py b/pygithub3/requests/users/followers.py
index f2bfc1a..5f59032 100644
--- a/pygithub3/requests/users/followers.py
+++ b/pygithub3/requests/users/followers.py
@@ -1,76 +1,40 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from . import Request, RequestValidationError
+from . import Request
 from pygithub3.resources.users import User
-from pygithub3.resources.base import Raw
 
 
 class List(Request):
 
     resource = User
+    uri = 'users/{user}/followers'
 
-    def validate(self):
-        pass
-
-    def set_uri(self):
-        if self.user:
-            return 'users/%s/followers' % self.user
-        else:
+    def clean_uri(self):
+        if not self.user:
             return 'user/followers'
 
 
 class Listfollowing(Request):
 
     resource = User
+    uri = 'users/{user}/following'
 
-    def validate(self):
-        pass
-
-    def set_uri(self):
-        if self.user:
-            return 'users/%s/following' % self.user
-        else:
+    def clean_uri(self):
+        if not self.user:
             return 'user/following'
 
 
 class Isfollowing(Request):
 
-    resource = Raw
-
-    def validate(self):
-        if not self.user:
-            raise RequestValidationError(
-                "'%s' request needs a user" % self.__class__.__name__)
-
-    def set_uri(self):
-        return 'user/following/%s' % self.user
+    uri = 'user/following/{user}'
 
 
 class Follow(Request):
 
-    resource = Raw
-
-    def validate(self):
-        if not self.user:
-            raise RequestValidationError(
-                "'%s' request needs a user" % self.__class__.__name__)
-
-    def set_uri(self):
-        return 'user/following/%s' % self.user
+    uri = 'user/following/{user}'
 
 
 class Unfollow(Request):
 
-    resource = User
-
-    def validate(self):
-        if not self.user:
-            raise RequestValidationError(
-                "'%s' request needs a user" % self.__class__.__name__)
-
-    def set_uri(self):
-        return 'user/following/%s' % self.user
-
-    def get_data(self):
-        pass
+    uri = 'user/following/{user}'
diff --git a/pygithub3/requests/users/keys.py b/pygithub3/requests/users/keys.py
index f5bb386..9afa3cd 100644
--- a/pygithub3/requests/users/keys.py
+++ b/pygithub3/requests/users/keys.py
@@ -1,79 +1,36 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from . import Request, json, RequestValidationError
+from . import Request
 from pygithub3.resources.users import Key
-from pygithub3.resources.base import Raw
 
 
 class List(Request):
 
     resource = Key
-
-    def validate(self):
-        pass
-
-    def set_uri(self):
-        return 'user/keys'
+    uri = 'user/key'
 
 
 class Get(Request):
 
     resource = Key
-
-    def validate(self):
-        if not self.key_id:
-            raise RequestValidationError("'%s' needs key_id"
-                                         % self.__class__.__name__)
-
-    def set_uri(self):
-        return 'user/keys/%s' % self.key_id
+    uri = 'user/keys/{key_id}'
 
 
 class Add(Request):
 
     resource = Key
-    valid = ('title', 'key')
-
-    def validate(self):
-        self.add_data = self._parse_simple_dict(self.add_data)
-
-    def set_uri(self):
-        return 'user/keys'
-
-    def get_data(self):
-        return json.dumps(self.add_data)
+    uri = 'user/keys'
+    body_schema = ('title', 'key')
 
 
 class Update(Request):
 
     resource = Key
-    valid = ('title', 'key')
-
-    def validate(self):
-        if not self.key_id:
-            raise RequestValidationError("'%s' needs key_id"
-                                         % self.__class__.__name__)
-        self.update_params = self._parse_simple_dict(self.update_with)
-
-    def set_uri(self):
-        return 'user/keys/%s' % self.key_id
-
-    def get_data(self):
-        return json.dumps(self.update_params)
+    body_schema = ('title', 'key')
+    uri = 'user/keys/{key_id}'
 
 
 class Delete(Request):
 
-    resource = Raw
-
-    def validate(self):
-        if not self.key_id:
-            raise RequestValidationError("'%s' needs key_id"
-                                         % self.__class__.__name__)
-
-    def set_uri(self):
-        return 'user/keys/%s' % self.key_id
-
-    def get_data(self):
-        return None
+    uri = 'user/keys/{key_id}'
diff --git a/pygithub3/requests/users/user.py b/pygithub3/requests/users/user.py
index d4b3667..78ab6cd 100644
--- a/pygithub3/requests/users/user.py
+++ b/pygithub3/requests/users/user.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from . import Request, json
+from . import Request, ValidationError
 from pygithub3.resources.users import User
 
 __all__ = ('Get', 'Update')
@@ -10,27 +10,16 @@ __all__ = ('Get', 'Update')
 class Get(Request):
 
     resource = User
+    uri = 'users/{user}'
 
-    def validate(self):
-        pass
-
-    def set_uri(self):
-        if self.user:
-            return 'users/%s' % self.user
-        else:
+    def clean_uri(self):
+        if not self.user:
             return 'user'
 
 
 class Update(Request):
 
     resource = User
-    valid = ('name', 'email', 'blog', 'company', 'location', 'hireable', 'bio')
-
-    def validate(self):
-        self.update_params = self._parse_simple_dict(self.update_with)
-
-    def get_data(self):
-        return json.dumps(self.update_params)
-
-    def set_uri(self):
-        return 'user'
+    uri = 'user'
+    body_schema = (
+        'name', 'email', 'blog', 'company', 'location','hireable', 'bio')