Breaking tag functionality into generic handlers.

jkeyes · jkeyes · commit b6859696b7f7 · 2015-03-28T11:40:05.000Z
diff --git a/intercom/generic_handlers/base_handler.py b/intercom/generic_handlers/base_handler.py
@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+
+import inspect
+
+
+class BaseHandler(type):
+
+    def __getattr__(cls, name):  # noqa
+        # ignore underscore attrs
+        if name[0] == "_":
+            return
+
+        import time
+        a = time.time()
+        # get the class heirarchy
+        klasses = inspect.getmro(cls)
+        b = time.time()
+        # find a class that can handle this attr
+        for klass in klasses:
+            if hasattr(klass, 'handles_attr') and klass.handles_attr(name):
+                c = time.time()
+                d = b - a
+                e = c - b
+                print("GETMRO %s HANDLE %s" % (d, e))
+                return klass._get(cls, name)
diff --git a/intercom/generic_handlers/tag.py b/intercom/generic_handlers/tag.py
@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+
+from intercom import utils
+
+
+class TagHandler():
+    def __init__(self, entity, name, context):
+        self.entity = entity
+        self.untag = name == "untag"
+        self.context = context
+
+    def __call__(self, *args, **kwargs):
+        return self.entity._tag_collection(
+            self.context, *args, untag=self.untag, **kwargs)
+
+
+class TagUntag():
+
+    @classmethod
+    def handles_attr(cls, name):
+        name, context = name.split('_', 1)
+        if name in ["tag", "untag"]:
+            return True
+
+    @classmethod
+    def _get(cls, entity, name):
+        name, context = name.split('_', 1)
+        return TagHandler(entity, name, context)
+
+    @classmethod
+    def _tag_collection(
+            cls, collection_name, name, objects, untag=False):
+        from intercom import Intercom
+        collection = utils.resource_class_to_collection_name(cls)
+        object_ids = []
+        for obj in objects:
+            if not hasattr(obj, 'keys'):
+                obj = {'id': obj}
+            if untag:
+                obj['untag'] = True
+            object_ids.append(obj)
+
+        params = {
+            'name': name,
+            collection_name: object_ids,
+        }
+        response = Intercom.post("/%s" % (collection), **params)
+        return cls(**response)
diff --git a/intercom/generic_handlers/tag_find_all.py b/intercom/generic_handlers/tag_find_all.py
@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+
+import re
+
+
+class FindAllHandler():
+    def __init__(self, entity, context):
+        self.entity = entity
+        self.context = context
+
+    def __call__(self, *args, **kwargs):
+        return self.entity._find_all_for(
+            self.context, *args, **kwargs)
+
+
+class TagFindAll():
+
+    find_matcher = re.compile(r'find_all_for_(\w+)')
+
+    @classmethod
+    def handles_attr(cls, name):
+        return cls.find_matcher.search(name) is not None
+
+    @classmethod
+    def _get(cls, entity, name):
+        match = cls.find_matcher.search(name)
+        context = match.group(1)
+        return FindAllHandler(entity, context)
+
+    @classmethod
+    def _find_all_for(cls, taggable_type, **kwargs):
+        params = {
+            'taggable_type': taggable_type
+        }
+        res_id = kwargs.pop('id', None)
+        if res_id:
+            params['taggable_id'] = res_id
+        params.update(kwargs)
+
+        return cls.find_all(**params)
diff --git a/intercom/tag.py b/intercom/tag.py
@@ -1,68 +1,18 @@
 # -*- coding: utf-8 -*-
 
-from intercom import utils
+import six
+
 from intercom.api_operations.all import All
 from intercom.api_operations.count import Count
 from intercom.api_operations.find import Find
 from intercom.api_operations.find_all import FindAll
 from intercom.api_operations.save import Save
+from intercom.generic_handlers.base_handler import BaseHandler
+from intercom.generic_handlers.tag import TagUntag
+from intercom.generic_handlers.tag_find_all import TagFindAll
 from intercom.traits.api_resource import Resource
 
 
-class Tag(Resource, All, Count, Find, FindAll, Save):
-
-    @classmethod
-    def _tag_collection(cls, name, collection_name, objects, untag=False):
-        from intercom import Intercom
-        collection = utils.resource_class_to_collection_name(cls)
-
-        object_ids = []
-        for obj in objects:
-            if not hasattr(obj, 'keys'):
-                obj = {'id': obj}
-            if untag:
-                obj['untag'] = True
-            object_ids.append(obj)
-
-        params = {
-            'name': name,
-            collection_name: object_ids,
-        }
-        response = Intercom.post("/%s" % (collection), **params)
-        return cls(**response)
-
-    @classmethod
-    def tag_users(cls, name, users):
-        return cls._tag_collection(name, 'users', users)
-
-    @classmethod
-    def untag_users(cls, name, users):
-        return cls._tag_collection(name, 'users', users, untag=True)
-
-    @classmethod
-    def tag_companies(cls, name, companies):
-        return cls._tag_collection(name, 'companies', companies)
-
-    @classmethod
-    def untag_companies(cls, name, companies):
-        return cls._tag_collection(name, 'companies', companies, untag=True)
-
-    @classmethod
-    def find_all_for_user(cls, **kwargs):
-        return cls.find_all_for('user', **kwargs)
-
-    @classmethod
-    def find_all_for_company(cls, **kwargs):
-        return cls.find_all_for('company', **kwargs)
-
-    @classmethod
-    def find_all_for(cls, taggable_type, **kwargs):
-        params = {
-            'taggable_type': taggable_type
-        }
-        res_id = kwargs.pop('id', None)
-        if res_id:
-            params['taggable_id'] = res_id
-        params.update(kwargs)
-
-        return Tag.find_all(**params)
+@six.add_metaclass(BaseHandler)
+class Tag(Resource, All, Count, Find, FindAll, Save, TagUntag, TagFindAll):
+    pass
