Fix: Also convert uuid strings to UUID objects

GetStream · Feb 27, 2017 · 51084c1 · 51084c1
1 parent 438e29d
commit 51084c1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/stream_django/enrich.py b/stream_django/enrich.py
@@ -2,6 +2,7 @@
 from collections import defaultdict
 import operator
 import itertools
+import uuid
 
 try:
     from django.apps import apps
@@ -122,7 +123,10 @@ def _inject_objects(self, activities, objects, fields):
             try:
                 f_id = int(f_id)
             except ValueError:
-                pass
+                try:
+                    f_id = uuid.UUID(f_id)
+                except ValueError:
+                    pass
 
             instance = objects[f_ct].get(f_id)
             if instance is None: