DZIALA

main.py

@@ -1,4 +1,4 @@
-from gsheet_api import list_sheets, get_sheet_data, append_unique_row
+from gsheet_api import GSheetAPI
 from mayo import MayoSession
 from dotenv import load_dotenv
 import os
@@ -6,48 +6,29 @@ import re
 
 load_dotenv()
 
-# --- konfiguracja ---pip list
+# --- konfiguracja ---
 DOC_NAME = os.getenv("DOC_NAME")
 MAYO_URL = os.getenv("MAYO_URL")
 LOGIN = os.getenv("MAYO_LOGIN")
 PASSWORD = os.getenv("MAYO_PASSWORD")
 RESULT_DOC = "gitary 2025"
 
-
-
 def normalize(text):
     if not text:
         return ""
-    # usuwa spacje, niełamliwe spacje, nowe linie itp.
     return re.sub(r"\s+", "", text)
 
 def get_finish_type(row_data):
-    """
-    Analizuje dane z listy wiersza, aby określić rodzaj wykończenia.
-
-    Args:
-        row_data: Lista zawierająca dane zamówienia, gdzie:
-                  - row_data[4] to 'color_top'
-                  - row_data[5] to 'color_body'
-
-    Returns:
-        Ciąg znaków: "GLOSS", "SATIN", "MIX", lub "MAT" na podstawie
-        ostatnich liter w kodach kolorów. Zwraca None, jeśli żaden
-        warunek nie jest spełniony.
-    """
     try:
         color_top = row_data["color_top"].strip()
         color_body = row_data["color_body"].strip()
 
-        # Sprawdzenie, czy dane są dostępne
         if not color_top or not color_body:
             return None
 
-        # Pobranie ostatniej litery z kodów kolorów
         top_last_char = color_top[-1]
         body_last_char = color_body[-1]
 
-        # Warunki logiczne
         if top_last_char == 'G' and body_last_char == 'G':
             return "GLOSS"
         elif top_last_char == 'S' and body_last_char == 'S':
@@ -56,29 +37,42 @@ def get_finish_type(row_data):
             return "MIX"
         elif top_last_char in ('M', 'R') and body_last_char in ('M', 'R'):
             return "MAT"
-    except:
-        return None # Zwraca None, jeśli żaden z powyższych warunków nie zostanie spełniony
-
-    
+    except (KeyError, AttributeError):
+        return None
+    return None
 
 def main():
-    print("📄 Pobieram listę arkuszy...")
-    sheets = list_sheets(DOC_NAME)
-    for i, name in enumerate(sheets):
-        print(f"{i+1}. {name}")
+    # Inicjalizuj API raz na początku
+    gsheet_api = GSheetAPI()
+
+    print("📄 Pobieram listę arkuszy...")
+    try:
+        sheets = gsheet_api.list_sheets(DOC_NAME)
+        for i, name in enumerate(sheets):
+            print(f"{i+1}. {name}")
+    except Exception as e:
+        print(f"❌ Błąd podczas pobierania listy arkuszy: {e}")
+        return
 
-    # wybór arkusza (na razie manualny)
     sheet_name = input("\nWybierz arkusz do przetworzenia: ")
 
     print(f"📋 Pobieram dane z arkusza: {sheet_name}")
-    rows = get_sheet_data(DOC_NAME, sheet_name)
+    try:
+        rows = gsheet_api.get_sheet_data(DOC_NAME, sheet_name)
+    except Exception as e:
+        print(f"❌ Błąd podczas pobierania danych z arkusza: {e}")
+        return
 
     mayo = MayoSession(MAYO_URL, LOGIN, PASSWORD)
     mayo.login()
 
+    rows_to_process = []
     counter = 1
     # Zakładamy: kolumna B = link, kolumna C = nr zam.
-    for row in rows[1:]:  # pomijamy nagłówek
+    for row in rows[1:]:
+        if len(row) < 3:
+            continue # Pomiń wiersze, które nie mają wystarczającej liczby kolumn
+
         link = row[1]
         nr_zam = row[2]
 
@@ -86,37 +80,46 @@ def main():
             continue
 
         print(f"\n🔗 Sprawdzam: {link}")
-        info = mayo.get_order_info(link)
-        order_number = info["order_number"]
-        model = info["model"]
+        try:
+            info = mayo.get_order_info(link)
+            order_number = info["order_number"]
+            model = info["model"]
 
-        print(f"Nr z arkusza: {nr_zam}")
-        print(f"Nr ze strony: {info['order_number']}")
-        print(f"Model: {info['model']}")
+            print(f"Nr z arkusza: {nr_zam}")
+            print(f"Nr ze strony: {order_number}")
+            print(f"Model: {model}")
 
-        if normalize(info["order_number"]) == normalize(nr_zam):
-            print("✅ Numer się zgadza")
-        else:
-            print("⚠️ Numer NIE pasuje!")
+            if normalize(order_number) == normalize(nr_zam):
+                print("✅ Numer się zgadza")
+            else:
+                print("⚠️ Numer NIE pasuje!")
 
-        row_data = [
-            counter,
-            link,
-            nr_zam,
-            model,
-            get_finish_type(info),
-            info["color_top"],
-            info["color_body"],
-            info["color_neck"],
-            info["color_head"],
-            info["finish"],
-        ]
-        
-        print(f"top: {info['color_top']}, body: {info['color_body']}, neck: {info['color_neck']}, head: {info['color_head']}, finish: {info["finish"]}")
-        print(f"WYKONCZENIE: {get_finish_type(info)}")
-        append_unique_row(RESULT_DOC, sheet_name, row_data)
+            row_data = [
+                counter,
+                link,
+                nr_zam,
+                model,
+                get_finish_type(info),
+                info.get("color_top"),
+                info.get("color_body"),
+                info.get("color_neck"),
+                info.get("color_head"),
+                info.get("finish"),
+            ]
+            
+            rows_to_process.append(row_data)
+            counter += 1
 
-        counter += 1
+        except Exception as e:
+            print(f"❌ Błąd podczas przetwarzania linku {link}: {e}")
+
+    # Po zakończeniu pętli, dodaj wszystkie zebrane wiersze za jednym razem
+    if rows_to_process:
+        print(f"\n\n--- Podsumowanie ---")
+        print(f"Zebrano {len(rows_to_process)} wierszy do przetworzenia.")
+        gsheet_api.batch_append_unique_rows(RESULT_DOC, sheet_name, rows_to_process)
+    else:
+        print("\nNie zebrano żadnych danych do przetworzenia.")
 
 if __name__ == "__main__":
-    main()
+    main()