Update Doors_Schedule.py

Doors_Schedule.py

@@ -1012,7 +1012,7 @@ def add_bluebeam_count_annotations_secondary(pdf_bytes, locations, main_info, se
     page = pdf_document[0]  # First page
     if len(main_info) == 2 and len(secondary_info) == 1:
         for loc in locations:
-            coor, lbl, clr, acous = loc
+            coor, lbl, acous, clr = loc
             clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
             for cor in coor:
                 #Create a Circle annotation (Count Markup)
@@ -1033,7 +1033,7 @@ def add_bluebeam_count_annotations_secondary(pdf_bytes, locations, main_info, se
 
     if len(main_info) == 2 and len(secondary_info) == 2:
         for loc in locations:
-            coor, lbl, clr, acous, fire = loc
+            coor, lbl, acous, fire, clr = loc
             clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
             for cor in coor:
                 #Create a Circle annotation (Count Markup)
@@ -1054,7 +1054,7 @@ def add_bluebeam_count_annotations_secondary(pdf_bytes, locations, main_info, se
 
     if len(main_info) == 3 and len(secondary_info) == 1:
         for loc in locations:
-            coor, lbl, w, clr, acous = loc
+            coor, lbl, w, acous, clr = loc
             clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
             for cor in coor:
                 #Create a Circle annotation (Count Markup)
@@ -1075,7 +1075,7 @@ def add_bluebeam_count_annotations_secondary(pdf_bytes, locations, main_info, se
 
     if len(main_info) == 3 and len(secondary_info) == 2:
         for loc in locations:
-            coor, lbl, w, clr, acous, fire = loc
+            coor, lbl, w, acous, fire, clr = loc
             clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
             for cor in coor:
                 #Create a Circle annotation (Count Markup)
@@ -1117,7 +1117,7 @@ def add_bluebeam_count_annotations_secondary(pdf_bytes, locations, main_info, se
 
     if len(main_info) == 4 and len(secondary_info) == 2:
         for loc in locations:
-            coor, lbl, w, h, clr, acous, fire = loc
+            coor, lbl, w, h, acous, fire, clr = loc
             clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
             for cor in coor:
                 #Create a Circle annotation (Count Markup)
@@ -1199,7 +1199,7 @@ import PyPDF2
 import io
 from PyPDF2.generic import TextStringObject  # ✅ Required for setting string values
 
-def add_bluebeam_count_annotations(pdf_bytes, locations):
+'''def add_bluebeam_count_annotations(pdf_bytes, locations):
     pdf_stream = io.BytesIO(pdf_bytes)  # Load PDF from bytes
     pdf_document = fitz.open("pdf", pdf_stream.read())  # Open PDF in memory
 
@@ -1271,6 +1271,81 @@ def add_bluebeam_count_annotations(pdf_bytes, locations):
     pdf_document.close()
 
     return output_stream.getvalue()  # Return the modified PDF as bytes
+'''
+def add_bluebeam_count_annotations(pdf_bytes, locations):
+    pdf_stream = io.BytesIO(pdf_bytes)  # Load PDF from bytes
+    pdf_document = fitz.open("pdf", pdf_stream.read())  # Open PDF in memory
+
+    page = pdf_document[0]  # First page
+    print(f"length of locations 0 from not sec presence: {len(locations[0])}")
+
+    for loc in locations:
+
+      if len(loc) == 3:
+            coor, lbl, clr = loc
+            clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
+            for cor in coor:
+                #Create a Circle annotation (Count Markup)
+                annot = page.add_circle_annot(
+                    fitz.Rect(cor[0] - 10, cor[1] - 10, cor[0] + 10, cor[1] + 10)  # Small circle
+                )
+
+                #Assign required Bluebeam metadata
+                annot.set_colors(stroke=clr, fill=(1, 1, 1))  # Set stroke color and fill white
+                annot.set_border(width=2)  # Border thickness
+                annot.set_opacity(1)  # Fully visible
+
+                #Set annotation properties for Bluebeam Count detection
+                annot.set_info("name", lbl)  # Unique name for each count
+                annot.set_info("subject", "Count")  #Bluebeam uses "Count" for Count markups
+                annot.set_info("title", lbl)  # Optional
+                annot.update()  # Apply changes
+      if len(loc) == 4:
+            coor, lbl, clr,w = loc
+            clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
+            for cor in coor:
+                #Create a Circle annotation (Count Markup)
+                annot = page.add_circle_annot(
+                    fitz.Rect(cor[0] - 10, cor[1] - 10, cor[0] + 10, cor[1] + 10)  # Small circle
+                )
+
+                #Assign required Bluebeam metadata
+                annot.set_colors(stroke=clr, fill=(1, 1, 1))  # Set stroke color and fill white
+                annot.set_border(width=2)  # Border thickness
+                annot.set_opacity(1)  # Fully visible
+
+                #Set annotation properties for Bluebeam Count detection
+                annot.set_info("name", lbl)  # Unique name for each count
+                annot.set_info("subject", "Count")  #Bluebeam uses "Count" for Count markups
+                annot.set_info("title", lbl)  # Optional
+                annot.update()  # Apply changes
+
+      if len(loc) == 5:
+            coor, lbl, clr,w,h = loc
+            clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
+            for cor in coor:
+                #Create a Circle annotation (Count Markup)
+                annot = page.add_circle_annot(
+                    fitz.Rect(cor[0] - 10, cor[1] - 10, cor[0] + 10, cor[1] + 10)  # Small circle
+                )
+
+                #Assign required Bluebeam metadata
+                annot.set_colors(stroke=clr, fill=(1, 1, 1))  # Set stroke color and fill white
+                annot.set_border(width=2)  # Border thickness
+                annot.set_opacity(1)  # Fully visible
+
+                #Set annotation properties for Bluebeam Count detection
+                annot.set_info("name", lbl)  # Unique name for each count
+                annot.set_info("subject", "Count")  #Bluebeam uses "Count" for Count markups
+                annot.set_info("title", lbl)  # Optional
+                annot.update()  # Apply changes
+
+    #Save modified PDF to a variable instead of a file
+    output_stream = io.BytesIO()
+    pdf_document.save(output_stream)
+    pdf_document.close()
+
+    return output_stream.getvalue()  # Return the modified PDF as bytes
 
 
 def get_user_input(user_words):