Update google_sheet_Legend.py

google_sheet_Legend.py

@@ -61,7 +61,7 @@ def create_new_sheet(path):
     return spreadsheetId
 
 #########################
-def update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath):
+def update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath,pdf_content=0):
     ws = gc.open_by_key(spreadsheetId)
     global pdfpath0
     pdfpath0=pdfpath
@@ -69,8 +69,14 @@ def update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath):
     worksheet.title = 'Legend and data created'
     worksheet.clear()
     ws.create_developer_metadata('path', pdfpath)
-    splittedpdfpath = re.split(r'[`\-=~!@#$%^&*()_+\[\]{};\'\\:"|<,/<>?]', pdfpath)
-    if splittedpdfpath[-2].startswith('2.2') or splittedpdfpath[-2].startswith('2.1'):
+    
+    if pdf_content:
+       splittedpdfpath = re.split(r'[`\-=~!@#$%^&*()_+\[\]{};\'\\:"|<,/<>?]', pdfpath)
+       section=splittedpdfpath[-2]
+    else:
+       section=pdfpath
+       print(section)
+    if section.startswith('2.2') or section.startswith('2.1'):
         worksheet.set_dataframe(start='A1', df=SimilarAreaDictionary)
     # if splittedpdfpath[-2].startswith('2.8'):
     #     DoorsLegend(SimilarAreaDictionary,spreadsheetId)
@@ -143,12 +149,12 @@ def update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath):
           worksheet.update_col(10, list(SimilarAreaDictionary['Texts']), row_offset=3)
           worksheet.update_col(11, list(SimilarAreaDictionary['Comments']), row_offset=3)
         
-        if splittedpdfpath[-2].startswith('1.0'):
+        if section.startswith('1.0'):
             colorsUsed = [
                 [SimilarAreaDictionary['R'].iloc[i], SimilarAreaDictionary['G'].iloc[i], SimilarAreaDictionary['B'].iloc[i]]
                 for i in range(len(SimilarAreaDictionary))
             ]
-        elif splittedpdfpath[-2].startswith('3.2') or splittedpdfpath[-2].startswith('3.3') or splittedpdfpath[-2].startswith('2.7'):
+        else:#if section.startswith('3.2') or section.startswith('3.3') or section.startswith('2.7'):
             colorsUsed = list(SimilarAreaDictionary['Color'])
         rowsLen = len(SimilarAreaDictionary.values.tolist())
         lastcell = worksheet.cell((rowsLen + 2, 1))
@@ -265,7 +271,7 @@ def update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath):
     pygsheets.DataRange('A2', 'K2', worksheet=worksheet).apply_format(model_cell)
 
 #########################
-def legendGoogleSheets(SimilarAreaDictionary, path, pdfpath, spreadsheetId=0):
+def legendGoogleSheets(SimilarAreaDictionary, path, pdfpath, spreadsheetId=0,pdf_content=0):
     titles = gc.spreadsheet_titles()
     if path in titles:
         ws = gc.open(path)
@@ -277,7 +283,10 @@ def legendGoogleSheets(SimilarAreaDictionary, path, pdfpath, spreadsheetId=0):
         ws=gc.open_by_key(spreadsheetId)
     global worksheetw
     worksheetw = ws.worksheet(0) #legend
-    update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath)
+    if pdf_content:
+      update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath,pdf_content)
+    else:
+       update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath)
     spreadsheet_url = f"https://docs.google.com/spreadsheets/d/{spreadsheetId}"
     drive_service.permissions().update(transferOwnership=True, fileId=spreadsheetId, permissionId='11OfoB4Z6wOVII8mYmbnCbbqTQs7rYA65')