Update Code_2_7.py

Code_2_7.py

@@ -85,8 +85,11 @@ This portion is used to convert vertices read from dxf to pixels in order to acc
 
 """PDF to image"""
 
-def pdftoimg(datadoc):
-  doc =fitz.open('pdf',datadoc)
+def pdftoimg(datadoc,pdf_content=0):
+  if pdf_content:
+      doc = fitz.open(stream=pdf_content, filetype="pdf")
+  else:
+      doc =fitz.open('pdf',datadoc)
   page=doc[0]
   pix = page.get_pixmap()  # render page to an image
   pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
@@ -119,9 +122,12 @@ def get_paper_size_in_inches(width, height):
             return size
     return "Unknown Size"
 
-def analyze_pdf(datadoc):
+def analyze_pdf(datadoc,pdf_content=0):
     # Open the PDF file
-    pdf_document = fitz.open('pdf',datadoc)
+    if pdf_content:
+        pdf_document = fitz.open(stream=pdf_content, filetype="pdf")
+    else:
+        pdf_document = fitz.open('pdf',datadoc)
 
     # Iterate through pages and print their sizes
     for page_number in range(len(pdf_document)):
@@ -176,10 +182,11 @@ def switch_case(argument):
 
 
 
-def RetriveRatio(datadoc,dxfpath):
-
-  width,height,paper_size = analyze_pdf (datadoc)
-
+def RetriveRatio(datadoc,dxfpath,pdf_content=0):
+  if pdf_content:
+    width,height,paper_size = analyze_pdf (datadoc,pdf_content)
+  else:
+      width,height,paper_size = analyze_pdf (datadoc)
   if(width > height ):
     bigger=width
   else:
@@ -1092,10 +1099,12 @@ def rotate_polygon(polygon, angle, pdfrotation,width,height,center_point=(0, 0))
 
 
 
-def Create_DF(dxfpath,datadoc,hatched_areas):
-
-  FinalRatio= RetriveRatio(datadoc,dxfpath)
+def Create_DF(dxfpath,datadoc,hatched_areas,pdf_content=0):
 
+  if pdf_content:
+    FinalRatio= RetriveRatio(datadoc,dxfpath,pdf_content)
+  else: 
+      FinalRatio= RetriveRatio(datadoc,dxfpath)
   # hatched_areas = get_hatched_areas(datadoc,dxfpath,FinalRatio)
 
   # hatched_areas=remove_duplicate_shapes(new_hatched_areas)
@@ -1334,19 +1343,27 @@ def calculate_distance(p1, p2):
 
 
 
-def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,SearchArray,Thickness,pdfpath=0,pdfname=0):
+def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,SearchArray,Thickness,pdfpath=0,pdfname=0,pdf_content=0):
   OutputPdfStage1='BB Trial.pdf'
-  FinalRatio= RetriveRatio(datadoc,dxfpath)
+  if pdf_content:
+    FinalRatio= RetriveRatio(datadoc,dxfpath,pdf_content)
+  else:
+      FinalRatio= RetriveRatio(datadoc,dxfpath)
 
   # hatched_areas = get_hatched_areas(datadoc,dxfpath,FinalRatio)
   # hatched_areas=remove_duplicate_shapes(new_hatched_areas)
-
-  img,pix2=pdftoimg(datadoc)
+  if pdf_content:
+    img,pix2=pdftoimg(datadoc,pdf_content)
+  else:
+      img,pix2=pdftoimg(datadoc)
   flipped_horizontal=flip(img)
   allcnts = []
   imgg = flipped_horizontal
   # imgtransparent1=imgg.copy()
-  doc = fitz.open('pdf',datadoc)
+  if pdf_content:
+      doc = fitz.open(stream=pdf_content, filetype="pdf")
+  else:
+    doc = fitz.open('pdf',datadoc)
   page2 = doc[0]
   rotationOld=page2.rotation
   derotationMatrix=page2.derotation_matrix
@@ -1374,7 +1391,10 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,SearchArray,Thickness,pdfpa
   allshapes=[]
   # Iterate through each polygon in metric units
   NewColors = []
-  SimilarAreaDictionary=Create_DF(dxfpath,datadoc,hatched_areas)
+  if pdf_content:
+      SimilarAreaDictionary=Create_DF(dxfpath,datadoc,hatched_areas,pdf_content)
+  else:
+      SimilarAreaDictionary=Create_DF(dxfpath,datadoc,hatched_areas)
   i=0
   flagcolor = 0
   ColorCounter = 0
@@ -1675,10 +1695,15 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,SearchArray,Thickness,pdfpa
   else:
         OutputPdfStage3 = remove_duplicate_annotations(OutputPdfStage2,threshold=10)
 
-
-  latestimg,pix=pdftoimg(OutputPdfStage3)
+  if pdf_content:
+    latestimg,pix=pdftoimg(OutputPdfStage3,pdf_content)
+  else:
+      latestimg,pix=pdftoimg(OutputPdfStage3)
   doc2 =fitz.open('pdf',OutputPdfStage3)
-  gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(grouped_df , pdfname,pdfpath)
+  if pdf_content:
+      gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(grouped_df , pdfname,pdfpath,pdf_content)
+  else:
+      gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(grouped_df , pdfname,pdfpath)
   list1=pd.DataFrame(columns=['content',  'id',  'subject','color'])
 
   # for page in doc: