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)