Update API.py

API.py

@@ -17,6 +17,15 @@ import pandas as pd
 from googleapiclient.discovery import build
 from google.oauth2 import service_account
 import pygsheets
+import numpy as np
+
+prjnamesURL= 'https://docs.google.com/spreadsheets/d/1I4P7wiCXTAwnld0YQfUJC8bPtuiETna1HGWVuHbuDBk/edit#gid=0'
+prjpartsURL= 'https://docs.google.com/spreadsheets/d/1I4P7wiCXTAwnld0YQfUJC8bPtuiETna1HGWVuHbuDBk/edit#gid=2102042200'
+prjsectionsURL= 'https://docs.google.com/spreadsheets/d/1I4P7wiCXTAwnld0YQfUJC8bPtuiETna1HGWVuHbuDBk/edit#gid=1708565170'
+
+dfNames=pd.DataFrame()
+dfParts=pd.DataFrame()
+
 def AuthorizeGoogleSheets():
   SCOPES = [
   'https://www.googleapis.com/auth/spreadsheets',
@@ -123,48 +132,42 @@ def AppendtablestoSheets(ws):
 # %%
 #-------------------------------------------------------------------------------CALL THIS FUNCTION TO BEGIN----------------------------------------------------------------------
 #--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-def getPrjNames():
-  ws,gc,spreadsheetId=createSheet()
-  AppendtablestoSheets(ws)
-  projectNamesSheet=ws.worksheet_by_title('Project Names')
-  prjnamesfromSheet=projectNamesSheet.get_col(2, returnas='matrix', include_tailing_empty=False)
-  # print(prjnamesfromSheet[2])
-  return prjnamesfromSheet[1:]
+def getPrjNames(url=prjnamesURL):
+  #url first sheet of names 
+  url_1 = url.replace('/edit#gid=', '/export?format=csv&gid=')
+  df = pd.read_csv(url_1)
+  return df['ProjectName'].tolist()
 
 # %%
-def getprjParts(chosenprjname):
+def getprjParts(chosenprjname,urlNames=prjnamesURL,urlparts=prjpartsURL):
+
+  url_1 = urlNames.replace('/edit#gid=', '/export?format=csv&gid=')
+  global dfNames
+  dfNames = pd.read_csv(url_1)
+  chosenprjname=chosenprjname.replace('"', '')
+  projectNameID=dfNames.iloc[np.where(dfNames['ProjectName'].astype(str)==chosenprjname)]['ProjectId']  #ProjectName == chosenprjname
+  url_2 = urlparts.replace('/edit#gid=', '/export?format=csv&gid=')
+  global dfParts
+  dfParts = pd.read_csv(url_2)
+  dfParts= dfParts.iloc[np.where(dfParts['ProjectId']==int(projectNameID))]
+  parts=dfParts['ProjectPart'].tolist()
+  print(parts)
 #Get projectName Id
-  ws,gc,spreadsheetId=createSheet()
-  projectNamesSheet=ws.worksheet_by_title('Project Names')
-  prjnamesfromSheet= getPrjNames()
-  index=prjnamesfromSheet.index(chosenprjname.replace('"', ''))
-  # index,indexE = [o for o, x in prjnamesfromSheet if x == str(chosenprjname)] #==clicked value
-  prjnameId=projectNamesSheet.cell((index+2 ,1)).value #retrieved Id
-  print('iddd',prjnameId)
-  #Get projectParts
-  projparts=[]
-  projectPartsSheet=ws.worksheet_by_title('Project Parts')
-  prjpartsfromSheet=projectPartsSheet.get_col(1, returnas='matrix', include_tailing_empty=False)
-
-  indices = [o for o, x in enumerate(prjpartsfromSheet[1:]) if x == prjnameId] #==clicked value
-  print('ind',indices)
-  for index in indices:
-    projparts.append([projectPartsSheet.cell((index+2 ,3)).value, projectPartsSheet.cell((index+2 ,2)).value]) #retrieved Id
-  return projparts
+  
+  return  parts
 
 # %%
-def getprjSections(chosenpart):
+def getprjSections(chosenprjname, chosenpart,urlSection=prjsectionsURL):
   #GetprojectSections
-  ws,gc,spreadsheetId=createSheet()
-  projsections=[]
-  projectSectionsSheet=ws.worksheet_by_title('Sections')
-  prjsectionsfromSheet=projectSectionsSheet.get_col(2, returnas='matrix', include_tailing_empty=False)
+  url_1 = urlSection.replace('/edit#gid=', '/export?format=csv&gid=')
+  dfSections = pd.read_csv(url_1)
+  chosenprjname=chosenprjname.replace('"', '')
   chosenpart=chosenpart.replace('"', '')
-  indices2 = [o for o, x in enumerate(prjsectionsfromSheet) if x == chosenpart] #part clicked on
-  print('indices2',indices2)
-  for index in indices2:
-    projsections.append(projectSectionsSheet.cell((index+2 ,3)).value) #retrieved Id
-  return projsections
+  projectId=dfNames.iloc[np.where(dfNames['ProjectName'].astype(str)==chosenprjname)]['ProjectId']  #ProjectName == chosenprjname
+  projectPartID=dfParts.iloc[np.where(dfParts['ProjectPart'].astype(str)==chosenpart)]['ProjectPartId']
+  dfSections=dfSections.iloc[np.where((dfSections['ProjectId']==int(projectId)) & (dfSections['ProjectPartId']==int(projectPartID)) ) ]
+  sections=dfSections['ProjectSection'].tolist()
+  return sections
 
 # %%
 # tableTrial=pd.DataFrame(data=[[0,'0x','part1'],[0,'1x','part2']],columns= ['ProjectId','ProjectPartId','ProjectPart'])
@@ -183,3 +186,11 @@ def getprjSections(chosenpart):
 
 
 
+
+##############################################################################
+def update_df(url): #get file from drive
+    
+    # threading.Timer(200.0,update_df).start() 
+    url_1 = url.replace('/edit#gid=', '/export?format=csv&gid=')
+    df = pd.read_csv(url_1)
+    return df