Refactor process_all_rows function to streamline row processing and improve readability

This commit is contained in:
2025-10-29 21:32:41 +01:00
parent 2d920f0257
commit 7cdd500cb6

27
main.py
View File

@@ -135,6 +135,18 @@ def get_sheet_data(gsheet_api, sheet_name):
print(f"❌ Błąd podczas pobierania danych z arkusza: {e}") print(f"❌ Błąd podczas pobierania danych z arkusza: {e}")
return None return None
def process_all_rows(rows, mayo):
"""Processes all rows from the sheet."""
rows_to_process = []
counter = 1
# Skip header row by starting from index 1
for row in rows[1:]:
processed_row = process_row(row, mayo, counter)
if processed_row:
rows_to_process.append(processed_row)
counter += 1
return rows_to_process
def main(): def main():
# Inicjalizuj API raz na początku # Inicjalizuj API raz na początku
gsheet_api = GSheetAPI() gsheet_api = GSheetAPI()
@@ -150,20 +162,13 @@ def main():
mayo = MayoSession(MAYO_URL, LOGIN, PASSWORD) mayo = MayoSession(MAYO_URL, LOGIN, PASSWORD)
mayo.login() mayo.login()
rows_to_process = [] processed_rows = process_all_rows(rows, mayo)
counter = 1
# Zakładamy: kolumna B = link, kolumna C = nr zam.
for row in rows[1:]:
processed_row = process_row(row, mayo, counter)
if processed_row:
rows_to_process.append(processed_row)
counter += 1
# Po zakończeniu pętli, dodaj wszystkie zebrane wiersze za jednym razem # Po zakończeniu pętli, dodaj wszystkie zebrane wiersze za jednym razem
if rows_to_process: if processed_rows:
print(f"\n\n--- Podsumowanie ---") print(f"\n\n--- Podsumowanie ---")
print(f"Zebrano {len(rows_to_process)} wierszy do przetworzenia.") print(f"Zebrano {len(processed_rows)} wierszy do przetworzenia.")
gsheet_api.batch_append_unique_rows(RESULT_DOC, sheet_name, rows_to_process) gsheet_api.batch_append_unique_rows(RESULT_DOC, sheet_name, processed_rows)
else: else:
print("\nNie zebrano żadnych danych do przetworzenia.") print("\nNie zebrano żadnych danych do przetworzenia.")