fix

app/model/tamp_diagnose_app.py

@@ -147,15 +147,16 @@ class HoldDiagnoseReport(Base, BaseModel):
             res = tamp_user_session.execute(sql)
             sql2 = "select customer_name from tamp_diagnose_app.customer_view where id = '{}'".format(self.customer_id)
             res2 = tamp_diagnose_app_session.execute(sql2)
-            if res.fetchone():
-                data['author_name'] = res.fetchone()[0]
-            else:
-                data['author_name'] = ''
-            if res2.fetchone():
-                data['customer_name'] = res2.fetchone()[0]
-            else:
-                data['customer_name'] = ''
-            data['type'] = 2
+            if data:
+                if res.fetchone():
+                    data['author_name'] = res.fetchone()[0]
+                else:
+                    data['author_name'] = ''
+                if res2.fetchone():
+                    data['customer_name'] = res2.fetchone()[0]
+                else:
+                    data['customer_name'] = ''
+                data['type'] = 2
         return data
 
 
@@ -187,15 +188,16 @@ class HoldReport(Base, BaseModel):
             res = tamp_user_session.execute(sql)
             sql2 = "select customer_name from tamp_diagnose_app.customer_view where id = '{}'".format(self.customer_id)
             res2 = tamp_diagnose_app_session.execute(sql2)
-            if res.fetchone():
-                data['author_name'] = res.fetchone()[0]
-            else:
-                data['author_name'] = ''
-            if res2.fetchone():
-                data['customer_name'] = res2.fetchone()[0]
-            else:
-                data['customer_name'] = ''
-            data['type'] = 1
+            if data:
+                if res.fetchone():
+                    data['author_name'] = res.fetchone()[0]
+                else:
+                    data['author_name'] = ''
+                if res2.fetchone():
+                    data['customer_name'] = res2.fetchone()[0]
+                else:
+                    data['customer_name'] = ''
+                data['type'] = 1
         return data
 
 
@@ -227,15 +229,16 @@ class PeriodicReport(Base, BaseModel):
             res = tamp_user_session.execute(sql)
             sql2 = "select customer_name from tamp_diagnose_app.customer_view where id = '{}'".format(self.customer_id)
             res2 = tamp_diagnose_app_session.execute(sql2)
-            if res.fetchone():
-                data['author_name'] = res.fetchone()[0]
-            else:
-                data['author_name'] = ''
-            if res2.fetchone():
-                data['customer_name'] = res2.fetchone()[0]
-            else:
-                data['customer_name'] = ''
-            data['type'] = 3
+            if data:
+                if res.fetchone():
+                    data['author_name'] = res.fetchone()[0]
+                else:
+                    data['author_name'] = ''
+                if res2.fetchone():
+                    data['customer_name'] = res2.fetchone()[0]
+                else:
+                    data['customer_name'] = ''
+                data['type'] = 3
         return data