__init__.py 10.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503
  1. """
  2. @author: luojunhui
  3. 任务常量配置文件
  4. """
  5. class ColdStartTaskConst:
  6. """
  7. 冷启动任务常量配置
  8. """
  9. PUBLISHED_STATUS = 2 # 文章已发布状态
  10. INIT_STATUS = 1 # 文章初始状态
  11. BAD_STATUS = 0 # 低质量文章状态
  12. # 常量
  13. ACCOUNT_GOOD_STATUS = 1
  14. # 账号是否每日抓取
  15. ACCOUNT_DAILY_SCRAPE = 1
  16. ACCOUNT_NOT_DAILY_SCRAPE = 0
  17. # 默认值
  18. DEFAULT_VIEW_COUNT = 0
  19. DEFAULT_LIKE_COUNT = 0
  20. DEFAULT_ARTICLE_STATUS = 1
  21. DEFAULT_TIMESTAMP = 1717171200
  22. # 标题sensitivity
  23. TITLE_SENSITIVE = 1
  24. TITLE_NOT_SENSITIVE = 0
  25. # 文章联想深度
  26. ARTICLE_ASSOCIATION_MAX_DEPTH = 4
  27. # 相关分百分位阈值
  28. PERCENT_THRESHOLD = 95
  29. # 相关性分阈值
  30. CORRELATION_THRESHOLD = 0.5
  31. # 阅读量阈值
  32. READ_COUNT_THRESHOLD = 1000
  33. # 阅读均值倍数阈值
  34. READ_AVG_THRESHOLD = 1.3
  35. # 群发类型
  36. BULK_PUBLISH_TYPE = 9
  37. # 种子文章数量
  38. SEED_ARTICLE_LIMIT_NUM = 60
  39. class updatePublishedMsgTaskConst:
  40. """
  41. 更新已发布文章消息常量配置
  42. """
  43. # 爬虫详情接口返回code
  44. ARTICLE_ILLEGAL_CODE = 25012
  45. ARTICLE_DELETE_CODE = 25005
  46. ARTICLE_SUCCESS_CODE = 0
  47. ARTICLE_UNKNOWN_CODE = 10000
  48. # 请求爬虫详情接口状态码
  49. # 记录默认状态
  50. DEFAULT_STATUS = 0
  51. # 请求接口失败状态
  52. REQUEST_FAIL_STATUS = -1
  53. # 文章被删除状态
  54. DELETE_STATUS = -2
  55. # 未知原因无信息返回状态
  56. UNKNOWN_STATUS = -3
  57. # 文章违规状态
  58. ILLEGAL_STATUS = -4
  59. # 公众号类型(订阅号 or 服务号)
  60. # 订阅号
  61. SUBSCRIBE_TYPE_SET = {0, 1}
  62. # 服务号
  63. SERVICE_TYPE = 2
  64. # 监测周期(秒)
  65. MONITOR_PERIOD = 60 * 60 * 24 * 3
  66. # 新号抓文章周期
  67. NEW_ACCOUNT_CRAWL_PERIOD = 60 * 60 * 24 * 30
  68. # 订阅号,抓取失败失败率报警阈值
  69. SUBSCRIBE_FAIL_RATE_THRESHOLD = 0.3
  70. class UpdateAccountReadRateTaskConst:
  71. """
  72. 更新账号阅读率常量配置
  73. """
  74. # 阅读率统计周期(秒)
  75. STATISTICS_PERIOD = 31 * 24 * 60 * 60
  76. # 一天的秒数
  77. ONE_DAY_IN_SECONDS = 60 * 60 * 24
  78. # 相对变化率阈值
  79. RELATIVE_VALUE_THRESHOLD = 0.1
  80. # 发文类型
  81. UNLIMITED_PUBLISH_TYPE = 10002
  82. BULK_PUBLISH_TYPE = 9
  83. # 文章位置
  84. ARTICLE_INDEX_LIST = [1, 2, 3, 4, 5, 6, 7, 8]
  85. # 默认粉丝
  86. DEFAULT_FANS = 0
  87. # 最低粉丝量
  88. MIN_FANS = 1000
  89. class UpdateAccountReadAvgTaskConst:
  90. """
  91. 更新账号阅读均值常量配置
  92. """
  93. # 投流账号
  94. TOULIU_ACCOUNTS = {
  95. "gh_93e00e187787",
  96. "gh_ac43e43b253b",
  97. "gh_68e7fdc09fe4",
  98. "gh_77f36c109fb1",
  99. "gh_b181786a6c8c",
  100. "gh_1ee2e1b39ccf",
  101. "gh_d3f039c9db2b",
  102. }
  103. # 发文模式
  104. ARTICLES_DAILY = 1
  105. TOULIU = 2
  106. # 默认粉丝
  107. DEFAULT_FANS = 0
  108. # index list
  109. ARTICLE_INDEX_LIST = [1, 2, 3, 4, 5, 6, 7, 8]
  110. # 默认点赞
  111. DEFAULT_LIKE = 0
  112. # 状态
  113. USING_STATUS = 1
  114. NOT_USING_STATUS = 0
  115. # 统计周期
  116. STAT_PERIOD = 30
  117. # default upper_quantile, confidence = 0.95
  118. DEFAULT_UPPER_QUANTILE = 0.975
  119. ACCOUNT_READ_RATE_TABLE = "long_articles_read_rate"
  120. ACCOUNT_READ_AVG_TABLE = "account_avg_info_v3"
  121. class WeixinVideoCrawlerConst:
  122. """
  123. 微信视频抓取常量配置
  124. """
  125. # 账号抓取状态
  126. ACCOUNT_CRAWL_STATUS = 1
  127. ACCOUNT_DO_NOT_CRAWL_STATUS = 0
  128. # 默认最早抓取时间戳(2024-01-01)
  129. DEFAULT_TIMESTAMP = 1704038400
  130. # 搜索爬虫最大页数
  131. MAX_SEARCH_PAGE_NUM = 10
  132. # 抓取每一页的等待时间
  133. SLEEP_SECONDS = 5
  134. # 种子标题最低阅读均值倍数
  135. READ_AVG_MULTIPLE = 1.3
  136. # 种子标题最低阅读量
  137. MIN_READ_COUNT = 2000
  138. # 获取种子标题的统计周期
  139. STAT_PERIOD = 7 * 24 * 60 * 60
  140. # 接口请求成功code
  141. REQUEST_SUCCESS = 0
  142. PUBLISHED_ILLEGAL_TITLE_CODE = 1015
  143. # 是否需要扫描查询源账号
  144. NEED_SCAN_SOURCE_ACCOUNT = 1
  145. DO_NOT_NEED_SOURCE_ACCOUNT = 0
  146. # 视频审核状态长文库
  147. VIDEO_AUDIT_INIT_STATUS = 0
  148. VIDEO_AUDIT_SUCCESS_STATUS = 1
  149. VIDEO_AUDIT_FAIL_STATUS = 2
  150. VIDEO_TITLE_GENERATE_FAIL_STATUS = 4
  151. VIDEO_AUDIT_PROCESSING_STATUS = -1
  152. # 票圈视频审核状态, 1 审核中,2 不通过 3 待修改,4 自己可见 5 通过
  153. PQ_AUDIT_PROCESSING_STATUS = 1
  154. PQ_AUDIT_FAIL_STATUS = 2
  155. PQ_AUDIT_WAIT_STATUS = 3
  156. PQ_AUDIT_SELF_VISIBLE_STATUS = 4
  157. PQ_AUDIT_SUCCESS_STATUS = 5
  158. # 默认账号
  159. DEFAULT_ACCOUNT_UID = 76862180
  160. # 每天发送的审核视频数量
  161. MAX_VIDEO_NUM = 1000
  162. # 单次发布视频审核量
  163. MAX_VIDEO_NUM_PER_PUBLISH = 350
  164. # 标题状态
  165. TITLE_DEFAULT_STATUS = 0
  166. TITLE_EXIT_STATUS = 1
  167. TITLE_FESTIVAL_STATUS = 2
  168. TITLE_SHORT_STATUS = 3
  169. # 标题最短长度
  170. TITLE_MIN_LENGTH = 15
  171. # safe score
  172. TITLE_SAFE_SCORE_THRESHOLD = 7
  173. # Task Status
  174. INIT_STATUS = 0
  175. PROCESSING_STATUS = 1
  176. SUCCESS_STATUS = 2
  177. FAIL_STATUS = 99
  178. class UpdateMiniProgramDetailConst(updatePublishedMsgTaskConst):
  179. """
  180. 更新小程序详情常量配置
  181. """
  182. # 账号联想
  183. class AccountAssociationTaskConst:
  184. """
  185. 账号联想任务常量配置
  186. """
  187. # 获取种子标题的统计周期
  188. STAT_PERIOD = 7 * 24 * 60 * 60
  189. # 阅读均值阈值
  190. READ_AVG_MULTIPLE = 1.3
  191. # 最小阅读量
  192. MIN_READ_COUNT = 2000
  193. # 种子数量限制
  194. SEED_TITLE_LIMIT = 100
  195. # 从aigc获取文章
  196. class ArticleCollectorConst:
  197. """
  198. 文章采集任务常量配置
  199. """
  200. # 发送方式
  201. # 手动推送
  202. MANUAL_PUSH = 1
  203. # 自动群发
  204. BULK_AUTO_PUSH = 2
  205. # 无限流推送
  206. UNLIMITED_PUSH = 3
  207. # 文章状态
  208. # 初始状态
  209. INIT_STATUS = 0
  210. # 成功状态
  211. SUCCESS_STATUS = 1
  212. # 失败状态
  213. FAIL_STATUS = -1
  214. # 发布状态
  215. PUBLISHED_STATUS = 2
  216. # 爬虫接口
  217. ARTICLE_ILLEGAL_CODE = 25012
  218. ARTICLE_DELETE_CODE = 25005
  219. ARTICLE_SUCCESS_CODE = 0
  220. ARTICLE_UNKNOWN_CODE = 10000
  221. class BaiduVideoCrawlerConst:
  222. """
  223. const for baidu video crawler
  224. """
  225. # account status
  226. BAIDU_ACCOUNT_GOOD_STATUS = 1
  227. BAIDU_ACCOUNT_BAD_STATUS = 0
  228. # earliest cursor, 2024-01-01 00:00:00
  229. DEFAULT_CURSOR = 17040384000000
  230. # no source account
  231. NO_SOURCE_ACCOUNT_STATUS = 0
  232. # timestamp To Cursor
  233. TIMESTAMP_TO_CURSOR = 10000
  234. # local path dir
  235. LOCAL_PATH_DIR = "static"
  236. class TitleRewriteTaskConst:
  237. """
  238. title rewrite task const
  239. """
  240. # title rewrite status
  241. TITLE_REWRITE_INIT_STATUS = 0
  242. TITLE_REWRITE_SUCCESS_STATUS = 1
  243. TITLE_REWRITE_FAIL_STATUS = 99
  244. TITLE_REWRITE_LOCK_STATUS = 101
  245. # article status
  246. ARTICLE_AUDIT_PASSED_STATUS = 1
  247. ARTICLE_POSITIVE_STATUS = 0
  248. # title useful status
  249. TITLE_USEFUL_STATUS = 1
  250. # prompt version
  251. PROMPT_VERSION = "xx_250228" # 信欣2025-02-28提供
  252. # block expire time 1h
  253. TITLE_REWRITE_LOCK_TIME = 60 * 60
  254. class ChannelVideoCrawlerConst:
  255. """
  256. const for baidu video crawler
  257. """
  258. # account status
  259. CHANNEL_ACCOUNT_GOOD_STATUS = 1
  260. CHANNEL_ACCOUNT_BAD_STATUS = 0
  261. # earliest cursor, 2024-01-01 00:00:00
  262. DEFAULT_CURSOR = 1704038400
  263. # no source account
  264. NO_SOURCE_ACCOUNT_STATUS = 0
  265. # local path dir
  266. LOCAL_PATH_DIR = "static"
  267. # title length min
  268. MIN_TITLE_LENGTH = 10
  269. # max video length(second)
  270. MAX_VIDEO_LENGTH = 600
  271. # sleep second
  272. SLEEP_SECOND = 2
  273. class ToutiaoVideoCrawlerConst:
  274. """
  275. const for toutiao video crawler
  276. """
  277. # platform
  278. PLATFORM = "toutiao"
  279. # account status
  280. TOUTIAO_ACCOUNT_GOOD_STATUS = 1
  281. TOUTIAO_ACCOUNT_BAD_STATUS = 0
  282. # earliest cursor, 2021-01-01 00:00:00
  283. DEFAULT_CURSOR = 1609430400
  284. # no source account
  285. NO_SOURCE_ACCOUNT_STATUS = 0
  286. # title length min
  287. MIN_TITLE_LENGTH = 10
  288. # max video length(second)
  289. MAX_VIDEO_LENGTH = 600
  290. # sleep second
  291. SLEEP_SECOND = 3
  292. class SohuVideoCrawlerConst:
  293. """
  294. const for sohu video crawler
  295. """
  296. # platform
  297. PLATFORM = "sohu"
  298. # account status
  299. GET_RECOMMEND_INIT_STATUS = 0
  300. GET_RECOMMEND_SUCCESS_STATUS = 1
  301. GET_RECOMMEND_FAIL_STATUS = 99
  302. # title length min
  303. MIN_TITLE_LENGTH = 10
  304. # max video length(second)
  305. MAX_VIDEO_LENGTH = 600
  306. # sleep second
  307. SLEEP_SECOND = 3
  308. # 获取推荐的最低相关性分
  309. GET_RECOMMEND_THRESHOLD_SCORE = 0.6
  310. # 审核状态
  311. AUDIT_SUCCESS_STATUS = 1
  312. # 视频状态
  313. VIDEO_NOT_BAD_STATUS = 0
  314. # PAGE_LIST
  315. PAGE_LIST = [i for i in range(1, 8)]
  316. class SingleVideoPoolPublishTaskConst:
  317. """
  318. const for single video pool publish task
  319. """
  320. TRANSFORM_INIT_STATUS = 0
  321. TRANSFORM_SUCCESS_STATUS = 1
  322. TRANSFORM_FAIL_STATUS = 99
  323. SUCCESS_STATUS = 2
  324. class GoogleVideoUnderstandTaskConst:
  325. # task batch size
  326. BATCH_SIZE = 100
  327. # task status
  328. INIT_STATUS = 0
  329. PROCESSING_STATUS = 1
  330. SUCCESS_STATUS = 2
  331. FAIL_STATUS = 99
  332. # sleep seconds
  333. SLEEP_SECONDS = 60
  334. # max processing time
  335. MAX_PROCESSING_TIME = 3600
  336. # task info
  337. TABLE_NAME = "long_articles_new_video_cover"
  338. TASK_NAME = "extract_video_best_frame_as_cover"
  339. DIR_NAME = "static"
  340. POOL_SIZE = 15
  341. class CategoryGenerationTaskConst:
  342. """
  343. const for category generation task
  344. """
  345. # MAX THREAD
  346. MAX_WORKERS = 10
  347. # task batch size
  348. BATCH_SIZE = 20
  349. # min batch
  350. MIN_BATCH_SIZE = 1
  351. # article status
  352. ARTICLE_GOOD_STATUS = 0
  353. # task status
  354. INIT_STATUS = 0
  355. PROCESSING_STATUS = 1
  356. SUCCESS_STATUS = 2
  357. FAIL_STATUS = 99
  358. # max processing time
  359. MAX_PROCESSING_TIME = 3600
  360. # task info
  361. VIDEO_TABLE_NAME = "publish_single_video_source"
  362. ARTICLE_TABLE_NAME = "crawler_meta_article"
  363. TASK_NAME = "generate_category_with_title"
  364. # article_status
  365. ARTICLE_INIT_STATUS = 1
  366. ARTICLE_PUBLISHED_STATUS = 2
  367. ARTICLE_BAD_STATUS = 0
  368. # limit score
  369. LIMIT_SCORE = 0.4