PYield 0.2.12__tar.gz → 0.2.14__tar.gz

{pyield-0.2.12 → pyield-0.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PYield
-Version: 0.2.12
+Version: 0.2.14
 Summary: A Python library for analysis of fixed income instruments in Brazil
 Keywords: fixed-income, brazil, finance, analysis, bonds
 Author-Email: Carlos Carvalho <cr.cj@outlook.com>

pyield-0.2.14/pyield/__about__.py

@@ -0,0 +1 @@
+__version__ = "0.2.14"

{pyield-0.2.12 → pyield-0.2.14}/pyield/di_futures.py

@@ -2,7 +2,7 @@ from typing import Literal
 from pathlib import Path
 import pandas as pd
 
-from . import di_html
+from . import di_url
 from . import di_xml
 from . import br_calendar as bc
 
@@ -109,7 +109,7 @@ def get_di(
     if source_type == "xml":
         df = di_xml.get_di(reference_date, data_path, return_raw)
     elif source_type == "html":
-        df = di_html.get_di(reference_date, return_raw)
+        df = di_url.get_di(reference_date, return_raw)
     else:
         raise ValueError("source_type must be either 'xml' or 'html'.")
 

pyield-0.2.12/pyield/di_html.py → pyield-0.2.14/pyield/di_url.py

@@ -112,7 +112,7 @@ def get_raw_di(reference_date: str | pd.Timestamp) -> pd.DataFrame:
 
     except Exception as e:
         warnings.warn(
-            f"A {type(e).__name__} occurred while reading the DI futures data for {reference_date.strftime("%d/%m/%Y")}. Returning an empty DataFrame."
+            f"A {type(e).__name__} occurred while reading the DI futures data for {reference_date.strftime('%d/%m/%Y')}. Returning an empty DataFrame."
         )
         return pd.DataFrame()
 

{pyield-0.2.12 → pyield-0.2.14}/pyield/di_xml.py

@@ -34,14 +34,19 @@ def extract_xml_file(zip_file: io.BytesIO) -> io.BytesIO:
 
     # Then, read the inner file
     with zipfile.ZipFile(outer_file, "r") as inner_zip:
-        inner_file_name = inner_zip.namelist()[0]
-        inner_file_content = inner_zip.read(inner_file_name)
+        filenames = inner_zip.namelist()
+        # Get first file that ends with ".xml"
+        xml_filename = [name for name in filenames if name.endswith(".xml")][0]
+        inner_file_content = inner_zip.read(xml_filename)
 
     return io.BytesIO(inner_file_content)
 
 
 def parse_xml_file(xml_file: io.BytesIO) -> list:
-    tree = etree.parse(xml_file)
+    parser = etree.XMLParser(
+        ns_clean=True, remove_blank_text=True, remove_comments=True, recover=True
+    )
+    tree = etree.parse(xml_file, parser)
     # Definir os namespaces (substitua 'ns' pelo prefixo apropriado se necessário)
     namespaces = {"ns": "urn:bvmf.217.01.xsd"}
 

{pyield-0.2.12 → pyield-0.2.14}/pyproject.toml

@@ -31,7 +31,7 @@ dependencies = [
     "lxml",
 ]
 dynamic = []
-version = "0.2.12"
+version = "0.2.14"
 
 [project.license]
 file = "LICENSE"

{pyield-0.2.12 → pyield-0.2.14}/tests/test_di.py

@@ -1,13 +1,13 @@
 import pandas as pd
 
 from pyield import di_futures as dif
-from pyield import di_html as dih
+from pyield import di_url as diu
 
 
 def test_valid_old_contract_code1():
     contact_code = "JAN3"  # Valid contract code
     reference_date = pd.Timestamp("2001-05-21")
-    result = dih.get_old_expiration_date(contact_code, reference_date)
+    result = diu.get_old_expiration_date(contact_code, reference_date)
     contract_expiration = pd.Timestamp("2003-01-02")
     assert result == contract_expiration
 
@@ -15,7 +15,7 @@ def test_valid_old_contract_code1():
 def test_valid_old_contract_code2():
     contact_code = "JAN3"  # Valid contract code
     reference_date = pd.Timestamp("1990-01-01")
-    result = dih.get_old_expiration_date(contact_code, reference_date)
+    result = diu.get_old_expiration_date(contact_code, reference_date)
     contract_expiration = pd.Timestamp("1993-01-04")
     assert result == contract_expiration
 
@@ -24,7 +24,7 @@ def test_invalid_old_contract_code():
     contact_code = "J3"  # Invalid contract code
     reference_date = pd.Timestamp("2001-01-02")
     # Must return NaT
-    result = dih.get_old_expiration_date(contact_code, reference_date)
+    result = diu.get_old_expiration_date(contact_code, reference_date)
     assert pd.isnull(result)
 
 
@@ -71,3 +71,8 @@ def test_settlement_rates_with_current_holiday_list():
     contract_codes = list(settlement_rates.keys())  # noqa: F841
     results = df.query("contract_code in @contract_codes")["settlement_rate"].to_list()
     assert results == list(settlement_rates.values())
+
+
+def test_invalid_date():
+    result = dif.get_di(reference_date="2023-12-24")
+    assert result.empty

pyield-0.2.12/pyield/__about__.py

@@ -1 +0,0 @@
-__version__ = "0.2.12"