From 2d920f025713e98ebf9c61e5b34c216353d92a1f Mon Sep 17 00:00:00 2001 From: bartool Date: Wed, 29 Oct 2025 21:32:04 +0100 Subject: [PATCH] Refactor get_sheet_data function to improve data fetching and error handling --- main.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 69256a0..cd72d22 100644 --- a/main.py +++ b/main.py @@ -126,6 +126,15 @@ def select_sheet(gsheet_api): print(f"❌ Błąd podczas pobierania listy arkuszy: {e}") return None +def get_sheet_data(gsheet_api, sheet_name): + """Fetches all data from a given sheet.""" + print(f"📋 Pobieram dane z arkusza: {sheet_name}") + try: + return gsheet_api.get_sheet_data(DOC_NAME, sheet_name) + except Exception as e: + print(f"❌ Błąd podczas pobierania danych z arkusza: {e}") + return None + def main(): # Inicjalizuj API raz na początku gsheet_api = GSheetAPI() @@ -134,11 +143,8 @@ def main(): if not sheet_name: return - print(f"📋 Pobieram dane z arkusza: {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}") + rows = get_sheet_data(gsheet_api, sheet_name) + if rows is None: return mayo = MayoSession(MAYO_URL, LOGIN, PASSWORD)