triangle-api 0.1.4__tar.gz → 0.1.5__tar.gz

{triangle_api-0.1.4 → triangle_api-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triangle-api
-Version: 0.1.4
+Version: 0.1.5
 Summary: Unofficial Python wrapper for Canadian Tire Triangle Mastercard API
 Home-page: https://github.com/diogobas/triangle-api
 Author: Diogo Bastos

{triangle_api-0.1.4 → triangle_api-0.1.5}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="triangle-api", # Nome que aparecerá no PyPI
-    version="0.1.4",
+    version="0.1.5",
     author="Diogo Bastos",
     author_email="seu-email@exemplo.com", # Opcional
     description="Unofficial Python wrapper for Canadian Tire Triangle Mastercard API",

triangle_api-0.1.5/triangle_api/__init__.py

@@ -0,0 +1,4 @@
+from .client import TriangleClient
+
+__version__ = "0.1.5"
+__all__ = ["TriangleClient"]

triangle_api-0.1.5/triangle_api/client.py

@@ -0,0 +1,173 @@
+from playwright.sync_api import sync_playwright
+import os
+import time
+import hashlib
+import json
+from datetime import datetime
+
+class TriangleClient:
+    def __init__(self, username=None, password=None, session_path="auth_state.json"):
+        self.username = username
+        self.password = password
+        self.session_path = session_path
+        self.account_data = None
+
+    def _get_csrf(self, context):
+        """Auxiliar para extrair o token CSRF dos cookies."""
+        cookies = context.cookies()
+        return next((c['value'] for c in cookies if c['name'] == 'csrftoken'), "")
+
+    def _fetch_account_data(self, page, csrf_token):
+        """Faz a chamada manual do retrieveAccount com CSRF."""
+        js_get_account = """
+        async (csrf) => {
+            const r = await fetch('/dash/v1/account/retrieveAccount', {
+                method: 'POST',
+                headers: { 
+                    'Content-Type': 'application/json',
+                    'CSRF-Token': csrf,
+                    'csrfToken': csrf
+                },
+                body: JSON.stringify({})
+            });
+            return r.ok ? await r.json() : null;
+        }
+        """
+        return page.evaluate(js_get_account, csrf_token)
+
+    def login(self, headless=True):
+        with sync_playwright() as p:
+            browser = p.chromium.launch(headless=headless)
+            has_session = os.path.exists(self.session_path)
+            storage = self.session_path if has_session else None
+            
+            context = browser.new_context(
+                viewport={'width': 1280, 'height': 800},
+                storage_state=storage,
+                user_agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+            )
+            page = context.new_page()
+
+            print("[API] Acessando o portal...")
+            page.goto("https://www.ctfs.com/content/dash/en/private/Summary.html", timeout=60000)
+
+            # Se cair no login, faz o processo de autenticação
+            if "login" in page.url.lower():
+                if not self.username or not self.password:
+                    browser.close()
+                    return "NEEDS_CREDENTIALS"
+
+                print("[API] Sessão expirada. Autenticando...")
+                try:
+                    # Lida com cookies
+                    if page.is_visible("#onetrust-accept-btn-handler"):
+                        page.click("#onetrust-accept-btn-handler", timeout=5000)
+
+                    page.wait_for_selector("#username", timeout=30000)
+                    page.fill("#username", self.username)
+                    page.fill("#password", self.password)
+                    page.focus("#password")
+                    page.keyboard.press("Enter")
+
+                    # Espera sair do login
+                    page.wait_for_function("() => !window.location.href.includes('login')", timeout=45000)
+
+                    if "challenge" in page.url or "otp" in page.url:
+                        sms_code = input("\n[MFA] Digite o código SMS: ")
+                        page.get_by_role("textbox").first.fill(sms_code)
+                        page.keyboard.press("Enter")
+                        page.wait_for_url("**/Summary.html", timeout=45000)
+                except Exception as e:
+                    print(f"[ERRO] Falha no login: {e}")
+                    browser.close()
+                    return None
+
+            # CAPTURA DE DADOS (Saldo)
+            print("[API] Capturando dados financeiros...")
+            try:
+                page.wait_for_load_state("domcontentloaded")
+                csrf = self._get_csrf(context)
+                self.account_data = self._fetch_account_data(page, csrf)
+                
+                if self.account_data:
+                    context.storage_state(path=self.session_path)
+                    print("[API] Dados capturados com sucesso.")
+            except Exception as e:
+                print(f"[AVISO] Erro na captura de saldo: {e}")
+
+            browser.close()
+            return self.account_data
+
+    def get_transactions(self, start_date=None, headless=True):
+        today = datetime.today()
+        if not start_date:
+            start_date = today.replace(day=1).strftime('%Y-%m-%d')
+        
+        with sync_playwright() as p:
+            browser = p.chromium.launch(headless=headless)
+            context = browser.new_context(storage_state=self.session_path)
+            page = context.new_page()
+
+            print("[API] Sincronizando para transações...")
+            page.goto("https://www.ctfs.com/content/dash/en/private/Summary.html", wait_until="domcontentloaded")
+            
+            csrf = self._get_csrf(context)
+            
+            # Se o account_data sumiu da memória, pegamos de novo agora que temos o CSRF
+            if not self.account_data:
+                self.account_data = self._fetch_account_data(page, csrf)
+
+            if not self.account_data:
+                browser.close()
+                return []
+
+            js_tx_script = """
+            async (params) => {
+                const call = async (cat, stmt = null) => {
+                    const body = { category: cat, transientReference: params.transient };
+                    if (stmt) body.statementDate = stmt;
+                    const r = await fetch('/dash/v1/account/retrieveTransactions', {
+                        method: 'POST',
+                        headers: { 'Content-Type': 'application/json', 'CSRF-Token': params.csrf, 'csrfToken': params.csrf },
+                        body: JSON.stringify(body)
+                    });
+                    return r.ok ? await r.json() : { transactions: [] };
+                };
+                const auth = await call('AUTHORIZED');
+                const post = await call('POSTED');
+                const stmt = params.lastStmt ? await call('STATEMENTED', params.lastStmt) : { transactions: [] };
+                return [...(auth.transactions || []), ...(post.transactions || []), ...(stmt.transactions || [])];
+            }
+            """
+            
+            raw_txs = page.evaluate(js_tx_script, {
+                "transient": self.account_data.get("transientReference"),
+                "lastStmt": self.account_data.get("lastStatementDate"),
+                "csrf": csrf
+            })
+            browser.close()
+
+            # Filtro e Deduplicação
+            unique_txs = {}
+            for t in raw_txs:
+                d = t.get("tranDate", "")
+                if d >= start_date:
+                    key = f"{d}|{t.get('merchant')}|{t.get('amount')}"
+                    unique_txs[key] = t
+            return list(unique_txs.values())
+
+    def get_normalized_transactions(self, start_date=None, headless=True):
+        txs = self.get_transactions(start_date=start_date, headless=headless)
+        normalized = []
+        for t in txs:
+            date, payee, amount = t.get("tranDate"), t.get("merchant", "Unknown"), float(t.get("amount", 0.0))
+            ext_id = hashlib.md5(f"{date}|{payee}|{amount}".encode()).hexdigest()
+            normalized.append({
+                "date": date, "payee": payee, "amount": amount, "currency": "cad",
+                "external_id": ext_id, "status": "cleared" if t.get("category") in ["POSTED", "STATEMENTED"] else "uncleared",
+                "notes": f"Type: {t.get('type')} | Imported via Triangle API"
+            })
+        return sorted(normalized, key=lambda x: x["date"], reverse=True)
+
+    def get_balance(self):
+        return self.account_data.get("currentBalanceAmt", 0.0) if self.account_data else "N/A"

{triangle_api-0.1.4 → triangle_api-0.1.5}/triangle_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triangle-api
-Version: 0.1.4
+Version: 0.1.5
 Summary: Unofficial Python wrapper for Canadian Tire Triangle Mastercard API
 Home-page: https://github.com/diogobas/triangle-api
 Author: Diogo Bastos

triangle_api-0.1.4/triangle_api/__init__.py

@@ -1,4 +0,0 @@
-from .client import TriangleClient
-
-__version__ = "0.1.4"
-__all__ = ["TriangleClient"]

triangle_api-0.1.4/triangle_api/client.py

@@ -1,232 +0,0 @@
-from playwright.sync_api import sync_playwright
-import os
-import time
-import hashlib
-from datetime import datetime
-
-class TriangleClient:
-    def __init__(self, username=None, password=None, session_path="auth_state.json"):
-        self.username = username
-        self.password = password
-        self.session_path = session_path
-        self.account_data = None
-
-    def login(self, headless=True):
-        with sync_playwright() as p:
-            # 1. Configuração do Navegador
-            browser = p.chromium.launch(headless=headless)
-            
-            # Verifica se existe sessão salva para tentar restaurar
-            has_session = os.path.exists(self.session_path)
-            storage = self.session_path if has_session else None
-            
-            context = browser.new_context(
-                viewport={'width': 1280, 'height': 800},
-                storage_state=storage,
-                user_agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-            )
-            page = context.new_page()
-
-            # 2. Definição da URL inicial
-            if has_session:
-                print("[API] Tentando restaurar sessão anterior...")
-                target_url = "https://www.ctfs.com/content/dash/en/private/Summary.html"
-            else:
-                print("[API] Iniciando novo login...")
-                target_url = "https://www.ctfs.com/content/dashpublic/en/login.html"
-
-            page.goto(target_url, wait_until="load", timeout=60000)
-
-            # 3. Lidar com Banner de Cookies (OneTrust)
-            # Esse banner bloqueia a interação com os campos de texto se não for fechado
-            try:
-                cookie_btn = page.locator("#onetrust-accept-btn-handler, text='Accept All'")
-                if cookie_btn.is_visible(timeout=8000):
-                    cookie_btn.click()
-                    print("[API] Banner de cookies aceito.")
-                    time.sleep(1) # Pausa para animação de fechamento
-            except:
-                pass
-
-            # 4. Fluxo de Autenticação (Caso a sessão tenha expirado ou não exista)
-            if "login" in page.url.lower() or not has_session:
-                if not self.username or not self.password:
-                    print("[API] Credenciais necessárias para continuar.")
-                    browser.close()
-                    return "NEEDS_CREDENTIALS"
-
-                print("[API] Tela de login detectada. Preenchendo credenciais...")
-                try:
-                    page.wait_for_selector("#username", timeout=30000)
-                    page.fill("#username", self.username)
-                    page.fill("#password", self.password)
-                    
-                    # Submissão via Teclado (Enter) é mais resiliente contra widgets de chat
-                    page.focus("#password")
-                    page.keyboard.press("Enter")
-                    
-                    # Clique de segurança via JavaScript caso o Enter falhe
-                    page.evaluate("document.querySelector('#signin-form')?.click()")
-
-                    # Espera sair da tela de login (mudar URL ou detectar MFA)
-                    print("[API] Formulário enviado. Aguardando resposta do banco...")
-                    page.wait_for_function("() => !window.location.href.includes('login')", timeout=45000)
-
-                    # --- Verificação de MFA (SMS/OTP) ---
-                    if "challenge" in page.url or "otp" in page.url or page.is_visible("input[aria-describedby*='otp']"):
-                        print("\n[MFA] Verificação de segurança detectada.")
-                        sms_code = input("Digite o código SMS recebido no celular: ")
-                        page.get_by_role("textbox").first.fill(sms_code)
-                        page.keyboard.press("Enter")
-                        page.wait_for_url("**/Summary.html", timeout=60000)
-
-                    print("[API] Login realizado com sucesso!")
-
-                except Exception as e:
-                    print(f"[ERRO] Falha no processo de login: {e}")
-                    page.screenshot(path="debug_login_error.png")
-                    browser.close()
-                    return None
-
-            # 5. CAPTURA FINAL DE DADOS (Saldo e transientReference)
-            print("[API] Capturando dados financeiros (retrieveAccount)...")
-            try:
-                # Esperamos a página carregar o básico
-                page.wait_for_load_state("domcontentloaded")
-                
-                # Em vez de 'esperar' uma resposta, nós 'pedimos' os dados
-                # Isso evita o erro de "Protocol error"
-                js_get_account = """
-                async () => {
-                    const r = await fetch('/dash/v1/account/retrieveAccount', {
-                        method: 'POST',
-                        headers: { 'Content-Type': 'application/json' },
-                        body: JSON.stringify({})
-                    });
-                    return r.ok ? await r.json() : null;
-                }
-                """
-                
-                # Executa o fetch dentro do contexto do navegador
-                self.account_data = page.evaluate(js_get_account)
-                
-                if self.account_data:
-                    # Agora que temos os dados, salvamos a sessão
-                    context.storage_state(path=self.session_path)
-                    print("[API] Dados financeiros capturados com sucesso.")
-                else:
-                    print("[AVISO] O servidor retornou vazio para o saldo.")
-                
-            except Exception as e:
-                print(f"[ERRO] Falha ao capturar saldo via JS: {e}")
-
-            browser.close()
-            return self.account_data
-
-    def get_normalized_transactions(self, start_date=None, headless=True):
-        """
-        Versão ultra-robusta para evitar erros de ordenação (NoneType).
-        """
-        # 1. Busca as transações brutas
-        raw_txs = self.get_transactions(start_date=start_date, headless=headless)
-        
-        normalized = []
-        seen_ids = set()
-
-        for tx in raw_txs:
-            # Pegamos a data. Se não existir, ignoramos o item.
-            t_date = tx.get("tranDate")
-            if not t_date:
-                continue
-
-            payee = tx.get("merchant", "Unknown Merchant")
-            amount = float(tx.get("amount", 0.0))
-            
-            # Gerar ID Único para deduplicação
-            identifier = f"{t_date}|{payee}|{amount}"
-            if identifier in seen_ids:
-                continue
-            seen_ids.add(identifier)
-
-            # MD5 para o Lunch Money
-            ext_id = hashlib.md5(identifier.encode("utf-8")).hexdigest()
-
-            # Certifique-se de que o dicionário normalized tenha a chave 'notes'
-            normalized.append({
-                "date": t_date,
-                "payee": payee,
-                "amount": amount,
-                "currency": "cad",
-                "external_id": ext_id,
-                "type": tx.get("type", "PURCHASE"),
-                "status": "cleared" if tx.get("category") in ["POSTED", "STATEMENTED"] else "uncleared",
-                "notes": f"Type: {tx.get('type')} | Imported via Triangle API"
-            })
-
-        # Ordenar garantindo que não existam Nones na chave de ordenação
-        return sorted(normalized, key=lambda x: x.get("date", ""), reverse=True)
-
-    def get_transactions(self, start_date=None, headless=True):
-        today = datetime.today()
-        if not start_date:
-            start_date = today.replace(day=1).strftime('%Y-%m-%d')
-        end_date = today.strftime('%Y-%m-%d')
-
-        print(f"[API] Buscando transações de {start_date} até {end_date}...")
-
-        with sync_playwright() as p:
-            browser = p.chromium.launch(headless=headless)
-            storage = self.session_path if os.path.exists(self.session_path) else None
-            context = browser.new_context(storage_state=storage)
-            page = context.new_page()
-
-            try:
-                print("[API] Sincronizando sessão para histórico...")
-                # Navega e pesca o retrieveAccount da janela atual
-                with page.expect_response("**/retrieveAccount", timeout=30000) as resp_acc:
-                    page.goto("https://www.ctfs.com/content/dash/en/private/Summary.html", wait_until="domcontentloaded")
-                
-                acc_info = resp_acc.value.json()
-                transient = acc_info.get("transientReference")
-                acc_id = acc_info.get("accountId")
-                last_stmt = acc_info.get("lastStatementDate")
-                
-                cookies = context.cookies()
-                csrf = next((c['value'] for c in cookies if c['name'] == 'csrftoken'), "")
-
-                # Script de busca tripla
-                js_script = """
-                async (params) => {
-                    const call = async (cat, stmt = null) => {
-                        const body = { category: cat, transientReference: params.transient, accountId: params.accId };
-                        if (stmt) body.statementDate = stmt;
-                        const r = await fetch('/dash/v1/account/retrieveTransactions', {
-                            method: 'POST',
-                            headers: { 'Content-Type': 'application/json', 'CSRF-Token': params.csrf, 'csrfToken': params.csrf },
-                            body: JSON.stringify(body)
-                        });
-                        return r.ok ? await r.json() : { transactions: [] };
-                    };
-                    const auth = await call('AUTHORIZED');
-                    const post = await call('POSTED');
-                    const stmt = params.lastStmt ? await call('STATEMENTED', params.lastStmt) : { transactions: [] };
-                    return [...(auth.transactions || []), ...(post.transactions || []), ...(stmt.transactions || [])];
-                }
-                """
-                
-                raw_txs = page.evaluate(js_script, {
-                    "transient": transient, "accId": acc_id, "lastStmt": last_stmt, "csrf": csrf
-                })
-                
-                # Filtro de data bruto
-                filtered = [t for t in raw_txs if t.get("tranDate", "") >= start_date]
-                browser.close()
-                return filtered
-
-            except Exception as e:
-                print(f"[ERRO] Falha na captura: {e}")
-                browser.close()
-                return []
-
-    def get_balance(self):
-        return self.account_data.get("currentBalanceAmt", 0.0) if self.account_data else "N/A"