Groups: Feature flags support (#45)

* Fix a documentation typo

* Feature flags & groups support

* Update examples

Author: macobo

example.py

@@ -17,6 +17,7 @@
 posthog.capture("distinct_id", "event", {"property1": "value", "property2": "value"})
 
 print(posthog.feature_enabled("beta-feature", "distinct_id"))
+print(posthog.feature_enabled("beta-feature", "distinct_id", groups={"company": "id:5"}))
 
 print("sleeping")
 time.sleep(5)
@@ -28,10 +29,16 @@
 posthog.alias("distinct_id", "new_distinct_id")
 
 posthog.capture("new_distinct_id", "event2", {"property1": "value", "property2": "value"})
+posthog.capture(
+    "new_distinct_id", "event-with-groups", {"property1": "value", "property2": "value"}, groups={"company": "id:5"}
+)
 
 # # Add properties to the person
 posthog.identify("new_distinct_id", {"email": "[email protected]"})
 
+# Add properties to a group
+posthog.group_identify("company", "id:5", {"employees": 11})
+
 # properties set only once to the person
 posthog.set_once("new_distinct_id", {"self_serve_signup": True})
 

posthog/__init__.py

@@ -172,7 +172,7 @@ def group_identify(
     """
     Set properties on a group
 
-     A `set_once` call requires
+     A `group_identify` call requires
      - `group_type` type of your group
      - `group_key` unique identifier of the group
      - `properties` with a dict with any key: value pairs
@@ -233,6 +233,7 @@ def feature_enabled(
     key,  # type: str,
     distinct_id,  # type: str,
     default=False,  # type: bool
+    groups={},  # type: dict
 ):
     # type: (...) -> bool
     """
@@ -242,11 +243,13 @@ def feature_enabled(
     ```python
     if posthog.feature_enabled('beta feature', 'distinct id'):
         # do something
+    if posthog.feature_enabled('groups feature', 'distinct id', groups={"organization": "id:5"}):
+        # do something
     ```
 
     You can call `posthog.load_feature_flags()` before to make sure you're not doing unexpected requests.
     """
-    return _proxy("feature_enabled", key=key, distinct_id=distinct_id, default=default)
+    return _proxy("feature_enabled", key=key, distinct_id=distinct_id, default=default, groups=groups)
 
 
 def page(*args, **kwargs):

posthog/client.py

@@ -348,9 +348,10 @@ def load_feature_flags(self):
             self.poller = Poller(interval=timedelta(seconds=self.poll_interval), execute=self._load_feature_flags)
             self.poller.start()
 
-    def feature_enabled(self, key, distinct_id, default=False):
+    def feature_enabled(self, key, distinct_id, default=False, *, groups={}):
         require("key", key, string_types)
         require("distinct_id", distinct_id, ID_TYPES)
+        require("groups", groups, dict)
 
         if not self.personal_api_key:
             self.log.warning("[FEATURE FLAGS] You have to specify a personal_api_key to use feature flags.")
@@ -375,6 +376,7 @@ def feature_enabled(self, key, distinct_id, default=False):
                     request_data = {
                         "distinct_id": distinct_id,
                         "personal_api_key": self.personal_api_key,
+                        "groups": groups,
                     }
                     resp_data = decide(self.api_key, self.host, timeout=10, **request_data)
                     response = key in resp_data["featureFlags"]