qgyd2021 commited on
Commit
31e8b01
·
1 Parent(s): b68dec5
toolbox/porter/tasks/douyin_live_to_bilibili_live_task.py CHANGED
@@ -83,19 +83,33 @@ class DouyinLiveToBilibiliLiveTask(BaseTask):
83
  self.success_rate_of_get_live_info_by_web_enter.append(0)
84
  return None
85
  data = js["data"]["data"]
 
86
  if len(data) == 0:
87
  self.success_rate_of_get_live_info_by_web_enter.append(1)
88
  return None
89
  data = data[0]
90
  status = data["status"]
91
  title = data["title"]
92
- stream_data = json.loads(data["stream_url"]["live_core_sdk_data"]["pull_data"]["stream_data"])
93
- stream_data = stream_data["data"]
 
 
 
 
 
 
 
 
 
 
 
 
94
 
95
  result = {
96
  "status": status,
97
  "title": title,
98
  "stream_data": stream_data,
 
99
  }
100
  self.success_rate_of_get_live_info_by_web_enter.append(1)
101
  return result
 
83
  self.success_rate_of_get_live_info_by_web_enter.append(0)
84
  return None
85
  data = js["data"]["data"]
86
+
87
  if len(data) == 0:
88
  self.success_rate_of_get_live_info_by_web_enter.append(1)
89
  return None
90
  data = data[0]
91
  status = data["status"]
92
  title = data["title"]
93
+
94
+ if status == 4:
95
+ paid_live_data = {}
96
+ stream_data = {}
97
+ else:
98
+ try:
99
+ paid_live_data = data["paid_live_data"]
100
+ stream_data = data["stream_url"]["live_core_sdk_data"]["pull_data"]["stream_data"]
101
+ except KeyError as error:
102
+ logger.info(f"data.keys: {data.keys()}, error type: {type(error)}, error text: {error}, traceback: {traceback.format_exc()}")
103
+ self.success_rate_of_get_live_info_by_web_enter.append(0)
104
+ return None
105
+ stream_data = json.loads(stream_data)
106
+ stream_data = stream_data["data"]
107
 
108
  result = {
109
  "status": status,
110
  "title": title,
111
  "stream_data": stream_data,
112
+ "paid_live_data": paid_live_data,
113
  }
114
  self.success_rate_of_get_live_info_by_web_enter.append(1)
115
  return result