ymcmy commited on
Commit
e2cac91
·
verified ·
1 Parent(s): fc6d630

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +12 -6
utils.py CHANGED
@@ -96,10 +96,16 @@ def gen_html(num):
96
  return all_q
97
 
98
  def convert_html_to_pdf(source_html, output_filename):
99
- result_file = open(output_filename, "w+b")
100
- pisa_status = pisa.CreatePDF(
101
- io.StringIO(source_html),
102
- dest=result_file
103
- )
104
- result_file.close()
 
 
 
 
 
 
105
  return pisa_status.err
 
96
  return all_q
97
 
98
  def convert_html_to_pdf(source_html, output_filename):
99
+ try:
100
+ result_file = open(output_filename, "w+b")
101
+ pisa_status = pisa.CreatePDF(
102
+ io.StringIO(source_html),
103
+ dest=result_file
104
+ )
105
+ result_file.close()
106
+ if pisa_status.err:
107
+ raise Exception("Error converting HTML to PDF")
108
+ except Exception as e:
109
+ print(f"Error: {e}")
110
+ raise
111
  return pisa_status.err