Update src/pyscripts/file_process.py

src/pyscripts/file_process.py

@@ -1,4 +1,5 @@
 import io
+import os
 import fitz  # PyMuPDF
 
 pdf_path_dict = {
@@ -8,8 +9,11 @@ pdf_path_dict = {
     "演讲": "src/assets/WWEC演讲嘉宾邀请函.pdf",
     "致辞": "src/assets/WWEC演讲嘉宾邀请函.pdf",
 }
-font_path = "src/assets/ZhouZiSongTi7000Zi-2.otf"
-font_path = "src/assets/FangZhengHeiTiJianTi-1.ttf"
+# font_path = "src/assets/NotoSansHans-Regular.otf"
+# other_font_path = "src/assets/SourceHanSansCN-VF-2.otf"
+# other_font_path = "src/assets/ZhouZiSongTi7000Zi-2.otf"
+font_path = "src/assets/字魂59号-创粗黑.ttf"
+other_font_path = "src/assets/FangZhengHeiTiJianTi-1.ttf"
 # font = fitz.Font(fontfile=font_path)
 
 def getInvitationPDF(data):
@@ -25,27 +29,29 @@ def getInvitationPDF(data):
     pdfStream = io.BytesIO()
     doc.save(pdfStream)
     doc.close()
-    # with open("output/output.pdf", 'wb') as f:
-    #     f.write(pdfStream.getvalue())
+    if os.path.exists("src/assets/output"):
+        with open("src/assets/output/output.pdf", 'wb') as f:
+            f.write(pdfStream.getvalue())
     return pdfStream
 
 
-def writeInvitationPDF(person, debug=False):
+def writeInvitationPDF(person):
     font_size=30
     pdf_path = pdf_path_dict[person.category]
     doc = fitz.open(pdf_path)
     page = doc.load_page(1)
+    page.insert_font(fontname="name_font", fontfile=font_path)
     page.insert_font(fontname="f", fontfile=font_path)
     page.insert_text((145, 80),
-        person.name+person.title, fontname="f", fontsize=min(font_size, font_size*5/len(person.name+person.title)), color=(1, 1, 1))
+        person.name+person.title, fontname="name_font", fontsize=min(font_size, font_size*5/len(person.name+person.title)), color=(1, 1, 1))
     if person.category == "演讲" or person.category == "致辞":
         page.insert_text((80, 375),
             person.conference, fontname="f", fontsize=min(font_size, font_size*14/len(person.conference)), color=(1, 1, 1))
     pdfStream = io.BytesIO()
     doc.save(pdfStream)
     doc.close()
-    if debug:
-        with open("output/output.pdf", 'wb') as f:
+    if os.path.exists("src/assets/output"):
+        with open("src/assets/output/output.pdf", 'wb') as f:
             f.write(pdfStream.getvalue())
     return pdfStream
 
@@ -53,11 +59,11 @@ def writeInvitationPDF(person, debug=False):
 if __name__ == "__main__":
     from collections import namedtuple
     Person = namedtuple('Person', ['name', 'title', 'category', 'conference', 'respond_person'])
-    person = Person(name="X专业测试XXX", title="先生", category="致辞", conference="8月21日中欧智慧论坛，做主题演讲", respond_person="Eazy")
+    person = Person(name="周鸿祎", title="先生", category="致辞", conference="8月21日中欧智慧论坛，做主题演讲", respond_person="Eazy")
 
     output_path = 'output/output.pdf'
 
     # getInvitationPDF(data)
-    writeInvitationPDF(person, debug=True)
+    writeInvitationPDF(person)
     # with open(output_path, 'wb') as f:
     #     f.write(pdfStream.getvalue())