Update pixelconversion.py

pixelconversion.py

@@ -45,14 +45,20 @@ def is_content_rotated(docpage):
       print("Page content does not appear visually rotated.")
       return False
 
-def drawisrotated(data,dpi=0):
-  docPath = fitz.open("pdf",data) #dropbox path
+def drawisrotated(data=0,pdf_content=0,dpi=0):
+  if pdf_content:
+    docPath = fitz.open(stream=pdf_content, filetype="pdf")
+  else:
+    docPath = fitz.open("pdf",data) #dropbox path
   pageDocPath=docPath[0]
   
   bool_T_F=is_content_rotated(pageDocPath)
 
   if bool_T_F==True:
-    docPath = fitz.open("pdf",data) #dropbox path
+    if pdf_content:
+      docPath = fitz.open(stream=pdf_content, filetype="pdf")
+    else:
+      docPath = fitz.open("pdf",data) #dropbox path
     pageDocPath=docPath[0]
     pageDocPath.draw_rect([250,250,pageDocPath.mediabox.width-500,pageDocPath.mediabox.height-500],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9)
     if dpi:
@@ -77,45 +83,57 @@ def drawisrotated(data,dpi=0):
     # docPath.save('drawnOn.pdf')
     return docPath,area,perimeter
   else:
-    if dpi:
-      doc,area,perimeter=openDrawPDF(data,dpi=dpi)
-      return doc,area,perimeter 
+    if pdf_content:
+      if dpi:
+        doc,area,perimeter=openDrawPDF(pdf_content=pdf_content,dpi=dpi)
+        return doc,area,perimeter 
+      else:
+        doc,area,perimeter=openDrawPDF(pdf_content=pdf_content)
+        return doc,area,perimeter
     else:
-      doc,area,perimeter=openDrawPDF(data)
-      return doc,area,perimeter
-
+      if dpi:
+        doc,area,perimeter=openDrawPDF(data,dpi=dpi)
+        return doc,area,perimeter 
+      else:
+        doc,area,perimeter=openDrawPDF(data)
+        return doc,area,perimeter
+    
 
-"""### Open PDF and draw a rectangle on it using Fitz"""
-def openDrawPDF(data,dpi=0):
-  
-  doc=fitz.open("pdf", data)
-  page = doc[0]
-  print('piixelsize',page.mediabox)
+    
 
-  if page.rect.height >page.rect.width:
-    rectText=fitz.Rect(300, 200, (page.mediabox.width-80),( page.mediabox.width+80) )
-  else:
-    rectText=fitz.Rect(500, 200, (page.mediabox.height-80),( page.mediabox.height+80) )
 
-  page.draw_rect([0+10,0+10,page.mediabox.width-10,page.mediabox.height-10],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
+def openDrawPDF(data=0,pdf_content=0,dpi=0):
   
-  if dpi:
-    pix=page.get_pixmap(dpi=dpi)
-    newwidth_ratio=pix.width/page.mediabox.width
-    newheight_ratio=pix.height/page.mediabox.height
-  else:
-    newwidth_ratio=1
-    newheight_ratio=1
+    if pdf_content:
+      doc = fitz.open(stream=pdf_content, filetype="pdf")
+    else:
+      doc = fitz.open("pdf",data) #dropbox path
+    page = doc[0]
+    print(page.rotation)
 
-  area=((page.mediabox.width-10-10)*newwidth_ratio)*((page.mediabox.height-10-10)*newheight_ratio)
-  perimeter=(((page.mediabox.width-10-10)*2)*newwidth_ratio)+(((page.mediabox.height-10-10)*2)*newheight_ratio)
+    if page.rect.height > page.rect.width:
+        rectText = fitz.Rect(300, 200, (page.mediabox.width - 80), (page.mediabox.width + 80))
+    else:
+        rectText = fitz.Rect(500, 200, (page.mediabox.height - 80), (page.mediabox.height + 80))
+    page.draw_rect([0 + 10, 0 + 10, page.mediabox.width - 10, page.mediabox.height - 10],color=(75 / 255, 0, 130 / 255), width=1, fill=(75 / 255, 0, 130 / 255), fill_opacity=0.9)
 
+    if dpi:
+        pix = page.get_pixmap(dpi=dpi)
+        newwidth_ratio = pix.width / page.mediabox.width
+        newheight_ratio = pix.height / page.mediabox.height
+    else:
+        print('No DPI specified')
+        newwidth_ratio = 1
+        newheight_ratio = 1
 
-  text = """Scale Document"""
-  annot1=page.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1,1,1), text_color=(1,1,1), rotate= page.rotation, align=1)
-  annot1.update()
-  print('fitzzzzzzzzzzzz',fitz.__version__)
-  # doc.save('drawnOn.pdf')#, encryption=encrypt, permissions=perm)
-  return  doc,area,perimeter
-"""### Extract color"""
+    # Apply the transformation to the rectangle's area and perimeter
+    area = ((page.mediabox.width - 10 - 10) * newwidth_ratio) * ((page.mediabox.height - 10 - 10) * newheight_ratio)
+    perimeter = (((page.mediabox.width - 10 - 10) * 2) * newwidth_ratio) + (((page.mediabox.height - 10 - 10) * 2) * newheight_ratio)
 
+    # Apply the transformation to the text annotation (shift and scale its position)
+    text = """Scale Document"""
+    annot1 = page.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1, 1, 1),text_color=(1, 1, 1), rotate=page.rotation, align=1)
+    annot1.update()
+    # print('fitzzzzzzzzzzzz',fitz.__version__)
+    # doc.save('drawnOn_transformed.pdf')
+    return doc, area, perimeter