sheet_utils.py

import os
from googleapiclient.discovery import build
from utils import load_and_refresh_credentials

SCOPES = ["https://www.googleapis.com/auth/spreadsheets"]
SPREADSHEET_ID = os.getenv("SPREADSHEET_ID")
SHEET_NAME = "Sheet1"

creds = load_and_refresh_credentials(scopes=SCOPES)


def append_to_sheet(values: list[str]):
    service = build('sheets', 'v4', credentials=creds)
    sheet = service.spreadsheets()

    body = {
        'values': [values]
    }

    result = sheet.values().append(
        spreadsheetId=SPREADSHEET_ID,
        range=f"{SHEET_NAME}!A2:G",
        valueInputOption="RAW",
        insertDataOption="INSERT_ROWS",
        body=body
    ).execute()
    return result


def update_to_sheet(range_to_update, values: list[str]):
    service = build('sheets', 'v4', credentials=creds)
    sheet = service.spreadsheets()

    body = {
        'values': [values]
    }

    result = sheet.values().update(
        spreadsheetId=SPREADSHEET_ID,
        range=range_to_update,
        valueInputOption="RAW",
        body=body
    ).execute()
    return result