获取内容类型字段

data_channel/douyin.py

@@ -58,8 +58,8 @@ class DY:
                 return list
             response.close()
             for i in range(len(data)):
-                entity_type = data[i].get('search_impr').get('entity_type')
-                if entity_type == 'GENERAL':
+                entity_type = data[i].get('media_type')
+                if entity_type == 4:
                     # is_top = data[i].get('is_top')  # 是否置顶
                     video_id = data[i].get('aweme_id')  # 文章id
                     status = sqlCollect.is_used(task_mark, video_id, mark, channel_id)