opencheck.py

import sys
import gspread
from google.oauth2.service_account import Credentials
from datetime import datetime


def next_empty_row(worksheet, column='A'):
	col_values = worksheet.col_values(ord(column.upper()) - 64)
	return len(col_values) + 1


def main():

	# parse input argumets
	if len(sys.argv) < 4:
		print("Usage: opencheck <jpc/pac> <time> <level/close>")
		return

	location = sys.argv[1].lower()
	time_input = sys.argv[2]
	action = sys.argv[3]

	if location != "jpc" and location != "pac":
		print("Invalid location. jpc or pac")
		return

	elif (location.lower() == "jpc"):
		
		openUrl = "https://docs.google.com/spreadsheets/d/1TUmrKavPb7jHt-xWukLbGb1DK4_EQzyBTXFLWCdQ16o/edit?gid=0#gid=0"
		oxyUrl = "https://docs.google.com/spreadsheets/d/1IIMtKaFydRukSqou1DZOrk2xno04M13hPqR9WGyCHO0/edit?gid=0#gid=0"
		aedUrl = "https://docs.google.com/spreadsheets/d/1pfw8Si6i12zwoxQPhlHrjMTTQQXiQzCzG_1KmDZBERQ/edit?gid=0#gid=0"	
		closeUrl = "https://docs.google.com/spreadsheets/d/1Igv2AEcDCMtU70yLuSbpsMNAOJmb3cvEq1K-hJH6XGQ/edit?gid=0#gid=0"
	
	else :
		
		openUrl = ""
		oxyUrl = ""
		aedUrl = ""
		closeUrl = ""

	now = datetime.now()
	date_str = str(now.date())
	day_of_month = str(now.day)

	if (day_of_month == "1" or day_of_month == "21" or day_of_month == "31"):
		
		day_of_month = day_of_month + "st"

	elif (day_of_month == "2" or day_of_month == "22"):
		
		day_of_month = day_of_month + "nd"

	elif (day_of_month == "3" or day_of_month == "23"):
	
		day_of_month = day_of_month + "rd"

	else:
		day_of_month = day_of_month + "th"


	print(date_str, day_of_month)

	# sheets and drive
	scopes = ["https://www.googleapis.com/auth/spreadsheets",
	"https://www.googleapis.com/auth/drive"]

	# give permissions for the api
	creds = Credentials.from_service_account_file("/home/caden-saltzberg/projects/auto/propane-flow-438700-p4-e6d6922109d3.json", scopes=scopes)

	# create client object
	client = gspread.authorize(creds)
	
	# debug statement
	print("Initialized API")

	if (action.lower() == "close"):

		print("Closing")

		print("Connecting to closing sheet")
		closing_sheet = client.open_by_url(closeUrl).worksheet(day_of_month)

		print("Updating cells")
		closing_sheet.update(range_name='B9', values=[[date_str]])
		closing_sheet.update(range_name='F9', values=[[time_input]])
		closing_sheet.update(range_name='A11', values=[['Complete']])
		closing_sheet.update(range_name='A12:A21', values=[['Yes']] * 10)
		closing_sheet.update(range_name='A22', values=[['Safety Inspection Completed By: Caden S']])

		print("Closing Sheet Complete")

	else:

		print("Opening")
		
		try:
			oxy_level = int(action)
		except:
			print("Oxygen level must be a number.")
			return

		print("Opening")

		print("Connecting to open sheet")
		opening_sheet = client.open_by_url(openUrl).worksheet(day_of_month)

		print("Updating cells")
		opening_sheet.update(range_name='B9', values=[[date_str]])
		opening_sheet.update(range_name='F9', values=[[time_input]])
		opening_sheet.update(range_name='A11', values=[['Complete']])
		opening_sheet.update(range_name='A12:A20', values=[['Yes']] * 9)
		opening_sheet.update(range_name='A21', values=[['Safety Inspection Completed By: Caden S']])

		print("Opening sheet complete")

		print("AED log")

		print("Connecting to AED sheet")
		aed_sheet = client.open_by_url(aedUrl).worksheet('AED Log')

		empty_cell = next_empty_row(aed_sheet, 'A')

		print(empty_cell)

		date_cell = "A"+str(empty_cell)
		time_cell = "B"+str(empty_cell)
		name_cell = "C"+str(empty_cell)
		yes_cells = "D"+str(empty_cell)+":J"+str(empty_cell)

		print(date_cell, time_cell, name_cell, yes_cells)

		print("Updating cells")
		aed_sheet.update(range_name=date_cell, values=[[date_str]])
		aed_sheet.update(range_name=time_cell, values=[[time_input]])
		aed_sheet.update(range_name=name_cell, values=[['CS']])
		aed_sheet.update(range_name=yes_cells, values=[['Yes'] * 7])

		print("AED sheet complete")

		print("Oxygen log")

		print("Connecting to Oxygen sheet")
		oxy_sheet = client.open_by_url(oxyUrl).worksheet('Oxygen Log')

		empty_cell = next_empty_row(oxy_sheet, 'A')

		date_cell = "A"+str(empty_cell)
		time_cell = "B"+str(empty_cell)
		name_cell = "C"+str(empty_cell)
		level_cell = "D"+str(empty_cell)
		yes_cells = "E"+str(empty_cell)+":L"+str(empty_cell)
		
		print(date_cell, time_cell, name_cell, yes_cells)

		print("Updating cells")
		oxy_sheet.update(range_name=date_cell, values=[[date_str]])
		oxy_sheet.update(range_name=time_cell, values=[[time_input]])
		oxy_sheet.update(range_name=name_cell, values=[['CS']])
		oxy_sheet.update(range_name=level_cell, values=[[oxy_level]])
		oxy_sheet.update(range_name=yes_cells, values=[['Yes'] * 8])

		print("Oxygen sheet complete")

		print("Opening Complete")



		



if __name__ == "__main__":
	main()