luxorasap 0.1.2__tar.gz → 0.1.3__tar.gz

{luxorasap-0.1.2 → luxorasap-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: luxorasap
-Version: 0.1.2
+Version: 0.1.3
 Summary: Luxor’s unified toolbox for data ingestion, querying and analytics.
 Author-email: Luxor Group <backoffice@luxor.com.br>
 License: Proprietary – All rights reserved

{luxorasap-0.1.2 → luxorasap-0.1.3}/pyproject.toml

@@ -10,7 +10,7 @@ build-backend = "setuptools.build_meta"
 #############################
 [project]
 name            = "luxorasap"
-version = "0.1.2"
+version = "0.1.3"
 description     = "Luxor’s unified toolbox for data ingestion, querying and analytics."
 readme          = "README.md"
 requires-python = ">=3.9"
@@ -74,7 +74,7 @@ exclude = ["tests*"]
 # bumpver (sem-ver)
 #############################
 [tool.bumpver]
-current_version = "0.1.2"
+current_version = "0.1.3"
 version_pattern = "MAJOR.MINOR.PATCH"
 
 # regex explícito – obrigatório no bumpver 2024+

{luxorasap-0.1.2 → luxorasap-0.1.3}/src/luxorasap/__init__.py

@@ -13,7 +13,7 @@ from types import ModuleType
 try:
     __version__: str = metadata.version(__name__)
 except metadata.PackageNotFoundError:  # editable install
-    __version__ = "0.1.2"
+    __version__ = "0.1.3"
 
 # ─── Lazy loader ─────────────────────────────────────────────────
 def __getattr__(name: str) -> ModuleType:

{luxorasap-0.1.2 → luxorasap-0.1.3}/src/luxorasap/btgapi/__init__.py

@@ -1,7 +1,7 @@
 """Wrapper para as APIs do BTG Pactual."""
 
 from .auth import get_access_token, BTGApiError
-from .reports import request_portfolio, await_report_ticket_result, process_zip_to_dfs, request_investors_transactions_report
+from .reports import request_portfolio, await_report_ticket_result, process_zip_to_dfs, request_investors_transactions_report, request_fundflow_report
 from .trades import submit_offshore_equity_trades, await_transaction_ticket_result
 
 __all__ = [
@@ -12,5 +12,6 @@ __all__ = [
     "submit_offshore_equity_trades",
     "await_transaction_ticket_result",
     "process_zip_to_dfs",
-    "request_investors_transactions_report"
+    "request_investors_transactions_report",
+    "request_fundflow_report",
 ]

{luxorasap-0.1.2 → luxorasap-0.1.3}/src/luxorasap/btgapi/reports.py

@@ -16,7 +16,8 @@ __all__ = [
     "check_report_ticket",
     "await_report_ticket_result",
     "process_zip_to_dfs",
-    "request_investors_transactions_report"
+    "request_investors_transactions_report",
+    "request_fundflow_report"
 ]
 
 _REPORT_ENDPOINT = "https://funds.btgpactual.com/reports/Portfolio"
@@ -24,6 +25,7 @@ _TICKET_ENDPOINT = "https://funds.btgpactual.com/reports/Ticket"
 _INVESTOR_TX_ENDPOINT = (
     "https://funds.btgpactual.com/reports/RTA/InvestorTransactionsFileReport"
 )
+_FUNDFLOW_ENDPOINT = "https://funds.btgpactual.com/reports/RTA/FundFlow"
 _REPORT_TYPES = {"excel": 10, "xml5": 81, "pdf": 2}
 
 
@@ -105,6 +107,15 @@ def check_report_ticket(token: str, ticket: str, *, page: Optional[int] = None)
             return _download_url(url)
         except Exception as exc:
             raise BTGApiError(f"Falha ao interpretar resultado: {exc}") from exc
+    
+    # 4. result pode ser uma lista de dados
+    if isinstance(result, list):
+        # Vamos tentar transformar num dataframe e retornar
+        try:
+            df = pd.DataFrame(result)
+            return df
+        except Exception as exc:
+            raise BTGApiError(f"Falha ao converter resultado em DataFrame: {exc}") from exc
 
     raise BTGApiError("Formato de resposta desconhecido")
 
@@ -186,4 +197,43 @@ def request_investors_transactions_report( token: str, query_date: dt.date, *,
         return r.json()["ticket"]
     raise BTGApiError(
         f"Erro InvestorTransactionsFileReport: {r.status_code} – {r.text}"
-    )
+    )
+    
+    
+def request_fundflow_report( token: str, start_date: dt.date,
+        end_date: dt.date, *, fund_name: str = "", date_type: str = "LIQUIDACAO", page_size: int = 100) -> str:
+    """Dispara geração do **Fund Flow** (RTA) e devolve *ticket*.
+
+    Args:
+        token: JWT obtido via :pyfunc:`luxorasap.btgapi.get_access_token`.
+        start_date,end_date: Datas do intervalo desejado.
+        fund_name: Nome do fundo conforme BTG. String vazia retorna as movimentacoes para todos os fundos.
+        date_type: Enum da API (`LIQUIDACAO`, `MOVIMENTO`, etc.).
+        page_size: Página retornada por chamada (default 100).
+
+    Returns
+    -------
+    str
+        ID do ticket a ser acompanhado em :pyfunc:`await_fundflow_ticket_result`.
+    """
+
+    body = {
+        "contract": {
+            "startDate": f"{start_date}T00:00:00Z",
+            "endDate": f"{end_date}T00:00:00Z",
+            "dateType": date_type,
+            "fundName": fund_name,
+        },
+        "pageSize": page_size,
+        "webhookEndpoint": "string",
+    }
+
+    r = requests.post(
+        _FUNDFLOW_ENDPOINT,
+        headers={"X-SecureConnect-Token": token, "Content-Type": "application/json"},
+        json=body,
+        timeout=30,
+    )
+    if r.ok:
+        return r.json()["ticket"]
+    raise BTGApiError(f"Erro FundFlow: {r.status_code} - {r.text}")

{luxorasap-0.1.2 → luxorasap-0.1.3}/src/luxorasap/ingest/cloud/__init__.py

@@ -20,8 +20,15 @@ def save_table(
     index_name: str = "index",
     normalize_columns: bool = True,
     directory: str = "enriched/parquet",
+    override=False
 ):
     """Salva DataFrame como Parquet em ADLS (sobrescrevendo)."""
+    
+    if override == False:
+        lq = LuxorQuery()
+        if lq.table_exists(table_name):
+            return
+    
     df = prep_for_save(df, index=index, index_name=index_name, normalize=normalize_columns)
     _client.write_df(df.astype(str), f"{directory}/{table_name}.parquet")
 

{luxorasap-0.1.2 → luxorasap-0.1.3}/src/luxorasap.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: luxorasap
-Version: 0.1.2
+Version: 0.1.3
 Summary: Luxor’s unified toolbox for data ingestion, querying and analytics.
 Author-email: Luxor Group <backoffice@luxor.com.br>
 License: Proprietary – All rights reserved

luxorasap-0.1.3/tests/test_btgapi_reports.py

@@ -0,0 +1,139 @@
+import datetime as dt, io, json, zipfile
+import pandas as pd
+import pytest
+
+from luxorasap.btgapi.reports import (
+    request_portfolio,
+    await_report_ticket_result,
+    process_zip_to_dfs,
+    check_report_ticket,
+    request_fundflow_report,
+)
+
+_TICKET_URL = "https://funds.btgpactual.com/reports/Ticket"
+_POST_URL   = "https://funds.btgpactual.com/reports/Portfolio"
+_FUNDFLOW_URL = "https://funds.btgpactual.com/reports/RTA/FundFlow"
+
+_TOKEN = "dummy-token"
+
+def test_request_portfolio_returns_ticket(requests_mock):
+    requests_mock.post(_POST_URL, json={"ticket": "ABC"})
+    tk = request_portfolio("tok", "FUND", dt.date(2025,1,1), dt.date(2025,1,31))
+    assert tk == "ABC"
+
+def test_await_ticket_inline_zip(requests_mock, monkeypatch):
+    # 1ª chamada: ainda processando   2ª: devolve ZIP binário
+    # Explicando o que requests_mock faz:
+    # 
+    requests_mock.get(_TICKET_URL, [
+        {"json": {"result": "Processando"}},
+        {"content": b"ZIP!"},
+    ])
+    monkeypatch.setattr("time.sleep", lambda *_: None)
+    out = await_report_ticket_result("tok", "ABC", attempts=2, interval=0)
+    assert out == b"ZIP!"
+
+def test_await_ticket_via_urldownload(requests_mock, monkeypatch):
+    dl_url = "https://download/file.zip"
+    # 1ª chamada ao ticket devolve JSON com UrlDownload
+    requests_mock.get(_TICKET_URL, json={"result": json.dumps({"UrlDownload": dl_url})})
+    requests_mock.get(dl_url, content=b"ZIP2", headers={"Content-Type": "application/zip"})
+    monkeypatch.setattr("time.sleep", lambda *_: None)
+    out = await_report_ticket_result("tok", "XYZ", attempts=1)
+    assert out == b"ZIP2"
+
+
+def test_process_zip_to_dfs():
+    buf = io.BytesIO()
+    with zipfile.ZipFile(buf, "w") as zf:
+        df = pd.DataFrame({"x": [1]})
+        zf.writestr("data.csv", df.to_csv(index=False))
+    dfs = process_zip_to_dfs(buf.getvalue())
+    assert dfs["data.csv"].iloc[0, 0] == 1
+    
+    
+def test_process_zip_latin1_csv():
+    import io, zipfile, pandas as pd
+    from luxorasap.btgapi.reports import process_zip_to_dfs
+
+    df_src = pd.DataFrame({"nome": ["ação", "æøå"]})
+    csv_latin1 = df_src.to_csv(index=False, encoding="latin1")
+
+    buf = io.BytesIO()
+    with zipfile.ZipFile(buf, "w") as zf:
+        zf.writestr("dados.csv", csv_latin1)
+
+    dfs = process_zip_to_dfs(buf.getvalue())
+    assert dfs["dados.csv"].equals(df_src)
+    
+
+def test_request_fundflow_report_success(requests_mock):
+    """POST devolvendo ticket e corpo montado corretamente."""
+    start, end = dt.date(2025, 5, 4), dt.date(2025, 6, 4)
+    fund_name   = "luxor lipizzaner fia"
+    expected_id = "TCK-123"
+
+    # valida corpo enviado
+    def _match(request):
+        payload = request.json()
+        c = payload["contract"]
+        assert c["startDate"] == f"{start}T00:00:00Z"
+        assert c["endDate"]   == f"{end}T00:00:00Z"
+        assert c["fundName"]  == fund_name
+        assert c["dateType"]  == "LIQUIDACAO"
+        return True
+
+    requests_mock.post(
+        _FUNDFLOW_URL,
+        additional_matcher=_match,
+        json={"ticket": expected_id},
+        status_code=200,
+    )
+
+    ticket = request_fundflow_report(
+        _TOKEN, start, end, fund_name=fund_name
+    )
+    assert ticket == expected_id
+    
+
+def test_check_report_ticket_returns_dataframe(requests_mock):
+    """GET devolvendo JSON-list deve virar DataFrame."""
+    ticket_id   = "TCK-LIST"
+    sample_rows = [
+        {"customerName": "A", "valueTotal": 1_000},
+        {"customerName": "B", "valueTotal": 2_000},
+    ]
+
+    # qualquer query param é aceito; podemos validar se quiser via matcher
+    requests_mock.get(
+        _TICKET_URL,
+        json={"result": sample_rows},
+        status_code=200,
+    )
+
+    df = check_report_ticket(_TOKEN, ticket_id)
+    assert isinstance(df, pd.DataFrame)
+    assert len(df) == 2
+    assert set(df.columns) == {"customerName", "valueTotal"}
+    
+    
+def test_await_report_ticket_polls_until_ready(requests_mock, monkeypatch):
+    """Primeira chamada ‘Processando’ → segunda retorna lista."""
+    ticket_id   = "TCK-POLL"
+    ready_rows  = [{"x": 1}]
+
+    # sequencia de respostas: processamento → pronto
+    requests_mock.get(
+        _TICKET_URL,
+        [
+            {"json": {"result": "Processando"}, "status_code": 200},
+            {"json": {"result": ready_rows},    "status_code": 200},
+        ],
+    )
+
+    # evita atraso real de sleep
+    monkeypatch.setattr("luxorasap.btgapi.reports.time.sleep", lambda *_: None)
+
+    df = await_report_ticket_result(_TOKEN, ticket_id, attempts=2, interval=0)
+    assert isinstance(df, pd.DataFrame)
+    assert df.iloc[0, 0] == 1

luxorasap-0.1.2/tests/test_btgapi_reports.py

@@ -1,62 +0,0 @@
-import datetime as dt, io, json, zipfile
-import pandas as pd
-import pytest
-
-from luxorasap.btgapi.reports import (
-    request_portfolio,
-    await_report_ticket_result,
-    process_zip_to_dfs,
-)
-
-_TICKET_URL = "https://funds.btgpactual.com/reports/Ticket"
-_POST_URL   = "https://funds.btgpactual.com/reports/Portfolio"
-
-def test_request_portfolio_returns_ticket(requests_mock):
-    requests_mock.post(_POST_URL, json={"ticket": "ABC"})
-    tk = request_portfolio("tok", "FUND", dt.date(2025,1,1), dt.date(2025,1,31))
-    assert tk == "ABC"
-
-def test_await_ticket_inline_zip(requests_mock, monkeypatch):
-    # 1ª chamada: ainda processando   2ª: devolve ZIP binário
-    # Explicando o que requests_mock faz:
-    # 
-    requests_mock.get(_TICKET_URL, [
-        {"json": {"result": "Processando"}},
-        {"content": b"ZIP!"},
-    ])
-    monkeypatch.setattr("time.sleep", lambda *_: None)
-    out = await_report_ticket_result("tok", "ABC", attempts=2, interval=0)
-    assert out == b"ZIP!"
-
-def test_await_ticket_via_urldownload(requests_mock, monkeypatch):
-    dl_url = "https://download/file.zip"
-    # 1ª chamada ao ticket devolve JSON com UrlDownload
-    requests_mock.get(_TICKET_URL, json={"result": json.dumps({"UrlDownload": dl_url})})
-    requests_mock.get(dl_url, content=b"ZIP2", headers={"Content-Type": "application/zip"})
-    monkeypatch.setattr("time.sleep", lambda *_: None)
-    out = await_report_ticket_result("tok", "XYZ", attempts=1)
-    assert out == b"ZIP2"
-
-
-def test_process_zip_to_dfs():
-    buf = io.BytesIO()
-    with zipfile.ZipFile(buf, "w") as zf:
-        df = pd.DataFrame({"x": [1]})
-        zf.writestr("data.csv", df.to_csv(index=False))
-    dfs = process_zip_to_dfs(buf.getvalue())
-    assert dfs["data.csv"].iloc[0, 0] == 1
-    
-    
-def test_process_zip_latin1_csv():
-    import io, zipfile, pandas as pd
-    from luxorasap.btgapi.reports import process_zip_to_dfs
-
-    df_src = pd.DataFrame({"nome": ["ação", "æøå"]})
-    csv_latin1 = df_src.to_csv(index=False, encoding="latin1")
-
-    buf = io.BytesIO()
-    with zipfile.ZipFile(buf, "w") as zf:
-        zf.writestr("dados.csv", csv_latin1)
-
-    dfs = process_zip_to_dfs(buf.getvalue())
-    assert dfs["dados.csv"].equals(df_src)