Helper for creating Position entities

zavod/parse/positions.py

@@ -0,0 +1,47 @@
+from typing import Optional
+from nomenklatura.entity import CE
+from followthemoney.util import join_text, make_entity_id
+from normality import slugify
+
+
+def make_position(
+    context: GenericZavod[CE, ZD],
+    name: Optional[str] = None,
+    summary: Optional[str] = None,
+    description: Optional[str] = None,
+    country: Optional[str] = None,
+    subnationalArea: Optional[str] = None,
+    organization: Optional[str] = None,
+    inceptionDate: Optional[str] = None,
+    dissolutionDate: Optional[str] = None,
+    numberOfSeats: Optional[str] = None,
+    wikidataId: Optional[str] = None,
+    sourceUrl: Optional[str] = None,
+    lang: Optional[str] = None,
+) -> CE:
+  position = context.make("Position")
+
+  position.add("name", name, lang=lang)
+  position.add("summary", summary, lang=lang)
+  position.add("description", description, lang=lang)
+  position.add("country", country, lang=lang)
+  position.add("organization", organization, lang=lang)
+  position.add("subnationalArea", subnationalArea, lang=lang)
+  position.add("inceptionDate", inceptionDate)
+  position.add("dissolutionDate", dissolutionDate)
+  position.add("numberOfSeats", numberOfSeats)
+  position.add("wikidataId", wikidataId)
+  position.add("sourceUrl", sourceUrl)
+
+  parts = [
+    name, 
+    organization.id if organization else None,
+    country,
+    inceptionDate,
+    dissolutionDate,
+  ]
+  hash_id = make_entity_id(*parts)
+  if hash_id is not None:
+      position.id = f"pos-{hash_id}"
+
+  return position