浏览代码

获取 rov 数据

罗俊辉 1 年之前
父节点
当前提交
2b6bc2c996
共有 1 个文件被更改,包括 12 次插入5 次删除
  1. 12 5
      data_process/update_mysql_data.py

+ 12 - 5
data_process/update_mysql_data.py

@@ -60,13 +60,20 @@ class DataProcessor(object):
             if label_obj:
                 total_return = label_obj.get('total_return', 0)
                 total_view = label_obj.get('total_view', 0)
-                if total_view == 0:
-                    label = None
-                else:
-                    if total_return == 0:
+                if total_return is not None and total_view is not None:
+                    total_return = float(total_return)
+                    total_view = float(total_view)
+                    if total_view == 0:
                         label = None
                     else:
-                        label = float(total_return) / float(total_view)
+                        if total_return == 0:
+                            label = None
+                        else:
+                            label = float(total_return) / float(total_view)
+                elif total_return is None and total_view is not None:
+                    label = 0
+                else:
+                    label = None
             else:
                 label = None
             return label, label_dt