src/maximum_roster_strategy/data_loader.py

import os
import pandas as pd


MRS_SHEET_ID = os.environ.get("MRS_SHEET_ID")


def get_google_sheet_data() -> pd.DataFrame:
    return get_sheet_data(0)


def get_sheet_data(sheet_id: int = 0):
    sheet_url = f"https://docs.google.com/spreadsheet/ccc?key={MRS_SHEET_ID}&output=csv&gid={sheet_id}"
    df = pd.read_csv(sheet_url)
    return df


def get_timeslot_labels() -> pd.DataFrame:
    return get_sheet_data(1875906423)