atlan-collect / spreadsheet / data_uploader / data_uploader.py
data_uploader.py
Raw
import json

def insert_data(worksheet, message):
    try:
        message = json.loads(message)

        name = message["name"]
        email = message["email"]
        phone = message["phone"]

        row_data = [name, email, phone]
        num_rows = len(worksheet.get_col(1, include_tailing_empty=False))
        next_row = num_rows

        worksheet.insert_rows(next_row, values=row_data)
        
        return f"The following data was successfully added to your spreadsheet successfully: {name}, {email}, & {phone}", None
    except Exception as err:
        print(err)
        return f"{err}", "error"