🩹 尝试修复简化信息匹配 #2

Cp0204 · Jun 11, 2024 · 6388e75 · 6388e75
1 parent afa0106
commit 6388e75
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/telecom_class.py b/telecom_class.py
@@ -200,24 +200,24 @@ def to_summary(self, data, phonenum=""):
             return {}
         phonenum = phonenum or self.phonenum
         # 总流量
-        flow_use = int(data["flowInfo"]["totalAmount"]["used"])
-        flow_balance = int(data["flowInfo"]["totalAmount"]["balance"])
+        flow_use = int(data["flowInfo"]["totalAmount"]["used"] or 0)
+        flow_balance = int(data["flowInfo"]["totalAmount"]["balance"] or 0)
         flow_total = flow_use + flow_balance
         # 通用流量
-        common_use = int(data["flowInfo"]["commonFlow"]["used"])
-        common_balance = int(data["flowInfo"]["commonFlow"]["balance"])
+        common_use = int(data["flowInfo"]["commonFlow"]["used"] or 0)
+        common_balance = int(data["flowInfo"]["commonFlow"]["balance"] or 0)
         common_total = common_use + common_balance
         # 专用流量
-        special_use = int(data["flowInfo"]["specialAmount"]["used"])
-        special_balance = int(data["flowInfo"]["specialAmount"]["balance"])
+        special_use = int(data["flowInfo"]["specialAmount"]["used"] or 0)
+        special_balance = int(data["flowInfo"]["specialAmount"]["balance"] or 0)
         special_total = special_use + special_balance
         # 语音通话
-        voice_usage = int(data["voiceInfo"]["voiceDataInfo"]["used"])
-        voice_balance = int(data["voiceInfo"]["voiceDataInfo"]["balance"])
-        voice_total = int(data["voiceInfo"]["voiceDataInfo"]["total"])
+        voice_usage = int(data["voiceInfo"]["voiceDataInfo"]["used"] or 0)
+        voice_balance = int(data["voiceInfo"]["voiceDataInfo"]["balance"] or 0)
+        voice_total = int(data["voiceInfo"]["voiceDataInfo"]["total"] or 0)
         # 余额
         balance = int(
-            float(data["balanceInfo"]["indexBalanceDataInfo"]["balance"]) * 100
+            float(data["balanceInfo"]["indexBalanceDataInfo"]["balance"] or 0) * 100
         )
         # 流量包列表
         flowItems = []