bartool/GoogleSheetBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

DZIALA

Browse Source
This commit is contained in:
bartool
2025-10-16 16:42:08 +02:00
parent b259442c20
commit 3b9f8fa038
3 changed files with 162 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
gsheet_api.py
View File

@@ -6,54 +6,75 @@ SCOPES = [
"https://www.googleapis.com/auth/drive"
]
def connect():
creds = Credentials.from_service_account_file("credentials.json", scopes=SCOPES)
client = gspread.authorize(creds)
return client
class GSheetAPI:
def __init__(self, credentials_file="credentials.json"):
"""Inicjalizuje klienta API przy tworzeniu obiektu."""
creds = Credentials.from_service_account_file(credentials_file, scopes=SCOPES)
self.client = gspread.authorize(creds)
print("✅ Połączono z Google Sheets API.")
def list_sheets(self, doc_name):
"""Zwraca listę arkuszy w danym dokumencie."""
spreadsheet = self.client.open(doc_name)
return [ws.title for ws in spreadsheet.worksheets()]
def list_sheets(doc_name):
client = connect()
spreadsheet = client.open(doc_name)
return [ws.title for ws in spreadsheet.worksheets()]
def get_sheet_data(self, doc_name, sheet_name):
"""Pobiera wszystkie dane z danego arkusza."""
sheet = self.client.open(doc_name).worksheet(sheet_name)
return sheet.get_all_values()
def ensure_worksheet(self, doc_name, sheet_name):
"""
Zwraca worksheet o danej nazwie.
Tworzy nowy, jeśli nie istnieje.
"""
spreadsheet = self.client.open(doc_name)
try:
ws = spreadsheet.worksheet(sheet_name)
except gspread.exceptions.WorksheetNotFound:
print(f"➕ Tworzę nowy arkusz: {sheet_name}")
ws = spreadsheet.add_worksheet(title=sheet_name, rows=100, cols=10)
ws.append_row(["#", "Link", "Nr zamówienia", "Model", "Wykończenie", "Kolor Top", "Kolor Body", "Kolor Neck", "Kolor Head", "Finish"])
return ws
def get_sheet_data(doc_name, sheet_name):
client = connect()
sheet = client.open(doc_name).worksheet(sheet_name)
return sheet.get_all_values()
def batch_append_unique_rows(self, doc_name, sheet_name, rows_data):
"""
Dodaje wiele wierszy za jednym razem, pomijając te,
których nr zamówienia (kolumna 3) już istnieje.
"""
if not rows_data:
print("ℹ️ Brak danych do dodania.")
return
ws = self.ensure_worksheet(doc_name, sheet_name)
def ensure_worksheet(doc_name, sheet_name):
"""
Zwraca worksheet o danej nazwie.
Tworzy nowy, jeśli nie istnieje.
"""
client = connect()
spreadsheet = client.open(doc_name)
try:
ws = spreadsheet.worksheet(sheet_name)
except gspread.exceptions.WorksheetNotFound:
print(f"➕ Tworzę nowy arkusz: {sheet_name}")
ws = spreadsheet.add_worksheet(title=sheet_name, rows=100, cols=10)
ws.append_row(["#", "Link", "Nr zamówienia", "Model"])
return ws
# 1. Pobierz wszystkie istniejące numery zamówień w JEDNYM zapytaniu
print("🔍 Sprawdzam istniejące numery zamówień w arkuszu docelowym...")
# existing_orders = set(ws.col_values(3))
existing_orders = {str(x).strip() for x in ws.col_values(3)}
print(f"Znaleziono {len(existing_orders)} istniejących numerów.\n existing_orders: {existing_orders}")
def append_unique_row(doc_name, sheet_name, row_data):
"""
Dodaje wiersz tylko, jeśli dany nr zamówienia (kolumna 3) jeszcze nie istnieje.
"""
client = connect()
ws = ensure_worksheet(doc_name, sheet_name)
# 2. Filtruj nowe wiersze, aby znaleźć tylko te unikalne
unique_rows_to_add = []
for row in rows_data:
order_number = str(row[2]).strip()
# print(f"order_number: '{order_number}'", end="")
if order_number not in existing_orders:
# print(f" not in existing_order!", end="")
unique_rows_to_add.append(row)
# Dodaj nowo dodany numer do seta, aby uniknąć duplikatów w ramach jednej paczki
existing_orders.add(order_number)
# print(" ")
existing = ws.col_values(3) # kolumna "Nr zamówienia"
order_number = str(row_data[2]).strip()
# 3. Dodaj wszystkie unikalne wiersze w JEDNYM zapytaniu
if unique_rows_to_add:
print(f"📝 Dodaję {len(unique_rows_to_add)} nowych unikalnych wierszy do arkusza {sheet_name}...")
ws.append_rows(unique_rows_to_add, value_input_option="USER_ENTERED") # type: ignore
print("✅ Zakończono dodawanie.")
else:
print("ℹ️ Nie znaleziono żadnych nowych wierszy do dodania.")
if order_number in existing:
print(f"⏭️ Pomijam – nr {order_number} już istnieje w arkuszu {sheet_name}")
return False
ws.append_row(row_data, value_input_option="USER_ENTERED")
print(f"📝 Dodano do {sheet_name}: {row_data}")
return True
skipped_count = len(rows_data) - len(unique_rows_to_add)
if skipped_count > 0:
print(f"⏭️ Pominięto {skipped_count} wierszy, które już istniały w arkuszu.")