corp-extractor 0.9.0__py3-none-any.whl → 0.9.3__py3-none-any.whl

statement_extractor/database/models.py

@@ -54,12 +54,17 @@ class PersonType(str, Enum):
     Used for categorizing people in the person database.
     """
     EXECUTIVE = "executive"  # CEOs, board members, C-suite
-    POLITICIAN = "politician"  # Elected officials, diplomats
+    POLITICIAN = "politician"  # Elected officials (presidents, MPs, mayors)
+    GOVERNMENT = "government"  # Civil servants, diplomats, appointed officials
+    MILITARY = "military"  # Military officers, armed forces personnel
+    LEGAL = "legal"  # Judges, lawyers, legal professionals
+    PROFESSIONAL = "professional"  # Known for their profession (doctors, engineers, architects)
     ACADEMIC = "academic"  # Professors, researchers
-    ARTIST = "artist"  # Musicians, actors, directors, writers
+    ARTIST = "artist"  # Traditional creatives (musicians, actors, painters, writers)
+    MEDIA = "media"  # Internet/social media personalities (YouTubers, influencers, podcasters)
     ATHLETE = "athlete"  # Sports figures
     ENTREPRENEUR = "entrepreneur"  # Founders, business owners
-    JOURNALIST = "journalist"  # Reporters, media personalities
+    JOURNALIST = "journalist"  # Reporters, news presenters, columnists
     ACTIVIST = "activist"  # Advocates, campaigners
     SCIENTIST = "scientist"  # Scientists, inventors
     UNKNOWN = "unknown"  # Type not determined
@@ -77,6 +82,8 @@ class CompanyRecord(BaseModel):
     source_id: str = Field(..., description="Unique identifier from source (LEI, CIK, CH number)")
     region: str = Field(default="", description="Geographic region/country (e.g., 'UK', 'US', 'DE')")
     entity_type: EntityType = Field(default=EntityType.UNKNOWN, description="Organization type classification")
+    from_date: Optional[str] = Field(default=None, description="Start date (ISO format YYYY-MM-DD)")
+    to_date: Optional[str] = Field(default=None, description="End date (ISO format YYYY-MM-DD)")
     record: dict[str, Any] = Field(default_factory=dict, description="Original record from source")
 
     @property
@@ -92,11 +99,13 @@ class CompanyRecord(BaseModel):
             "source_id": self.source_id,
             "region": self.region,
             "entity_type": self.entity_type.value,
+            "from_date": self.from_date or "",
+            "to_date": self.to_date or "",
             "record": self.record,
         }
 
 
-PersonSourceType = Literal["wikidata"]
+PersonSourceType = Literal["wikidata", "sec_edgar", "companies_house"]
 
 
 class PersonRecord(BaseModel):
@@ -111,8 +120,13 @@ class PersonRecord(BaseModel):
     source_id: str = Field(..., description="Unique identifier from source (Wikidata QID)")
     country: str = Field(default="", description="Country code or name (e.g., 'US', 'Germany')")
     person_type: PersonType = Field(default=PersonType.UNKNOWN, description="Person type classification")
-    known_for_role: str = Field(default="", description="Primary role from Wikipedia (e.g., 'CEO', 'President')")
-    known_for_org: str = Field(default="", description="Primary org from Wikipedia (e.g., 'Apple Inc', 'Tesla')")
+    known_for_role: str = Field(default="", description="Primary role (e.g., 'CEO', 'President')")
+    known_for_org: str = Field(default="", description="Primary org (e.g., 'Apple Inc', 'Tesla')")
+    known_for_org_id: Optional[int] = Field(default=None, description="Foreign key to organizations table")
+    from_date: Optional[str] = Field(default=None, description="Start date of role (ISO format YYYY-MM-DD)")
+    to_date: Optional[str] = Field(default=None, description="End date of role (ISO format YYYY-MM-DD)")
+    birth_date: Optional[str] = Field(default=None, description="Date of birth (ISO format YYYY-MM-DD)")
+    death_date: Optional[str] = Field(default=None, description="Date of death (ISO format YYYY-MM-DD) - if set, person is historic")
     record: dict[str, Any] = Field(default_factory=dict, description="Original record from source")
 
     @property
@@ -120,6 +134,11 @@ class PersonRecord(BaseModel):
         """Generate canonical ID in format source:source_id."""
         return f"{self.source}:{self.source_id}"
 
+    @property
+    def is_historic(self) -> bool:
+        """Return True if the person is deceased (has a death date)."""
+        return self.death_date is not None and self.death_date != ""
+
     def model_dump_for_db(self) -> dict[str, Any]:
         """Convert to dict suitable for database storage."""
         return {
@@ -130,6 +149,11 @@ class PersonRecord(BaseModel):
             "person_type": self.person_type.value,
             "known_for_role": self.known_for_role,
             "known_for_org": self.known_for_org,
+            "known_for_org_id": self.known_for_org_id,  # Can be None
+            "from_date": self.from_date or "",
+            "to_date": self.to_date or "",
+            "birth_date": self.birth_date or "",
+            "death_date": self.death_date or "",
             "record": self.record,
         }