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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triangle-api
-Version: 0.1.5
+Version: 0.1.6
 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.5 → triangle_api-0.1.6}/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.5",
+    version="0.1.6",
     author="Diogo Bastos",
     author_email="seu-email@exemplo.com", # Opcional
     description="Unofficial Python wrapper for Canadian Tire Triangle Mastercard API",

triangle_api-0.1.6/triangle_api/__init__.py

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

{triangle_api-0.1.5 → triangle_api-0.1.6}/triangle_api/client.py

@@ -13,12 +13,10 @@ class TriangleClient:
         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', {
@@ -33,13 +31,17 @@ class TriangleClient:
             return r.ok ? await r.json() : null;
         }
         """
-        return page.evaluate(js_get_account, csrf_token)
+        try:
+            return page.evaluate(js_get_account, csrf_token)
+        except:
+            return None
 
     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
+            
+            # SEGURANÇA: Só carrega se o arquivo existir
+            storage = self.session_path if os.path.exists(self.session_path) else None
             
             context = browser.new_context(
                 viewport={'width': 1280, 'height': 800},
@@ -51,15 +53,13 @@ class TriangleClient:
             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...")
+                print("[API] Sessão expirada ou inexistente. Autenticando...")
                 try:
-                    # Lida com cookies
                     if page.is_visible("#onetrust-accept-btn-handler"):
                         page.click("#onetrust-accept-btn-handler", timeout=5000)
 
@@ -69,7 +69,6 @@ class TriangleClient:
                     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:
@@ -82,7 +81,6 @@ class TriangleClient:
                     browser.close()
                     return None
 
-            # CAPTURA DE DADOS (Saldo)
             print("[API] Capturando dados financeiros...")
             try:
                 page.wait_for_load_state("domcontentloaded")
@@ -91,7 +89,7 @@ class TriangleClient:
                 
                 if self.account_data:
                     context.storage_state(path=self.session_path)
-                    print("[API] Dados capturados com sucesso.")
+                    print("[API] Dados capturados e sessão salva.")
             except Exception as e:
                 print(f"[AVISO] Erro na captura de saldo: {e}")
 
@@ -99,13 +97,15 @@ class TriangleClient:
             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')
+            start_date = datetime.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)
+            
+            # SEGURANÇA: Só carrega se o arquivo existir
+            storage = self.session_path if os.path.exists(self.session_path) else None
+            context = browser.new_context(storage_state=storage)
             page = context.new_page()
 
             print("[API] Sincronizando para transações...")
@@ -113,11 +113,11 @@ class TriangleClient:
             
             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:
+                print("[ERRO] Não foi possível obter dados da conta para transações.")
                 browser.close()
                 return []
 
@@ -147,7 +147,6 @@ class TriangleClient:
             })
             browser.close()
 
-            # Filtro e Deduplicação
             unique_txs = {}
             for t in raw_txs:
                 d = t.get("tranDate", "")
@@ -167,7 +166,7 @@ class TriangleClient:
                 "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)
+        return sorted(normalized, key=lambda x: x.get("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.5 → triangle_api-0.1.6}/triangle_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triangle-api
-Version: 0.1.5
+Version: 0.1.6
 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.5/triangle_api/__init__.py

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