Refactor process_all_rows function to streamline row processing and improve readability
This commit is contained in:
27
main.py
27
main.py
@@ -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.")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user