first commit

Author: amar-jaiswal-haptik

.gitignore

@@ -0,0 +1,9 @@
+**sublime**
+*.pyc
+dist
+*.egg-info
+dist
+MANIFEST
+build
+.eggs
+venv

LICENSE

@@ -0,0 +1 @@
+# Interakt.Track.PythonSdk

README.md

@@ -0,0 +1,6 @@
+[metadata]
+description-file = README.md
+license_file = LICENSE
+
+[bdist_wheel]
+universal = 1

setup.cfg

@@ -0,0 +1,70 @@
+
+import os
+import sys
+
+try:
+    from setuptools import setup
+except ImportError:
+    from distutils.core import setup
+
+# Don't import interakt-track-python module here, since deps may not be installed
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'track'))
+from track.version import VERSION  # noqa
+
+
+# 'setup.py publish' shortcut
+if sys.argv[-1] == 'publish':
+    os.system('python3 setup.py sdist bdist_wheel')
+    os.system('twine upload dist/*')
+    sys.exit()
+
+# TODO: update long description in README.md
+with open(file='README.md', mode='r', encoding='utf-8') as f:
+    readme = f.read()
+
+install_requires = [
+    "requests>=2.20,<3.0"
+]
+
+tests_require = [
+    "mock==4.0.2",
+    "pylint==2.6.0",
+    "flake8==3.8.3",
+    "coverage==5.2.1"
+]
+
+
+setup(
+    name='interakt-track-python',
+    packages=['track'],
+    version=VERSION,
+    # TODO : Add repository url
+    url='',
+    author="Amar Jaiswal",
+    author_email="amar.j@cawstudios.com",
+    maintainer="interakt.ai",
+    # Chose a license from here: https://help.github.com/articles/licensing-a-repository
+    # TODO : Add License
+    license='',
+    description='The easy way to integrate track apis for interakt',
+    keywords=['INTERAKT', 'KIWI'],
+    install_requires=install_requires,
+    long_description=readme,
+    long_description_content_type='text/markdown',
+    classifiers=[
+        # TODO: Update Dev status
+        # Chose either "3 - Alpha", "4 - Beta" or "5 - Production/Stable"
+        'Development Status :: 3 - Alpha',
+        'Intended Audience :: Developers',
+        'Topic :: Software Development :: Build Tools',
+        # TODO: Add license
+        # 'License :: OSI Approved :: MIT License',   # Again, pick a license
+        "Operating System :: OS Independent",
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.4',
+        'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
+    ],
+)

setup.py

@@ -0,0 +1,32 @@
+from track.version import VERSION
+from track.client import Client
+__version__ = VERSION
+
+"""Settings"""
+write_key = None
+default_client = None
+host = None
+sync_mode = True
+debug = False
+timeout = 10
+
+
+def identify(user_id=None, country_code='+91', phone_number=None, traits={}):
+    """Send an identify call for customer"""
+    return _proxy('identify', user_id=user_id, country_code=country_code, phone_number=phone_number, traits=traits)
+
+
+def event(user_id=None, event=None, traits={}):
+    """Send an event track call."""
+    return _proxy('event', user_id=user_id, event=event, traits=traits)
+
+
+def _proxy(method, *args, **kwargs):
+    """Create an analytics client if one doesn't exist and send to it."""
+    global default_client
+    if not default_client:
+        default_client = Client(
+            write_key, host=host, debug=debug, sync_mode=sync_mode, timeout=timeout)
+
+    fn = getattr(default_client, method)
+    return fn(*args, **kwargs)

track/__init__.py

@@ -0,0 +1,54 @@
+from track.utils import require, stringify
+from track.request import post
+from track.const import ApiPaths
+import logging
+import numbers
+ID_TYPES = (numbers.Number, str)
+
+
+class Client(object):
+    logger = logging.getLogger('interakt')
+
+    def __init__(self, write_key=None, host=None, debug=False, sync_mode=True, timeout=10):
+        """Create a new interakt client"""
+        require('write_key', write_key, str)
+        self.write_key = write_key
+        self.debug = debug
+        self.host = host
+        self.sync_mode = sync_mode
+        self.timeout = timeout
+        if debug:
+            self.logger.setLevel(logging.DEBUG)
+
+    def identify(self, user_id=None, country_code='+91', phone_number=None, traits={}):
+        """Tie a user to their actions and record traits about them."""
+        if not user_id and not phone_number:
+            raise AssertionError("Either user_id or phone_number is required")
+        if user_id:
+            require('user_id', user_id, ID_TYPES)
+        if phone_number:
+            require('phone_number', phone_number, str)
+        require('traits', traits, dict)
+        msg = {
+            'userId': stringify(val=user_id),
+            'countryCode': country_code,
+            'phoneNumber': phone_number,
+            'traits': traits
+        }
+        return self.__send_request(path=ApiPaths.Identify.value, msg=msg)
+
+    def event(self, user_id=None, event=None, traits={}):
+        """To record user events"""
+        traits = traits or {}
+        require('user_id', user_id, ID_TYPES)
+        require('traits', traits, dict)
+        require('event', event, str)
+        msg = {
+            'userId': stringify(val=user_id),
+            'event': event,
+            'traits': traits
+        }
+        return self.__send_request(path=ApiPaths.Event.value, msg=msg)
+
+    def __send_request(self, path, msg):
+        return post(self.write_key, host=self.host, path=path, body=msg, timeout=self.timeout)

track/client.py

@@ -0,0 +1,8 @@
+import enum
+
+DEFAULT_HOST = 'https://api.interakt.ai'
+
+
+class ApiPaths(enum.Enum):
+    Identify = '/v1/public/track/users/'
+    Event = '/v1/public/track/events/'

track/const.py

@@ -0,0 +1,27 @@
+import logging
+
+from requests import sessions
+from requests.auth import HTTPBasicAuth
+
+from track.utils import remove_trailing_slash
+from track.version import VERSION
+from track.const import DEFAULT_HOST
+
+_session = sessions.session()
+logger = logging.getLogger('interakt')
+
+
+def post(write_key, host=None, path=None, body=None, timeout=10):
+    """Post the msg to the API"""
+    headers = {
+        'Content-Type': 'application/json',
+        'User-Agent': f'interakt-track-python/{VERSION}',
+        'Authorization': write_key
+    }
+    url = remove_trailing_slash(host or DEFAULT_HOST) + path
+    logger.debug(f'Making request: {body}')
+    response = _session.post(url=url, headers=headers,
+                             json=body, timeout=timeout)
+    payload = response.json()
+    logger.debug(f'Received response: {payload}')
+    return response

track/request.py

@@ -0,0 +1,24 @@
+import logging
+
+logger = logging.getLogger('interakt')
+
+
+def require(name, field, data_type):
+    """Require that the named `field` has the right `data_type`"""
+    if not isinstance(field, data_type):
+        msg = '{0} must have {1}, got: {2}'.format(name, data_type, field)
+        raise AssertionError(msg)
+
+
+def remove_trailing_slash(host):
+    if host.endswith('/'):
+        return host[:-1]
+    return host
+
+
+def stringify(val):
+    if val is None:
+        return None
+    if isinstance(val, str):
+        return val
+    return str(val)