优化业务汇总再分类

app/api/export/crud.py

@@ -268,14 +268,15 @@ class ReferenceTypeClassification():
             else:
                 self.income.append(op)
         print(f"出账：{self.outcome}, 入账：{self.income}")
-        outcome = pd.DataFrame(self.outcome)
-        income = pd.DataFrame(self.income)
-        ogs = outcome.groupby("nickname").sum()
-        igs = income.groupby("nickname").sum()
-        # 格式化
         res_list = []
-        for k,v in ogs.to_dict().get('money').items():
-            res_list.append({"type": "出账", "name": k, "money": v})
-        for k,v in igs.to_dict().get('money').items():
-            res_list.append({"type": "入账", "name": k, "money": v})
+        if self.outcome:
+            outcome = pd.DataFrame(self.outcome)
+            ogs = outcome.groupby("nickname").sum()
+            for k, v in ogs.to_dict().get('money').items():
+                res_list.append({"type": "出账", "name": k, "money": v})
+        if self.income:
+            income = pd.DataFrame(self.income)
+            igs = income.groupby("nickname").sum()
+            for k,v in igs.to_dict().get('money').items():
+                res_list.append({"type": "入账", "name": k, "money": v})
         return res_list