Fix: Allow UUID model IDs

GetStream · Feb 27, 2017 · 438e29d · 438e29d
1 parent 4d0acd0
commit 438e29d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/stream_django/enrich.py b/stream_django/enrich.py
@@ -118,7 +118,13 @@ def _inject_objects(self, activities, objects, fields):
             if not self.is_ref(activity, field):
                 continue
             f_ct, f_id = activity[field].split(':')
-            instance = objects[f_ct].get(int(f_id))
+
+            try:
+                f_id = int(f_id)
+            except ValueError:
+                pass
+
+            instance = objects[f_ct].get(f_id)
             if instance is None:
                 activity.track_not_enriched_field(field, activity[field])
             else: