读取数据库理财师默认模版

app/service/report_service.py

@@ -15,7 +15,8 @@ from sqlalchemy import and_, or_
 # from sqlalchemy import union_all
 
 from app.api.engine import TAMP_SQL, tamp_diagnose_app_engine, template_folder, temp_img_save_folder, pdf_save_folder
-from app.model.tamp_diagnose_app import HoldReport, HoldDiagnoseReport, PeriodicReport, Customer, ReportView
+from app.model.tamp_diagnose_app import HoldReport, HoldDiagnoseReport, PeriodicReport, Customer, ReportView, \
+    FundReportManange
 from app.utils.jinjia2html_v2 import DataIntegrate
 from app.utils.celery import save_pdf
 
@@ -188,8 +189,9 @@ def make_report(args):
     args['create_time'] = datetime.datetime.now()
     args['update_status'] = 1
 
-    # args.pop('type')
+    args.pop('type')
     record_id = 0
+    default_template = ''
     with TAMP_SQL(tamp_diagnose_app_engine) as tamp_diagnose_app:
         tamp_diagnose_session = tamp_diagnose_app.session
         report_record = None
@@ -204,11 +206,18 @@ def make_report(args):
         tamp_diagnose_session.add(report_record)
         tamp_diagnose_session.flush()
         record_id = report_record.id
-
+        template = tamp_diagnose_session.query(FundReportManange).filter(and_(
+            FundReportManange.ifa_id == ifa_id,
+            FundReportManange.type == type,
+            FundReportManange.default == 1,
+            FundReportManange.default_template.isnot(None)
+        )).first()
+        if template:
+            default_template = json.loads(template.default_template)
     # save_pdf.delay(record_id, ifa_id, customer_id, pdf_name, type)
     try:
         dt = DataIntegrate(ifa_id=ifa_id, customer_id=customer_id, pdf_name=pdf_name, type=type)
-        report_data = dt.get_template_data()
+        report_data = dt.get_template_data(default_template=default_template)
         report_data = json.dumps(report_data).replace(template_folder, temp_img_save_folder)
         report_data = json.loads(report_data)
         dt.render_data(data=report_data)