Send distinctID to the decide endpoint to determin if user should have features enabled

Author: fuziontech

posthog/client.py

@@ -5,12 +5,13 @@
 import atexit
 import hashlib
 
+import requests
 from dateutil.tz import tzutc
 from six import string_types
 
 from posthog.utils import guess_timezone, clean
 from posthog.consumer import Consumer
-from posthog.request import post, get, APIError
+from posthog.request import batch_post, decide, get, APIError
 from posthog.version import VERSION
 from posthog.poller import Poller
 
@@ -188,7 +189,7 @@ def _enqueue(self, msg):
 
         if self.sync_mode:
             self.log.debug('enqueued with blocking %s.', msg['event'])
-            post(self.api_key, self.host, gzip=self.gzip,
+            batch_post(self.api_key, self.host, gzip=self.gzip,
                  timeout=self.timeout, batch=[msg])
 
             return True, msg
@@ -256,7 +257,6 @@ def load_feature_flags(self):
     def feature_enabled(self, key, distinct_id, default=False):
         require('key', key, string_types)
         require('distinct_id', distinct_id, ID_TYPES)
-        error = False
 
         if not self.personal_api_key:
             self.log.warning('[FEATURE FLAGS] You have to specify a personal_api_key to use feature flags.')
@@ -266,7 +266,6 @@ def feature_enabled(self, key, distinct_id, default=False):
         # If loading in previous line failed
         if not self.feature_flags:
             response = default
-            error = True
         else:
             try:
                 feature_flag = [flag for flag in self.feature_flags if flag['key'] == key][0]
@@ -277,13 +276,17 @@ def feature_enabled(self, key, distinct_id, default=False):
                 response = _hash(key, distinct_id) <= (feature_flag['rollout_percentage'] / 100)
             else:
                 try:
-                    request = get(self.api_key, '/decide/', self.host, timeout=1)
-                    response = key in request['featureFlags']
+                    request_data = {
+                        "distinct_id": distinct_id,
+                        "personal_api_key": self.personal_api_key,
+                    }
+                    request = decide(self.api_key, self.host, timeout=10, **request_data)
+                    resp_data = request.json()
+                    response = key in resp_data['featureFlags']
                 except Exception as e:
                     response = default
                     self.log.warning('[FEATURE FLAGS] Unable to get data for flag %s, because of the following error:' % key)
                     self.log.warning(e)
-                    error = True
 
         self.capture(distinct_id, '$feature_flag_called', {'$feature_flag': key, '$feature_flag_response': response})
         return response

posthog/consumer.py

@@ -4,7 +4,7 @@
 import backoff
 import json
 
-from posthog.request import post, APIError, DatetimeSerializer
+from posthog.request import batch_post, APIError, DatetimeSerializer
 
 try:
     from queue import Empty
@@ -128,7 +128,7 @@ def fatal_exception(exc):
             max_tries=self.retries + 1,
             giveup=fatal_exception)
         def send_request():
-            post(self.api_key, self.host, gzip=self.gzip,
+            batch_post(self.api_key, self.host, gzip=self.gzip,
                  timeout=self.timeout, batch=batch)
 
         send_request()

posthog/request.py

@@ -15,12 +15,13 @@
 DEFAULT_HOST = 'https://app.posthog.com'
 USER_AGENT = 'posthog-python/' + VERSION
 
-def post(api_key, host=None, gzip=False, timeout=15, **kwargs):
+
+def post(api_key, host=None, path=None, gzip=False, timeout=15, **kwargs):
     """Post the `kwargs` to the API"""
     log = logging.getLogger('posthog')
     body = kwargs
     body["sentAt"] = datetime.utcnow().replace(tzinfo=tzutc()).isoformat()
-    url = remove_trailing_slash(host or DEFAULT_HOST) + '/batch/'
+    url = remove_trailing_slash(host or DEFAULT_HOST) + path
     body['api_key'] = api_key
     data = json.dumps(body, cls=DatetimeSerializer)
     log.debug('making request: %s', data)
@@ -44,13 +45,38 @@ def post(api_key, host=None, gzip=False, timeout=15, **kwargs):
         log.debug('data uploaded successfully')
         return res
 
+
+def decide(api_key, host=None, gzip=False, timeout=15, **kwargs):
+    """Post the `kwargs to the decide API endpoint"""
+    log = logging.getLogger('posthog')
+    res = post(api_key, host, '/decide/', gzip, timeout, **kwargs)
+    if res.status_code == 200:
+        log.debug('Feature flags decided successfully')
+        return res
     try:
         payload = res.json()
         log.debug('received response: %s', payload)
         raise APIError(res.status_code, payload['detail'])
     except ValueError:
         raise APIError(res.status_code, res.text)
 
+
+def batch_post(api_key, host=None, gzip=False, timeout=15, **kwargs):
+    """Post the `kwargs` to the batch API endpoint for events"""
+    log = logging.getLogger('posthog')
+    res = post(api_key, host, '/batch/', gzip, timeout, **kwargs)
+
+    if res.status_code == 200:
+        log.debug('data uploaded successfully')
+        return res
+    try:
+        payload = res.json()
+        log.debug('received response: %s', payload)
+        raise APIError(res.status_code, payload['detail'])
+    except ValueError:
+        raise APIError(res.status_code, res.text)
+
+
 def get(api_key, url, host=None, timeout=None):
     log = logging.getLogger('posthog')
     url = remove_trailing_slash(host or DEFAULT_HOST) + url

posthog/test/request.py

@@ -3,14 +3,14 @@
 import json
 import requests
 
-from posthog.request import post, DatetimeSerializer
+from posthog.request import batch_post, DatetimeSerializer
 from posthog.test.utils import TEST_API_KEY
 
 
 class TestRequests(unittest.TestCase):
 
     def test_valid_request(self):
-        res = post(TEST_API_KEY, batch=[{
+        res = batch_post(TEST_API_KEY, batch=[{
             'distinct_id': 'distinct_id',
             'event': 'python event',
             'type': 'track'
@@ -38,7 +38,7 @@ def test_date_serialization(self):
         self.assertEqual(result, expected)
 
     def test_should_not_timeout(self):
-        res = post(TEST_API_KEY, batch=[{
+        res = batch_post(TEST_API_KEY, batch=[{
             'distinct_id': 'distinct_id',
             'event': 'python event',
             'type': 'track'
@@ -47,7 +47,7 @@ def test_should_not_timeout(self):
 
     def test_should_timeout(self):
         with self.assertRaises(requests.ReadTimeout):
-            post('key', batch=[{
+            batch_post('key', batch=[{
                 'distinct_id': 'distinct_id',
                 'event': 'python event',
                 'type': 'track'