Initial commit

venv/lib/python3.10/site-packages/edgar/thirteenf/parsers/primary_xml.py

@@ -0,0 +1,118 @@
+"""Parser for 13F primary document XML format."""
+
+from datetime import datetime
+from decimal import Decimal
+from functools import lru_cache
+
+from edgar._party import Address
+from edgar.thirteenf.models import (
+    FilingManager,
+    OtherManager,
+    CoverPage,
+    SummaryPage,
+    Signature,
+    PrimaryDocument13F
+)
+from edgar.xmltools import child_text, find_element
+
+__all__ = ['parse_primary_document_xml']
+
+
+@lru_cache(maxsize=8)
+def parse_primary_document_xml(primary_document_xml: str):
+    """
+    Parse the primary 13F XML document.
+
+    Args:
+        primary_document_xml: XML content of the primary document
+
+    Returns:
+        PrimaryDocument13F: Parsed primary document data
+    """
+    root = find_element(primary_document_xml, "edgarSubmission")
+    # Header data
+    header_data = root.find("headerData")
+    filer_info = header_data.find("filerInfo")
+    report_period = datetime.strptime(child_text(filer_info, "periodOfReport"), "%m-%d-%Y")
+
+    # Form Data
+    form_data = root.find("formData")
+    cover_page_el = form_data.find("coverPage")
+
+    report_calendar_or_quarter = child_text(form_data, "reportCalendarOrQuarter")
+    report_type = child_text(cover_page_el, "reportType")
+
+    # Filing Manager
+    filing_manager_el = cover_page_el.find("filingManager")
+
+    # Address
+    address_el = filing_manager_el.find("address")
+    address = Address(
+        street1=child_text(address_el, "street1"),
+        street2=child_text(address_el, "street2"),
+        city=child_text(address_el, "city"),
+        state_or_country=child_text(address_el, "stateOrCountry"),
+        zipcode=child_text(address_el, "zipCode")
+    )
+    filing_manager = FilingManager(name=child_text(filing_manager_el, "name"), address=address)
+    # Other managers
+    other_manager_info_el = cover_page_el.find("otherManagersInfo")
+    other_managers = [
+        OtherManager(
+            cik=child_text(other_manager_el, "cik"),
+            name=child_text(other_manager_el, "name"),
+            file_number=child_text(other_manager_el, "form13FFileNumber")
+        )
+        for other_manager_el in other_manager_info_el.find_all("otherManager")
+    ] if other_manager_info_el else []
+
+    # Summary Page
+    summary_page_el = form_data.find("summaryPage")
+    if summary_page_el:
+        other_included_managers_count = child_text(summary_page_el,
+                                                   "otherIncludedManagersCount")
+        if other_included_managers_count:
+            other_included_managers_count = int(other_included_managers_count)
+
+        total_holdings = child_text(summary_page_el, "tableEntryTotal")
+        if total_holdings:
+            total_holdings = int(total_holdings)
+
+        total_value = child_text(summary_page_el, "tableValueTotal")
+        if total_value:
+            total_value = Decimal(total_value)
+    else:
+        other_included_managers_count = 0
+        total_holdings = 0
+        total_value = 0
+
+    # Signature Block
+    signature_block_el = form_data.find("signatureBlock")
+    signature = Signature(
+        name=child_text(signature_block_el, "name"),
+        title=child_text(signature_block_el, "title"),
+        phone=child_text(signature_block_el, "phone"),
+        city=child_text(signature_block_el, "city"),
+        signature=child_text(signature_block_el, "signature"),
+        state_or_country=child_text(signature_block_el, "stateOrCountry"),
+        date=child_text(signature_block_el, "signatureDate")
+    )
+
+    parsed_primary_doc = PrimaryDocument13F(
+        report_period=report_period,
+        cover_page=CoverPage(
+            filing_manager=filing_manager,
+            report_calendar_or_quarter=report_calendar_or_quarter,
+            report_type=report_type,
+            other_managers=other_managers
+        ),
+        signature=signature,
+        summary_page=SummaryPage(
+            other_included_managers_count=other_included_managers_count or 0,
+            total_holdings=total_holdings or 0,
+            total_value=total_value or 0
+        ),
+        additional_information=child_text(cover_page_el, "additionalInformation")
+    )
+
+    return parsed_primary_doc