Add entities to events

predicthq · Jul 4, 2019 · 51fc43b · 51fc43b
1 parent db2f113
commit 51fc43b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/predicthq/endpoints/v1/events/schemas.py b/predicthq/endpoints/v1/events/schemas.py
@@ -38,6 +38,17 @@ class Options:
     brand_unsafe = ModelType(BrandUnsafe)
 
 
+class Entities(Model):
+
+    class Options:
+        serialize_when_none = True
+
+    entity_id = StringType()
+    name = StringType()
+    type = StringType()
+    formatted_address = StringType()
+
+
 class Event(Model):
 
     class Options:
@@ -55,6 +66,7 @@ class Options:
     country = StringType()
     rank = IntType()
     local_rank = IntType()
+    entities = ListType(ModelType(Entities))
     location = GeoJSONPointType()
     place_hierarchies = ListType(ListType(StringType()))
     scope = StringType()