pyield 0.48.8__tar.gz → 0.48.9__tar.gz

{pyield-0.48.8 → pyield-0.48.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyield
-Version: 0.48.8
+Version: 0.48.9
 Summary: A Python library for analysis of fixed income instruments in Brazil
 Project-URL: Homepage, https://github.com/crdcj/PYield
 Project-URL: Documentation, https://crdcj.github.io/PYield

pyield-0.48.9/pyield/__about__.py

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

{pyield-0.48.8 → pyield-0.48.9}/pyield/b3/boletim.py

@@ -102,6 +102,13 @@ COLUNAS_PRICE_REPORT: list[tuple[str, str, type[pl.DataType]]] = [
 SCHEMA_PRICE_REPORT = {nome: tipo for _, nome, tipo in COLUNAS_PRICE_REPORT}
 
 
+_SESSAO = requests.Session()
+_SESSAO.headers["User-Agent"] = (
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
+    " (KHTML, like Gecko) Chrome/144.0.0.0 Safari/537.36"
+)
+
+
 @ttl_cache()
 @retry_padrao
 def _baixar_zip_url(data: dt.date, boletim_completo: bool) -> bytes:
@@ -113,7 +120,7 @@ def _baixar_zip_url(data: dt.date, boletim_completo: bool) -> bytes:
             f"https://www.b3.com.br/pesquisapregao/download?filelist=SPRD{data_str}.zip"
         )
 
-    resposta = requests.get(url, timeout=(5, 30))
+    resposta = _SESSAO.get(url, timeout=(5, 30))
     resposta.raise_for_status()
 
     if len(resposta.content) < MIN_TAMANHO_ZIP_BYTES:

{pyield-0.48.8 → pyield-0.48.9}/pyield/b3/futuro/__init__.py

@@ -13,12 +13,12 @@ def futuro_enriquecer(
     """Enriquece DataFrame bruto do Price Report (PR) da B3.
 
     Aceita um DataFrame com colunas no schema original da B3
-    (ex.: ``TradDt``, ``TckrSymb``) ou já renomeadas para o padrão
-    PYield. Adiciona data de vencimento, dias úteis/corridos e
-    colunas derivadas (dv01, taxa_forward) conforme o contrato.
+    (ex.: ``TradDt``, ``TckrSymb``). Filtra pelo contrato informado,
+    adiciona data de vencimento, dias úteis/corridos e colunas
+    derivadas (dv01, taxa_forward) conforme o contrato.
 
     Args:
-        df: DataFrame com dados do PR da B3.
+        df: DataFrame com dados brutos do PR da B3.
         contrato: Contrato futuro
             (ex.: "DI1", "DOL").
 

{pyield-0.48.8 → pyield-0.48.9}/pyield/b3/futuro/historico.py

@@ -76,6 +76,9 @@ _COLUNAS_CONTRATO_TAXA = (
     "taxa_forward",
 )
 
+# Comprimento padrão de ticker de futuro na B3 (prefixo 3 + mês 1 + ano 2).
+_COMPRIMENTO_TICKER = 6
+
 # Normaliza o schema XML bruto da B3 para o padrão de colunas deste módulo.
 _RENOMEAR_COLUNAS_PR = {
     "TradDt": "data_referencia",
@@ -125,7 +128,11 @@ def _enriquecer_dados(df: pl.DataFrame, contrato: str) -> pl.DataFrame:
 
     if contrato in {"DI1", "DAP"}:
         df = df.with_columns(
-            taxa_forward=forwards(dias_uteis=df["dias_uteis"], taxas=df["taxa_ajuste"])
+            taxa_forward=forwards(
+                dias_uteis=df["dias_uteis"],
+                taxas=df["taxa_ajuste"],
+                agrupar_por=df["data_referencia"],
+            )
         )
 
     return df
@@ -156,12 +163,12 @@ def enriquecer(df: pl.DataFrame, contrato: str) -> pl.DataFrame:
     """Enriquece DataFrame bruto do Price Report (PR) da B3.
 
     Aceita um DataFrame com colunas no schema original da B3 (ex.:
-    ``TradDt``, ``TckrSymb``) ou já renomeadas para o padrão PYield.
-    Adiciona data de vencimento, dias úteis/corridos e colunas derivadas
-    (dv01, taxa_forward) conforme o contrato.
+    ``TradDt``, ``TckrSymb``). Filtra pelo contrato informado,
+    adiciona data de vencimento, dias úteis/corridos e colunas
+    derivadas (dv01, taxa_forward) conforme o contrato.
 
     Args:
-        df: DataFrame com dados do PR da B3.
+        df: DataFrame com dados brutos do PR da B3.
         contrato: Contrato futuro (ex.: "DI1", "DOL").
 
     Returns:
@@ -171,6 +178,12 @@ def enriquecer(df: pl.DataFrame, contrato: str) -> pl.DataFrame:
         return pl.DataFrame()
 
     df = df.rename(_RENOMEAR_COLUNAS_PR)
+    df = df.filter(
+        pl.col("codigo_negociacao").str.starts_with(contrato),
+        pl.col("codigo_negociacao").str.len_chars() == _COMPRIMENTO_TICKER,
+    )
+    if df.is_empty():
+        return pl.DataFrame()
     df = adicionar_vencimento(df, contrato, coluna_ticker="codigo_negociacao")
     df = _enriquecer_dados(df, contrato)
     df = _selecionar_colunas_saida(df, contrato)

pyield-0.48.8/pyield/__about__.py

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