importer.py

from csv import reader
from distutils.util import strtobool
from os import getenv

import gspread
from google.oauth2.service_account import Credentials

# Input vars
csv_path = getenv("INPUT_CSV_PATH")
spreadsheet_id = getenv("INPUT_SPREADSHEET_ID")
worksheet_id = int(getenv("INPUT_WORKSHEET"))
append_content = strtobool(getenv("INPUT_APPEND_CONTENT", "False"))

service_account_info = {
    "token_uri": "https://oauth2.googleapis.com/token",
    "client_email": getenv("INPUT_GOOGLE_SERVICE_ACCOUNT_EMAIL"),
    "private_key": getenv("INPUT_GOOGLE_SERVICE_ACCOUNT_PRIVATE_KEY").replace(
        "\\n", "\n"
    ),
}
scopes = [
    "https://spreadsheets.google.com/feeds",
]


def next_available_row(worksheet):
    """
    Return the next available row in the first column
    gspread doesn't have a method to return the last row used
    """
    str_list = list(filter(None, worksheet.col_values(1)))
    return str(len(str_list) + 1)


# Read csv file as a list of lists
with open(csv_path, "r") as read_obj:
    # pass the file object to reader() to get the reader object
    csv_reader = reader(read_obj)
    # Pass reader object to list() to get a list of lists
    list_of_rows = list(csv_reader)

creds = Credentials.from_service_account_info(
    service_account_info, scopes=scopes
)

client = gspread.authorize(creds)

spreadsheet = client.open_by_key(spreadsheet_id)
ws = spreadsheet.get_worksheet(worksheet_id)

if append_content:
    start_from = f"A{next_available_row(ws)}"
else:
    ws.clear()
    start_from = "A1"

ws.update(start_from, list_of_rows)