Update google_sheet_Legend.py

google_sheet_Legend.py

@@ -73,7 +73,7 @@ def update_sheet(spreadsheetId, SimilarAreaDictionary, pdfpath):
     if splittedpdfpath[-2].startswith('2.2') or splittedpdfpath[-2].startswith('2.1'):
         worksheet.set_dataframe(start='A1', df=SimilarAreaDictionary)
     if splittedpdfpath[-2].startswith('2.8'):
-        DoorsLegend(SimilarAreaDictionary,spreadsheetId,worksheet)
+        DoorsLegend(SimilarAreaDictionary,spreadsheetId)
         return
     
     else:
@@ -685,7 +685,9 @@ def deletefromlegend(deletedrows, SimilarAreaDictionarycopy, section, areaPermAr
     return SimilarAreaDictionarycopy
 
 #########################
-def DoorsLegend(Dictionary,spreadsheetId,worksheet):
+def DoorsLegend(Dictionary,spreadsheetId):
+    global worksheetw
+    worksheetw.clear()
     top_header_format = [
             {'mergeCells': {  # legend and data created
                 'mergeType': 'MERGE_ROWS',
@@ -698,10 +700,11 @@ def DoorsLegend(Dictionary,spreadsheetId,worksheet):
                 }
             }}
     ]
-    spreadsheet_service.spreadsheets().batchUpdate(spreadsheetId=spreadsheetId, body={'requests': top_header_format}).execute()
-    worksheet.cell((1, 1)).value = 'Legend and Data Created'
-    worksheet.set_dataframe(start='A2', df=Dictionary)
-
+    print('here')
+    # spreadsheet_service.spreadsheets().batchUpdate(spreadsheetId=spreadsheetId, body={'requests': top_header_format}).execute()
+    worksheetw.cell((1, 1)).value = 'Legend and Data Created'
+    worksheetw.set_dataframe(start='A2', df=Dictionary)
+    print('here1')
     body2={ 
       "requests": [
         {
@@ -771,48 +774,38 @@ def DoorsLegend(Dictionary,spreadsheetId,worksheet):
         }
       ]
     }
-    spreadsheet_service.spreadsheets().batchUpdate(spreadsheetId=spreadsheetId, body=body2).execute()
-    model_cell = worksheet.cell('A1')
+    print('here2')
+    # spreadsheet_service.spreadsheets().batchUpdate(spreadsheetId=spreadsheetId, body=body2).execute()
+    model_cell = worksheetw.cell('A1')
     model_cell.set_text_format('bold', True)
     model_cell.set_horizontal_alignment(pygsheets.custom_types.HorizontalAlignment.CENTER)
     model_cell.color = (213 / 255, 219 / 255, 255 / 255)
-    pygsheets.DataRange('A2', 'B2', worksheet=worksheet).apply_format(model_cell)
+    pygsheets.DataRange('A2', 'B2', worksheet=worksheetw).apply_format(model_cell)
+    print('here3')
+    return Dictionary
 #########################
 
-def delete3_2(deletedrows,dictionary):
+def deletedoors(deletedrows,dictionary):
 
   items=[]
-  # print('dletefromlegend3.2')
+  # print('dletefromlegend2.8')
   idx=0
   myDict=eval(dictionary)
   df_doors=pd.DataFrame(myDict)
-  # deletedrows=eval(deletedrows)
   strings=deletedrows['content']
+  print('strings',strings)
   for item in strings:
     newitem=str(item).split('\n \n')
-    if 'single' in str(newitem):
+    if 'Single' in str(newitem):
       current_value = df_doors.loc[df_doors['Type'] == 'Single Doors', 'Quantity'].values[0]
-
-      # Step 2: Subtract one from the current value
       new_value = int(current_value) - 1
-
-      # Step 3: Update the DataFrame with the new value
       df_doors.loc[df_doors['Type'] == 'Single Doors', 'Quantity'] = str(new_value)
-
-      # Print the updated DataFrame to verify the change
-      # print(df_doors)
-    elif 'double' in str(newitem):
+    elif 'Double' in str(newitem):
       current_value = df_doors.loc[df_doors['Type'] == 'Double Doors', 'Quantity'].values[0]
-      # print(current_value)
-      # Step 2: Subtract one from the current value
       new_value = int(current_value) - 1
-      # print(new_value)
-      # Step 3: Update the DataFrame with the new value
       df_doors.loc[df_doors['Type'] == 'Double Doors', 'Quantity'] = str(new_value)
-
-      # Print the updated DataFrame to verify the change
-      # print(df_doors)
   df_doors = df_doors[['Type', 'Quantity']]
+  print(df_doors)
   return df_doors
 
 #########################