J1147
/
AiOpsWechat
關注 1
收藏 0
複製 0
程式碼 問題 0 合併請求 0 版本發佈 0 Wiki 活動
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 提交
5 分支
3.7MB
目錄樹: 3abf5a4572
分支列表 標籤
${ item.name }
建立分支 ${ searchTerm }
從 '3abf5a4572'
${ noResults }
AiOpsWechat/tasks.py

1145 line
59KB
原始文件 Blame 歷史記錄 

  1. from celery_app import celery_app

  2. from fastapi import Request,FastAPI

  3. import time,datetime

  4. from celery import Celery

  5. import celery.schedules

  6. from redbeat import RedBeatSchedulerEntry

  7. from datetime import timedelta

  8. 

  9. from services.redis_service import RedisService

  10. from services.kafka_service import KafkaService

  11. from services.gewe_service import GeWeService

  12. # from common.log import logger

  13. from common.utils import *

  14. import asyncio,random

  15. from model.models import AddGroupContactsHistory

  16. import logging

  17. from model.models import AgentConfig

  18. 

  19. import logging

  20. import sys,traceback

  21. 

  22. logger = logging.getLogger('redbeat')

  23. 

  24. 

  25. @celery_app.task(name='tasks.add_task', bind=True, acks_late=True)

  26. def add_task(self, x, y):

  27.     

  28.     time.sleep(5)  # 模拟长时间计算

  29.     logger.info('add')

  30.     return x + y

  31. 

  32. 

  33. @celery_app.task(name='tasks.mul_task', bind=True, acks_late=True)

  34. def mul_task(self, x, y):

  35.     time.sleep(5)  # 模拟长时间计算

  36.     return x * y    

  37. 

  38. 

  39. # @celery.task(name='app.tasks.sync_contacts', bind=True, acks_late=True)

  40. # async def sync_contacts_task(self,app):

  41. #     login_keys = list(await app.state.redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'))

  42. #     return login_keys

  43. #     # for k in login_keys:

  44. #     #     print(k)

  45. 

  46. @celery_app.task(name='tasks.sync_contacts', bind=True, acks_late=True)

  47. async def sync_contacts_task(self, redis_service):

  48.     # Use the redis_service passed as an argument

  49.     login_keys = list(await redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'))

  50.     return login_keys

  51. 

  52. 

  53. @celery_app.task(name='tasks.background_worker_task', bind=True, acks_late=True)

  54. def background_worker_task(self, redis_config, kafka_config, gewe_config):

  55.     async def task():

  56.         redis_service = RedisService()

  57.         await redis_service.init(**redis_config)

  58.         login_keys = []

  59.         async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  60.             login_keys.append(key)

  61.         print(login_keys)

  62.     

  63.     asyncio.run(task())

  64. 

  65. # @celery.task(name='tasks.background_worker_task', bind=True, acks_late=True)

  66. # async def background_worker_task(self, redis_config, kafka_config, gewe_config):

  67. #     # Initialize services inside the task

  68. #     redis_service = RedisService()

  69. #     await redis_service.init(**redis_config)

  70. 

  71. #     login_keys = []

  72. #     async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):  # 使用 async for 遍历异步生成器

  73. #         login_keys.append(key)

  74. 

  75. #     print(login_keys)

  76. 

  77.     # kafka_service = KafkaService(**kafka_config)

  78.     # await kafka_service.start()

  79. 

  80.     # gewe_service = await GeWeService.get_instance(None, gewe_config['api_url'])

  81. 

  82.     # # Task logic

  83.     # lock_name = "background_wxchat_thread_lock"

  84.     # lock_identifier = str(time.time())

  85.     # while True:

  86.     #     if await redis_service.acquire_lock(lock_name, timeout=10):

  87.     #         try:

  88.     #             logger.info("分布式锁已成功获取")

  89.     #             # Perform task logic

  90.     #         finally:

  91.     #             await redis_service.release_lock(lock_name, lock_identifier)

  92.     #             break

  93.     #     else:

  94.     #         logger.info("获取分布式锁失败，等待10秒后重试...")

  95.     #         await asyncio.sleep(10)

  96. 

  97. 

  98. 

  99. # @celery_app.task(name='tasks.scheduled_task', bind=True, acks_late=True)

  100. # def scheduled_task(self):

  101. #     print("定时任务执行成功！~~~~~~~~~~~~~~~~~")

  102. #     return "Hello from Celery Beat + RedBeat!"

  103. 

  104. 

  105. # @celery_app.task(name='tasks.scheduled_task_sync_wx', bind=True, acks_late=True)

  106. # def scheduled_task_sync_wx(self,redis_service,kafka_service,gewe_service):

  107. #     print("scheduled_task_sync_wx 定时任务执行成功！")

  108. #     return "Hello from Celery Beat + RedBeat!"

  109. 

  110. # @celery_app.task(name='tasks.scheduled_task_sync_wx_info_1', bind=True, acks_late=True)

  111. # def scheduled_task_sync_wx_info_1(self,redis_config, kafka_config, gewe_config):

  112. #     '''

  113. #     定时获取微信号资料

  114. #     '''

  115. #     loop = asyncio.new_event_loop()

  116. #     asyncio.set_event_loop(loop)

  117. #     async def task():

  118. #         try:

  119. #             redis_service = RedisService()

  120. #             await redis_service.init(**redis_config)

  121. #             # gewe_service = await GeWeService.get_instance(None, gewe_config['api_url'])

  122. #             login_keys = []

  123. #             async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  124. #                 login_keys.append(key)

  125. 

  126. #             print(login_keys)

  127. #             # for k in login_keys:

  128. #             #     r = await redis_service.get_hash(k)

  129. #             #     app_id = r.get("appId")

  130. #             #     token_id = r.get("tokenId")

  131. #             #     wxid = r.get("wxid")

  132. #             #     status = r.get('status')

  133. #             #     if status == '0':

  134. #             #         continue

  135. #             #     ret, msg, profile = await gewe_service.get_profile_async(token_id, app_id)

  136. #             #     if ret != 200:

  137. #             #         logger.warning(f"同步微信号 {wxid} 资料失败: {ret}-{msg}")

  138. #             #         continue

  139. #             #     nickname=profile.get("nickName")

  140. #             #     head_img_url=profile.get("smallHeadImgUrl")

  141. #             #     r.update({"nickName":nickname,"headImgUrl":head_img_url,"modify_at":int(time.time())})

  142. #             #     cleaned_login_info = {k: (v if v is not None else '') for k, v in r.items()}

  143. #             #     await redis_service.set_hash(k, cleaned_login_info)

  144. #             #     logger.info(f"同步微信号 {wxid} 资料 成功")

  145. #                 # redis_service.update_hash_field(k,"nickName",nickname)

  146. #                 # redis_service.update_hash_field(k,"headImgUrl",head_img_url)

  147. #                 # redis_service.update_hash_field(k,"modify_at",int(time.time()))

  148. #         except Exception as e:

  149. #             logger.error(f"任务执行过程中发生异常: {e}")

  150. 

  151. #         print("scheduled_task_sync_wx_info 定时任务执行成功！")

  152. #         return "Hello from Celery Beat + RedBeat!"

  153.     

  154. #     loop.run_until_complete(task())

  155. #     loop.close()

  156. 

  157. 

  158. 

  159. # @celery_app.task(name='tasks.scheduled_task_sync_wx_info', bind=True, acks_late=True)

  160. # def scheduled_task_sync_wx_info(self, redis_config, kafka_config, gewe_config):

  161. #     '''

  162. #     定时获取微信号资料

  163. #     '''

  164. #     async def task():

  165. #         try:

  166. #             redis_service = RedisService()

  167. #             await redis_service.init(**redis_config)

  168. #             gewe_service = await GeWeService.get_instance(redis_service,gewe_config['api_url'])

  169. 

  170. #             login_keys = []

  171. #             async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  172. #                 login_keys.append(key)

  173. 

  174. #             #print(login_keys)

  175. #             for k in login_keys:

  176. #                 r = await redis_service.get_hash(k)

  177. #                 app_id = r.get("appId")

  178. #                 token_id = r.get("tokenId")

  179. #                 wxid = r.get("wxid")

  180. #                 status = r.get('status')

  181. #                 if status == '0':

  182. #                     logger.warning(f"微信号 {wxid} 已经离线")

  183. #                     continue

  184. #                 ret, msg, profile = await gewe_service.get_profile_async(token_id, app_id)

  185. #                 if ret != 200:

  186. #                     logger.warning(f"同步微信号 {wxid} 资料失败: {ret}-{msg}")

  187. #                     continue

  188. #                 nickname=profile.get("nickName")

  189. #                 head_img_url=profile.get("smallHeadImgUrl")

  190. #                 # print(nickname)

  191. 

  192. #                 nickname=profile.get("nickName")

  193. #                 head_img_url=profile.get("smallHeadImgUrl")

  194. #                 r.update({"nickName":nickname,"headImgUrl":head_img_url,"modify_at":int(time.time())})

  195. #                 cleaned_login_info = {k: (v if v is not None else '') for k, v in r.items()}

  196. #                 await redis_service.set_hash(k, cleaned_login_info)

  197. #                 logger.info(f"定时同步微信号{wxid}-昵称{nickname} 资料成功")

  198. 

  199. #         except Exception as e:

  200. #             logger.error(f"任务执行过程中发生异常: {e}")

  201. 

  202. #     loop = asyncio.get_event_loop()

  203. #     if loop.is_closed():

  204. #         loop = asyncio.new_event_loop()

  205. #         asyncio.set_event_loop(loop)

  206. 

  207. #     loop.run_until_complete(task())  # 在现有事件循环中运行任务

  208. 

  209. 

  210. @celery_app.task(name='tasks.scheduled_task_sync_wx_info', bind=True, acks_late=True)

  211. def scheduled_task_sync_wx_info(self, redis_config, kafka_config, gewe_config):

  212.     '''

  213.     定时获取微信号资料

  214.     '''

  215.     async def process_key(redis_service, gewe_service, semaphore, key):

  216.         async with semaphore:  # 使用 Semaphore 控制并发

  217.             try:

  218.                 r = await redis_service.get_hash(key)

  219.                 app_id = r.get("appId")

  220.                 token_id = r.get("tokenId")

  221.                 wxid = r.get("wxid")

  222.                 status = r.get('status')

  223.                 if status == '0':

  224.                     logger.warning(f"微信号 {wxid} 已经离线")

  225.                     return

  226.                 ret, msg, profile = await gewe_service.get_profile_async(token_id, app_id)

  227.                 if ret != 200:

  228.                     logger.warning(f"同步微信号 {wxid} 资料失败: {ret}-{msg}")

  229.                     return

  230.                 nickname = profile.get("nickName")

  231.                 head_img_url = profile.get("smallHeadImgUrl")

  232.                 r.update({"nickName": nickname, "headImgUrl": head_img_url, "modify_at": int(time.time())})

  233.                 cleaned_login_info = {k: (v if v is not None else '') for k, v in r.items()}

  234.                 await redis_service.set_hash(key, cleaned_login_info)

  235.                 logger.info(f"定时同步微信号{wxid}-昵称{nickname} 资料成功")

  236.             except Exception as e:

  237.                 logger.error(f"处理键 {key} 时发生异常: {e}")

  238. 

  239.     async def task():

  240.         try:

  241.             redis_service = RedisService()

  242.             await redis_service.init(**redis_config)

  243.             gewe_service = await GeWeService.get_instance(redis_service, gewe_config['api_url'])

  244.             login_keys = []

  245.             async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  246.                 login_keys.append(key)

  247. 

  248.             # 设置 Semaphore，限制并发数为 10

  249.             semaphore = asyncio.Semaphore(10)

  250.             # 使用 asyncio.gather 并发处理所有键

  251.             await asyncio.gather(*[process_key(redis_service, gewe_service, semaphore, key) for key in login_keys])

  252.         except Exception as e:

  253.             logger.error(f"任务执行过程中发生异常: {e}")

  254. 

  255.     loop = asyncio.get_event_loop()

  256.     if loop.is_closed():

  257.         loop = asyncio.new_event_loop()

  258.         asyncio.set_event_loop(loop)

  259.     loop.run_until_complete(task())  # 在现有事件循环中运行任务

  260. 

  261. 

  262. 

  263. 

  264. # @celery_app.task(name='tasks.scheduled_task_add_contacts_from_chatrooms', bind=True, acks_late=True)

  265. # def scheduled_task_add_contacts_from_chatrooms(self, redis_config, kafka_config, gewe_config):

  266. 

  267. #     '''

  268. #     关于群加好友的请求规则: 每个智能体一次最多30人,间隔2小时做1次,即最多90人/天。

  269. #     加好友规则:每天处理次数、间隔时间(分钟)、每次加好友人数这3个参数都可以设置。目前默认只是上面的设置。

  270. #     '''

  271. #     async def task():

  272. #         try:

  273. #             now = datetime.now()

  274. #             if now.hour < 8:

  275. #                 logger.info(f"定时群成员定时添好友任务不启动,当前时间为 {now.strftime('%Y-%m-%d %H:%M:%S')}，早于8点")

  276. #                 return

  277. 

  278. #             logger.info('定时群成员定时添好友任务开始')

  279. #             redis_service = RedisService()

  280. #             await redis_service.init(**redis_config)

  281. 

  282. #             gewe_service = await GeWeService.get_instance(redis_service,gewe_config['api_url'])

  283. 

  284. #             KAFKA_BOOTSTRAP_SERVERS=kafka_config['bootstrap_servers']

  285. #             KAFKA_TOPIC=kafka_config['topic']

  286. #             KAFKA_GROUP_ID=kafka_config['group_id']

  287.             

  288. #             kafka_service= KafkaService(KAFKA_BOOTSTRAP_SERVERS, KAFKA_TOPIC, KAFKA_TOPIC,KAFKA_GROUP_ID)

  289. #             await kafka_service.start_producer()

  290. 

  291. #             global_config=await gewe_service.get_global_config_from_cache_async()

  292. #             scheduled_task_add_contacts_from_chatrooms_config=global_config.get('scheduledTaskAddContactsFromChatrooms',{})

  293.             

  294. #             oneday_add_contacts_total=90

  295. #             once_add_contacts_total=30

  296. #             #oneday_times=3

  297. 

  298. #             if scheduled_task_add_contacts_from_chatrooms_config:

  299. #                 oneday_add_contacts_total=scheduled_task_add_contacts_from_chatrooms_config.get('oneDayAddContactsTotal',90)

  300. #                 once_add_contacts_total=scheduled_task_add_contacts_from_chatrooms_config.get('onceAddContactsTotal',30)

  301. #                 #oneday_times=scheduled_task_add_contacts_from_chatrooms_config.get('oneDayTimes',3)

  302. 

  303. #             # cache_task_run_time_logs= await gewe_service.get_task_run_time_async('scheduled_task_add_contacts_from_chatrooms')

  304. #             # if cache_task_run_time_logs:

  305. #             #     sorted_tasks = sorted(cache_task_run_time_logs, key=lambda x: x.get("runTime"), reverse=True)

  306. #             #     last_run_time=sorted_tasks[0].get("runTime")

  307. 

  308. #             #     if last_run_time > 1e12:  # 毫秒级时间戳

  309. #             #         last_run_time = last_run_time / 1000  # 转换为秒

  310.                 

  311. #             #     # 将时间戳转换为 datetime 对象

  312. #             #     last_run_time = datetime.fromtimestamp(last_run_time)

  313.                 

  314. #             #     # 获取当前时间

  315. #             #     current_time = datetime.now()

  316.                 

  317. #             #     # 计算时间差

  318. #             #     time_difference = current_time - last_run_time

  319.                 

  320. #             #     # 判断是否相差2小时

  321. #             #     if time_difference < timedelta(hours=2):

  322. #             #         logger.info(f"上次定时群成员定时添好友任务在2小时内，不再执行")

  323. #             #         return 

  324. 

  325. #             # time_difference_seconds = today_seconds_remaining()

  326. #             # cache_task_run_time_logs.append({"runTime":int(time.time())})

  327. #             # await gewe_service.save_task_run_time_async('scheduled_task_add_contacts_from_chatrooms',cache_task_run_time_logs,time_difference_seconds)

  328.             

  329. #             login_keys = []

  330. #             async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  331. #                 login_keys.append(key)

  332. 

  333. #             wixd_add_contacts_from_chatrooms_times = {}

  334. 

  335. #             for k in login_keys:

  336. #                 r = await redis_service.get_hash(k)

  337. #                 app_id = r.get("appId")

  338. #                 token_id = r.get("tokenId")

  339. #                 wxid = r.get("wxid")

  340. #                 status = r.get('status')

  341. #                 if status == '0':

  342. #                     logger.warning(f"微信号 {wxid} 已经离线，群成员不能定时添加")

  343. #                     continue

  344. 

  345. #                 config=await gewe_service.get_wxchat_config_from_cache_async(wxid)

  346. #                 validated_config = AgentConfig.model_validate(config)

  347. #                 if not validated_config.agentEnabled:

  348. #                     logger.warning(f"微信号 {wxid} 取消了托管，群成员不能定时添加")

  349. #                     continue

  350. 

  351. 

  352. #                 # 判断是否过于频繁

  353. #                 is_wx_expection = await gewe_service.get_wx_expection_async(wxid,"addGroupMemberAsFriend")

  354. #                 if is_wx_expection:

  355. #                     logger.info(f"{wxid} 本次任务接口addGroupMemberAsFriend异常, {is_wx_expection}，本次群好友邀请任务未开始,跳过任务。")   

  356. #                     continue

  357.                 

  358. 

  359. #                 cache_task_run_time_wxid_logs= await gewe_service.get_task_run_time_by_wxid_async(wxid,'scheduled_task_add_contacts_from_chatrooms')

  360. #                 if cache_task_run_time_wxid_logs:

  361. #                     sorted_tasks = sorted(cache_task_run_time_wxid_logs, key=lambda x: x.get("runTime"), reverse=True)

  362. #                     last_run_time=sorted_tasks[0].get("runTime")

  363. 

  364. #                     if last_run_time > 1e12:  # 毫秒级时间戳

  365. #                         last_run_time = last_run_time / 1000  # 转换为秒

  366.                     

  367. #                     # 将时间戳转换为 datetime 对象

  368. #                     last_run_time = datetime.fromtimestamp(last_run_time)

  369.                     

  370. #                     # 获取当前时间

  371. #                     current_time = datetime.now()

  372.                     

  373. #                     # 计算时间差

  374. #                     time_difference = current_time - last_run_time

  375.                     

  376. #                     # 判断是否相差2小时

  377. #                     if time_difference < timedelta(hours=2):

  378. #                         logger.info(f"{wxid}上次定时群成员定时添好友任务在2小时内，不再执行")

  379. #                         continue 

  380. 

  381. 

  382. #                 cache_task_run_time_wxid_logs.append({"runTime":int(time.time())})

  383. #                 await gewe_service.save_task_run_time_by_wxid_async(wxid,'scheduled_task_add_contacts_from_chatrooms',cache_task_run_time_wxid_logs,3600*2)

  384.                 

  385. 

  386. 

  387. #                 c = await gewe_service.get_wxchat_config_from_cache_async(wxid)

  388. #                 contacts = await gewe_service.get_contacts_brief_from_cache_async(wxid)

  389.             

  390. #                 contact_wxids = [c.get('userName') for c in contacts]

  391. #                 chatrooms = c.get('addContactsFromChatroomIdWhiteList', [])

  392. #                 logger.info(f'{wxid} 定时群成员定时添好友任务开始')

  393. #                 wixd_add_contacts_from_chatrooms_times[wxid] = 0

  394. #                 for chatroom_id in chatrooms:

  395. #                     chatroom = await gewe_service.get_group_info_from_cache_async(wxid, chatroom_id)

  396. #                     chatroom_member=await gewe_service.get_group_members_from_cache_async(wxid, chatroom_id)

  397.                     

  398. #                     chatroom_nickname = chatroom.get('nickName')

  399. #                     chatroom_owner_wxid = chatroom_member.get('chatroomOwner', None)

  400. 

  401. #                     admin_wxids = chatroom_member.get('adminWxid', []) 

  402. 

  403. #                     admin_wxids = chatroom_member.get('adminWxid')

  404. #                     if admin_wxids is None:

  405. #                         admin_wxids = []  # 如果 admin_wxids 是 None，将其初始化为空列表

  406.                     

  407. #                     logger.info(f'{chatroom_nickname} 的群主是 {chatroom_owner_wxid}，管理员是{admin_wxids}')

  408. #                     contact_wxids_set = set(contact_wxids)

  409. #                     # for admin_wxid in admin_wxids:

  410. #                     #     contact_wxids_set.add(admin_wxid)

  411. #                     if admin_wxids:

  412. #                         contact_wxids_set.update(set(admin_wxids))

  413. #                     if chatroom_owner_wxid is not None:

  414. #                         contact_wxids_set.add(chatroom_owner_wxid)

  415.                     

  416. #                     contact_wxids_set.add(wxid)

  417. 

  418. #                     # unavailable_wixds=await gewe_service.check_wixd_group_add_contacts_history_async(wxid,chatroom_id)

  419. #                     # contact_wxids_set.update(set(unavailable_wixds))

  420. 

  421. #                     # chatroom_member_list = chatroom.get('memberList', [])

  422. 

  423. 

  424. #                     unavailable_wixds=await gewe_service.check_wixd_group_add_contacts_history_async(wxid,chatroom_id)

  425. #                     if unavailable_wixds:

  426. #                         contact_wxids_set.update(set(unavailable_wixds))

  427.                         

  428. #                     chatroom_member_list = chatroom.get('memberList', [])

  429. #                     if chatroom_member_list is None:

  430. #                         chatroom_member_list = []  # 如果 memberList 是 None，将其初始化为空列表

  431. #                     elif not isinstance(chatroom_member_list, list):

  432. #                         chatroom_member_list = list(chatroom_member_list)  # 如果 memberList 不是列表，将其转换为列表                    

  433. 

  434. 

  435. #                     remaining_chatroot_members = [x for x in chatroom_member_list if x.get('wxid') not in contact_wxids_set]

  436.                 

  437. #                     nickname = next((member['nickName'] for member in chatroom_member_list if member['wxid'] == wxid), None)

  438. 

  439. #                     if not remaining_chatroot_members:

  440. #                         logger.info(f'{nickname}-{wxid} 在 {chatroom_nickname} 群里没有好友可以邀请')

  441. #                         # 任务状态推送到kafka 

  442. #                         k_message=wx_add_contacts_from_chatroom_task_status_message(wxid,chatroom_id,2)

  443. #                         await kafka_service.send_message_async(k_message)

  444. #                         continue

  445. 

  446. #                     logger.info(f'{nickname}-{wxid} 在 {chatroom_nickname} 群里还可以邀请的好友有：{[x.get("nickName") for x in remaining_chatroot_members]}')

  447. #                     for m in remaining_chatroot_members:

  448. #                         # 判断本次任务是否已经邀请了30个好友

  449. #                         if wixd_add_contacts_from_chatrooms_times[wxid] == once_add_contacts_total:

  450. #                             logger.info(f"{wxid} 本次任务已经邀请了{once_add_contacts_total}人,不再邀请")

  451. #                             continue 

  452. #                         # 判断当天群成员是否已经加了90个好友

  453. #                         is_add_group_times = await gewe_service.is_group_add_contacts_history_one_day_async(wxid,oneday_add_contacts_total)

  454. #                         if is_add_group_times:

  455. #                             logger.info(f"当天 {wxid} 所有群的成员已经加了{oneday_add_contacts_total}个好友，不再添加")

  456. #                             continue

  457.                         

  458. #                         # 判断是否过于频繁

  459. #                         is_wx_expection = await gewe_service.get_wx_expection_async(wxid,"addGroupMemberAsFriend")

  460. #                         if is_wx_expection:

  461. #                             logger.info(f"{wxid} 本次任务接口addGroupMemberAsFriend异常,不再邀请,{is_wx_expection}")   

  462. #                             continue

  463. 

  464. #                         contact_wxid= m.get('wxid')

  465. #                         member_nickname=m.get("nickName")

  466. #                         group_add_contacts_history = await gewe_service.get_group_add_contacts_history_async(wxid,chatroom_id,contact_wxid)

  467.                         

  468. #                         if group_add_contacts_history:

  469. #                             sorted_history = sorted(group_add_contacts_history, key=lambda x: x.addTime, reverse=True)

  470.                             

  471. #                             # 已经邀请过两次，不再邀请

  472. #                             if len(sorted_history)==2:

  473. #                                 logger.info(f'{nickname}-{wxid}在{chatroom_nickname}-{chatroom_id} 群的 {member_nickname}-{contact_wxid} 已经邀请过2次，不再邀请')

  474. #                                 continue

  475. 

  476. #                             # 当天邀请过，不再邀请

  477. #                             if len(sorted_history) > 0:

  478. #                                 last_add_time = sorted_history[0].addTime

  479. #                                 def is_add_time_more_than_one_day(addTime: int) -> bool:

  480. #                                     """

  481. #                                     判断 addTime 是否与当前时间相隔大于 3600 × 24 秒

  482. #                                     :param addTime: Unix 时间戳

  483. #                                     :return: 如果 addTime 与当前时间相隔大于 3600 × 24 秒，返回 True；否则返回 False

  484. #                                     """

  485. #                                     # 获取当前时间的时间戳

  486. #                                     current_time = time.time()

  487.                                     

  488. #                                     # 计算时间戳差值

  489. #                                     time_difference = abs(current_time - addTime)

  490.                                     

  491. #                                     # 检查是否大于 3600 × 24 秒

  492. #                                     return time_difference > 3600 * 24

  493.                             

  494. #                                 is_more_than_one_day= is_add_time_more_than_one_day(last_add_time)

  495. 

  496. #                                 if not is_more_than_one_day:

  497. #                                     logger.info(f'{nickname}-{wxid}在{chatroom_nickname}-{chatroom_id} 群的 {member_nickname}-{contact_wxid} 已经当天邀请，不再邀请')

  498. #                                     continue

  499. 

  500. #                         ret, msg, data = await gewe_service.add_group_member_as_friend_async(token_id, app_id, chatroom_id, m.get('wxid'), f'我是群聊"{chatroom_nickname}"群的{nickname}')

  501. #                         if ret!=200:

  502. #                             logger.warning(f'群好友邀请失败原因：{ret} {data}')

  503. #                             if msg in '操作过于频繁，请稍后再试。':

  504. #                                 await gewe_service.save_wx_expection_async(wxid,"addGroupMemberAsFriend",msg,today_seconds_remaining())

  505. #                                 logger.warning(f'{nickname}-{wxid} 操作过于频繁，本次群好友邀请任务未完成跳过。')

  506. #                                 continue 

  507. 

  508. #                         history=AddGroupContactsHistory.model_validate({

  509. #                             "chatroomId":chatroom_id,

  510. #                             "wxid":wxid,

  511. #                             "contactWixd":contact_wxid,

  512. #                             "addTime":int(time.time())

  513. #                         })

  514. #                         await gewe_service.save_group_add_contacts_history_async(wxid,chatroom_id,contact_wxid,history)

  515. #                         wixd_add_contacts_from_chatrooms_times[wxid]+=1

  516. #                         logger.info(f'{nickname} 向 {chatroom_nickname}-{chatroom_id} 群的  {m.get("nickName")}-{m.get("wxid")} 发送好友邀请  {msg}')

  517. #                         # 推送到kafka

  518. 

  519. #                         k_message = wx_add_contacts_from_chatroom_message(history.wxid,history.chatroomId,history.contactWixd,history.addTime)

  520. #                         await kafka_service.send_message_async(k_message)

  521. #                         #await asyncio.sleep(random.uniform(1.5, 3)) 

  522. #                         await asyncio.sleep(random.uniform(30,60)) 

  523. 

  524. #                     # 任务状态推送到kafka 

  525. #                     task_status=await gewe_service.wx_add_contacts_from_chatroom_task_status_async(wxid,chatroom_id)

  526. #                     k_message=wx_add_contacts_from_chatroom_task_status_message(wxid,chatroom_id,task_status)

  527. #                     await kafka_service.send_message_async(k_message)

  528. #                     # 下一个群

  529. #                     await asyncio.sleep(random.uniform(1.5, 3)) 

  530. #         except Exception as e:

  531. #             # 获取当前的堆栈跟踪

  532. #             tb = sys.exc_info()[2]

  533. #             # 为异常附加堆栈跟踪

  534. #             e = e.with_traceback(tb)

  535. #             # 输出详细的错误信息

  536. #             logger.error(f"任务执行过程中发生异常: {e}\n异常类型: {type(e).__name__}\n异常信息: {str(e)}\n堆栈跟踪: {traceback.format_exc()}")

  537. #         finally:

  538. #             await kafka_service.stop_producer()

  539.     

  540. #     loop = asyncio.get_event_loop()

  541. #     if loop.is_closed():

  542. #         loop = asyncio.new_event_loop()

  543. #         asyncio.set_event_loop(loop)

  544. 

  545. #     loop.run_until_complete(task())  # 在现有事件循环中运行任务

  546. 

  547. 

  548. @celery_app.task(name='tasks.scheduled_task_add_contacts_from_chatrooms', bind=True, acks_late=True)

  549. def scheduled_task_add_contacts_from_chatrooms(self, redis_config, kafka_config, gewe_config):

  550. 

  551.     '''

  552.     关于群加好友的请求规则: 每个智能体一次最多30人,间隔2小时做1次,即最多90人/天。

  553.     加好友规则:每天处理次数、间隔时间(分钟)、每次加好友人数这3个参数都可以设置。目前默认只是上面的设置。

  554.     '''

  555.     async def process_login_key(redis_service:RedisService, gewe_service: GeWeService, kafka_service:KafkaService, k, gewe_config, oneday_add_contacts_total, once_add_contacts_total, semaphore):

  556.         async with semaphore:  # 使用 Semaphore 控制并发

  557.             r = await redis_service.get_hash(k)

  558.             app_id = r.get("appId")

  559.             token_id = r.get("tokenId")

  560.             wxid = r.get("wxid")

  561.             status = r.get('status')

  562. 

  563.             if status == '0':

  564.                 logger.warning(f"微信号 {wxid} 已经离线，群成员不能定时添加")

  565.                 return

  566. 

  567.             config = await gewe_service.get_wxchat_config_from_cache_async(wxid)

  568.             validated_config = AgentConfig.model_validate(config)

  569.             if not validated_config.agentEnabled:

  570.                 logger.warning(f"微信号 {wxid} 取消了托管，群成员不能定时添加")

  571.                 return

  572. 

  573.             # 判断是否过于频繁

  574.             is_wx_expection = await gewe_service.get_wx_expection_async(wxid, "addGroupMemberAsFriend")

  575.             if is_wx_expection:

  576.                 logger.info(f"{wxid} 本次任务接口addGroupMemberAsFriend异常, {is_wx_expection}，本次群好友邀请任务未开始,跳过任务。")

  577.                 return

  578. 

  579.             cache_task_run_time_wxid_logs = await gewe_service.get_task_run_time_by_wxid_async(wxid, 'scheduled_task_add_contacts_from_chatrooms')

  580.             if cache_task_run_time_wxid_logs:

  581.                 sorted_tasks = sorted(cache_task_run_time_wxid_logs, key=lambda x: x.get("runTime"), reverse=True)

  582.                 last_run_time = sorted_tasks[0].get("runTime")

  583.                 if last_run_time > 1e12:  # 毫秒级时间戳

  584.                     last_run_time = last_run_time / 1000  # 转换为秒

  585. 

  586.                 # 将时间戳转换为 datetime 对象

  587.                 last_run_time = datetime.fromtimestamp(last_run_time)

  588. 

  589.                 # 获取当前时间

  590.                 current_time = datetime.now()

  591. 

  592.                 # 计算时间差

  593.                 time_difference = current_time - last_run_time

  594. 

  595.                 # 判断是否相差2小时

  596.                 if time_difference < timedelta(hours=2):

  597.                     logger.info(f"{wxid}上次定时群成员定时添好友任务在2小时内，不再执行")

  598.                     return

  599. 

  600.             cache_task_run_time_wxid_logs.append({"runTime": int(time.time())})

  601.             await gewe_service.save_task_run_time_by_wxid_async(wxid, 'scheduled_task_add_contacts_from_chatrooms', cache_task_run_time_wxid_logs, 3600 * 2)

  602.             c:dict = await gewe_service.get_wxchat_config_from_cache_async(wxid)

  603.             contacts = await gewe_service.get_contacts_brief_from_cache_async(wxid)

  604. 

  605.             contact_wxids = [c.get('userName') for c in contacts]

  606.             chatrooms = c.get('addContactsFromChatroomIdWhiteList', [])

  607.             logger.info(f'{wxid} 定时群成员定时添好友任务开始')

  608.             wixd_add_contacts_from_chatrooms_times = {wxid: 0}

  609.             for chatroom_id in chatrooms:

  610.                 chatroom = await gewe_service.get_group_info_from_cache_async(wxid, chatroom_id)

  611.                 chatroom_member = await gewe_service.get_group_members_from_cache_async(wxid, chatroom_id)

  612. 

  613.                 chatroom_nickname = chatroom.get('nickName')

  614.                 chatroom_owner_wxid = chatroom_member.get('chatroomOwner', None)

  615.                 admin_wxids = chatroom_member.get('adminWxid', [])

  616.                 admin_wxids = chatroom_member.get('adminWxid')

  617.                 if admin_wxids is None:

  618.                     admin_wxids = []  # 如果 admin_wxids 是 None，将其初始化为空列表

  619. 

  620.                 logger.info(f'{chatroom_nickname} 的群主是 {chatroom_owner_wxid}，管理员是{admin_wxids}')

  621.                 contact_wxids_set = set(contact_wxids)

  622.                 if admin_wxids:

  623.                     contact_wxids_set.update(set(admin_wxids))

  624.                 if chatroom_owner_wxid is not None:

  625.                     contact_wxids_set.add(chatroom_owner_wxid)

  626. 

  627.                 contact_wxids_set.add(wxid)

  628.                 unavailable_wixds = await gewe_service.check_wixd_group_add_contacts_history_async(wxid, chatroom_id)

  629.                 if unavailable_wixds:

  630.                     contact_wxids_set.update(set(unavailable_wixds))

  631. 

  632.                 chatroom_member_list = chatroom.get('memberList', [])

  633.                 if chatroom_member_list is None:

  634.                     chatroom_member_list = []  # 如果 memberList 是 None，将其初始化为空列表

  635.                 elif not isinstance(chatroom_member_list, list):

  636.                     chatroom_member_list = list(chatroom_member_list)  # 如果 memberList 不是列表，将其转换为列表

  637. 

  638.                 remaining_chatroot_members = [x for x in chatroom_member_list if x.get('wxid') not in contact_wxids_set]

  639. 

  640.                 nickname = next((member['nickName'] for member in chatroom_member_list if member['wxid'] == wxid), None)

  641.                 if not remaining_chatroot_members:

  642.                     logger.info(f'{nickname}-{wxid} 在 {chatroom_nickname} 群里没有好友可以邀请')

  643.                     # 任务状态推送到kafka

  644.                     k_message = wx_add_contacts_from_chatroom_task_status_message(wxid, chatroom_id, 2)

  645.                     await kafka_service.send_message_async(k_message)

  646.                     continue 

  647.                 

  648.                 logger.info(f'{nickname}-{wxid} 在 {chatroom_nickname} 群里还可以邀请的好友有：{[x.get("nickName") for x in remaining_chatroot_members]}')

  649.                 for m in remaining_chatroot_members:

  650.                     # 判断本次任务是否已经邀请了30个好友

  651.                     if wixd_add_contacts_from_chatrooms_times[wxid] == once_add_contacts_total:

  652.                         logger.info(f"{wxid} 本次任务已经邀请了{once_add_contacts_total}人,不再邀请")

  653.                         return

  654. 

  655.                     # 判断当天群成员是否已经加了90个好友

  656.                     is_add_group_times = await gewe_service.is_group_add_contacts_history_one_day_async(wxid, oneday_add_contacts_total)

  657.                     if is_add_group_times:

  658.                         logger.info(f"当天 {wxid} 所有群的成员已经加了{oneday_add_contacts_total}个好友，不再添加")

  659.                         return

  660. 

  661.                     # 判断是否过于频繁

  662.                     is_wx_expection = await gewe_service.get_wx_expection_async(wxid, "addGroupMemberAsFriend")

  663.                     if is_wx_expection:

  664.                         logger.info(f"{wxid} 本次任务接口addGroupMemberAsFriend异常,不再邀请,{is_wx_expection}")

  665.                         return

  666.                     

  667.                     contact_wxid = m.get('wxid')

  668.                     member_nickname = m.get("nickName")

  669.                     group_add_contacts_history = await gewe_service.get_group_add_contacts_history_async(wxid, chatroom_id, contact_wxid)

  670. 

  671.                     if group_add_contacts_history:

  672.                         sorted_history = sorted(group_add_contacts_history, key=lambda x: x.addTime, reverse=True)

  673. 

  674.                         # 已经邀请过两次，不再邀请

  675.                         if len(sorted_history) == 2:

  676.                             logger.info(f'{nickname}-{wxid}在{chatroom_nickname}-{chatroom_id} 群的 {member_nickname}-{contact_wxid} 已经邀请过2次，不再邀请')

  677.                             continue

  678.                         

  679.                         # 当天邀请过，不再邀请

  680.                         if len(sorted_history) > 0:

  681.                             last_add_time = sorted_history[0].addTime

  682. 

  683.                             def is_add_time_more_than_one_day(addTime: int) -> bool:

  684.                                 """

  685.                                 判断 addTime 是否与当前时间相隔大于 3600 × 24 秒

  686.                                 :param addTime: Unix 时间戳

  687.                                 :return: 如果 addTime 与当前时间相隔大于 3600 × 24 秒，返回 True；否则返回 False

  688.                                 """

  689.                                 # 获取当前时间的时间戳

  690.                                 current_time = time.time()

  691. 

  692.                                 # 计算时间戳差值

  693.                                 time_difference = abs(current_time - addTime)

  694. 

  695.                                 # 检查是否大于 3600 × 24 秒

  696.                                 return time_difference > 3600 * 24

  697. 

  698.                             is_more_than_one_day = is_add_time_more_than_one_day(last_add_time)

  699.                             if not is_more_than_one_day:

  700.                                 logger.info(f'{nickname}-{wxid}在{chatroom_nickname}-{chatroom_id} 群的 {member_nickname}-{contact_wxid} 已经当天邀请，不再邀请')

  701.                                 continue 

  702. 

  703.                     ret, msg, data = await gewe_service.add_group_member_as_friend_async(token_id, app_id, chatroom_id, m.get('wxid'),

  704.                                                                                         f'我是群聊"{chatroom_nickname}"群的{nickname}')

  705.                     if ret != 200:

  706.                         logger.warning(f'群好友邀请失败原因：{ret} {msg} {data}')

  707.                         if '操作过于频繁' in data.get('msg'):

  708.                             await gewe_service.save_wx_expection_async(wxid, "addGroupMemberAsFriend", msg, today_seconds_remaining())

  709.                             logger.warning(f'{nickname}-{wxid} 操作过于频繁，本次群好友邀请任务未完成跳过。')

  710.                             return 

  711.                         

  712.                     history = AddGroupContactsHistory.model_validate({

  713.                         "chatroomId": chatroom_id,

  714.                         "wxid": wxid,

  715.                         "contactWixd": contact_wxid,

  716.                         "addTime": int(time.time())

  717.                     })

  718.                     await gewe_service.save_group_add_contacts_history_async(wxid, chatroom_id, contact_wxid, history)

  719.                     wixd_add_contacts_from_chatrooms_times[wxid] += 1

  720.                     logger.info(f'{nickname} 向 {chatroom_nickname}-{chatroom_id} 群的  {m.get("nickName")}-{m.get("wxid")} 发送好友邀请  {msg}')

  721.                     # 推送到kafka

  722.                     k_message = wx_add_contacts_from_chatroom_message(history.wxid, history.chatroomId, history.contactWixd, history.addTime)

  723.                     await kafka_service.send_message_async(k_message)

  724.                     # await asyncio.sleep(random.uniform(1.5, 3))

  725.                     await asyncio.sleep(random.uniform(30, 60))

  726.                 # 任务状态推送到kafka

  727.                 task_status = await gewe_service.wx_add_contacts_from_chatroom_task_status_async(wxid, chatroom_id)

  728.                 k_message=wx_add_contacts_from_chatroom_task_status_message(wxid, chatroom_id, task_status)

  729.                 await kafka_service.send_message_async(k_message)

  730.                 # 下一个群

  731.                 await asyncio.sleep(random.uniform(1.5, 3))

  732. 

  733.     async def task():

  734.         try:

  735.             now = datetime.now()

  736.             if now.hour < 8:

  737.                 logger.info(f"定时群成员定时添好友任务不启动,当前时间为 {now.strftime('%Y-%m-%d %H:%M:%S')}，早于8点")

  738.                 return

  739. 

  740.             logger.info('定时群成员定时添好友任务开始')

  741.             redis_service = RedisService()

  742.             await redis_service.init(**redis_config)

  743.             gewe_service = await GeWeService.get_instance(redis_service, gewe_config['api_url'])

  744.             KAFKA_BOOTSTRAP_SERVERS = kafka_config['bootstrap_servers']

  745.             KAFKA_TOPIC = kafka_config['topic']

  746.             KAFKA_GROUP_ID = kafka_config['group_id']

  747. 

  748.             kafka_service = KafkaService(KAFKA_BOOTSTRAP_SERVERS, KAFKA_TOPIC, KAFKA_TOPIC, KAFKA_GROUP_ID)

  749.             await kafka_service.start_producer()

  750.             global_config = await gewe_service.get_global_config_from_cache_async()

  751.             scheduled_task_add_contacts_from_chatrooms_config = global_config.get('scheduledTaskAddContactsFromChatrooms', {})

  752. 

  753.             oneday_add_contacts_total = 90

  754.             once_add_contacts_total = 30

  755.             # oneday_times=3

  756.             if scheduled_task_add_contacts_from_chatrooms_config:

  757.                 oneday_add_contacts_total = scheduled_task_add_contacts_from_chatrooms_config.get('oneDayAddContactsTotal', 90)

  758.                 once_add_contacts_total = scheduled_task_add_contacts_from_chatrooms_config.get('onceAddContactsTotal', 30)

  759.                 # oneday_times=scheduled_task_add_contacts_from_chatrooms_config.get('oneDayTimes',3)

  760. 

  761.             login_keys = []

  762.             async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  763.                 login_keys.append(key)

  764. 

  765.             # 设置 Semaphore，限制并发任务数量

  766.             semaphore = asyncio.Semaphore(10)  # 例如，限制同时运行的任务数量为 10

  767.             # 使用 asyncio.gather 并发处理每个 login_key

  768.             await asyncio.gather(*[process_login_key(redis_service, gewe_service, kafka_service, k, gewe_config, oneday_add_contacts_total, once_add_contacts_total, semaphore) for k in login_keys])

  769. 

  770.         except Exception as e:

  771.             # 获取当前的堆栈跟踪

  772.             tb = sys.exc_info()[2]

  773.             # 为异常附加堆栈跟踪

  774.             e = e.with_traceback(tb)

  775.             # 输出详细的错误信息

  776.             logger.error(f"任务执行过程中发生异常: {e}\n异常类型: {type(e).__name__}\n异常信息: {str(e)}\n堆栈跟踪: {traceback.format_exc()}")

  777.         finally:

  778.             await kafka_service.stop_producer()

  779. 

  780.     loop = asyncio.get_event_loop()

  781.     if loop.is_closed():

  782.         loop = asyncio.new_event_loop()

  783.         asyncio.set_event_loop(loop)

  784.     loop.run_until_complete(task())  # 在现有事件循环中运行任务

  785. 

  786. 

  787. 

  788. # @celery_app.task(name='tasks.scheduled_task_add_contacts_from_chatrooms', bind=True, acks_late=True)

  789. # def scheduled_task_add_contacts_from_chatrooms(self, redis_config, kafka_config, gewe_config):

  790. 

  791. #     '''

  792. #     关于群加好友的请求规则: 每个智能体一次最多30人,间隔2小时做1次,即最多90人/天。

  793. #     加好友规则:每天处理次数、间隔时间(分钟)、每次加好友人数这3个参数都可以设置。目前默认只是上面的设置。

  794. #     '''

  795. #     async def process_login_key(redis_service:RedisService, gewe_service: GeWeService, kafka_service:KafkaService, k, gewe_config, oneday_add_contacts_total, once_add_contacts_total):

  796. #         r = await redis_service.get_hash(k)

  797. #         app_id = r.get("appId")

  798. #         token_id = r.get("tokenId")

  799. #         wxid = r.get("wxid")

  800. #         status = r.get('status')

  801. 

  802. #         if status == '0':

  803. #             logger.warning(f"微信号 {wxid} 已经离线，群成员不能定时添加")

  804. #             return

  805. 

  806. #         config = await gewe_service.get_wxchat_config_from_cache_async(wxid)

  807. #         validated_config = AgentConfig.model_validate(config)

  808. #         if not validated_config.agentEnabled:

  809. #             logger.warning(f"微信号 {wxid} 取消了托管，群成员不能定时添加")

  810. #             return

  811. 

  812. #         # 判断是否过于频繁

  813. #         is_wx_expection = await gewe_service.get_wx_expection_async(wxid, "addGroupMemberAsFriend")

  814. #         if is_wx_expection:

  815. #             logger.info(f"{wxid} 本次任务接口addGroupMemberAsFriend异常, {is_wx_expection}，本次群好友邀请任务未开始,跳过任务。")

  816. #             return

  817. 

  818. #         cache_task_run_time_wxid_logs = await gewe_service.get_task_run_time_by_wxid_async(wxid, 'scheduled_task_add_contacts_from_chatrooms')

  819. #         if cache_task_run_time_wxid_logs:

  820. #             sorted_tasks = sorted(cache_task_run_time_wxid_logs, key=lambda x: x.get("runTime"), reverse=True)

  821. #             last_run_time = sorted_tasks[0].get("runTime")

  822. #             if last_run_time > 1e12:  # 毫秒级时间戳

  823. #                 last_run_time = last_run_time / 1000  # 转换为秒

  824. 

  825. #             # 将时间戳转换为 datetime 对象

  826. #             last_run_time = datetime.fromtimestamp(last_run_time)

  827. 

  828. #             # 获取当前时间

  829. #             current_time = datetime.now()

  830. 

  831. #             # 计算时间差

  832. #             time_difference = current_time - last_run_time

  833. 

  834. #             # 判断是否相差2小时

  835. #             if time_difference < timedelta(hours=2):

  836. #                 logger.info(f"{wxid}上次定时群成员定时添好友任务在2小时内，不再执行")

  837. #                 return

  838. 

  839. #         cache_task_run_time_wxid_logs.append({"runTime": int(time.time())})

  840. #         await gewe_service.save_task_run_time_by_wxid_async(wxid, 'scheduled_task_add_contacts_from_chatrooms', cache_task_run_time_wxid_logs, 3600 * 2)

  841. #         c:dict = await gewe_service.get_wxchat_config_from_cache_async(wxid)

  842. #         contacts = await gewe_service.get_contacts_brief_from_cache_async(wxid)

  843. 

  844. #         contact_wxids = [c.get('userName') for c in contacts]

  845. #         chatrooms = c.get('addContactsFromChatroomIdWhiteList', [])

  846. #         logger.info(f'{wxid} 定时群成员定时添好友任务开始')

  847. #         wixd_add_contacts_from_chatrooms_times = {wxid: 0}

  848. #         for chatroom_id in chatrooms:

  849. #             chatroom = await gewe_service.get_group_info_from_cache_async(wxid, chatroom_id)

  850. #             chatroom_member = await gewe_service.get_group_members_from_cache_async(wxid, chatroom_id)

  851. 

  852. #             chatroom_nickname = chatroom.get('nickName')

  853. #             chatroom_owner_wxid = chatroom_member.get('chatroomOwner', None)

  854. #             admin_wxids = chatroom_member.get('adminWxid', [])

  855. #             admin_wxids = chatroom_member.get('adminWxid')

  856. #             if admin_wxids is None:

  857. #                 admin_wxids = []  # 如果 admin_wxids 是 None，将其初始化为空列表

  858. 

  859. #             logger.info(f'{chatroom_nickname} 的群主是 {chatroom_owner_wxid}，管理员是{admin_wxids}')

  860. #             contact_wxids_set = set(contact_wxids)

  861. #             if admin_wxids:

  862. #                 contact_wxids_set.update(set(admin_wxids))

  863. #             if chatroom_owner_wxid is not None:

  864. #                 contact_wxids_set.add(chatroom_owner_wxid)

  865. 

  866. #             contact_wxids_set.add(wxid)

  867. #             unavailable_wixds = await gewe_service.check_wixd_group_add_contacts_history_async(wxid, chatroom_id)

  868. #             if unavailable_wixds:

  869. #                 contact_wxids_set.update(set(unavailable_wixds))

  870. 

  871. #             chatroom_member_list = chatroom.get('memberList', [])

  872. #             if chatroom_member_list is None:

  873. #                 chatroom_member_list = []  # 如果 memberList 是 None，将其初始化为空列表

  874. #             elif not isinstance(chatroom_member_list, list):

  875. #                 chatroom_member_list = list(chatroom_member_list)  # 如果 memberList 不是列表，将其转换为列表

  876. 

  877. #             remaining_chatroot_members = [x for x in chatroom_member_list if x.get('wxid') not in contact_wxids_set]

  878. 

  879. #             nickname = next((member['nickName'] for member in chatroom_member_list if member['wxid'] == wxid), None)

  880. #             if not remaining_chatroot_members:

  881. #                 logger.info(f'{nickname}-{wxid} 在 {chatroom_nickname} 群里没有好友可以邀请')

  882. #                 # 任务状态推送到kafka

  883. #                 k_message = wx_add_contacts_from_chatroom_task_status_message(wxid, chatroom_id, 2)

  884. #                 await kafka_service.send_message_async(k_message)

  885. #                 return 

  886.             

  887. #             logger.info(f'{nickname}-{wxid} 在 {chatroom_nickname} 群里还可以邀请的好友有：{[x.get("nickName") for x in remaining_chatroot_members]}')

  888. #             for m in remaining_chatroot_members:

  889. #                 # 判断本次任务是否已经邀请了30个好友

  890. #                 if wixd_add_contacts_from_chatrooms_times[wxid] == once_add_contacts_total:

  891. #                     logger.info(f"{wxid} 本次任务已经邀请了{once_add_contacts_total}人,不再邀请")

  892. #                     return

  893. 

  894. #                 # 判断当天群成员是否已经加了90个好友

  895. #                 is_add_group_times = await gewe_service.is_group_add_contacts_history_one_day_async(wxid, oneday_add_contacts_total)

  896. #                 if is_add_group_times:

  897. #                     logger.info(f"当天 {wxid} 所有群的成员已经加了{oneday_add_contacts_total}个好友，不再添加")

  898. #                     return

  899. 

  900. #                 # 判断是否过于频繁

  901. #                 is_wx_expection = await gewe_service.get_wx_expection_async(wxid, "addGroupMemberAsFriend")

  902. #                 if is_wx_expection:

  903. #                     logger.info(f"{wxid} 本次任务接口addGroupMemberAsFriend异常,不再邀请,{is_wx_expection}")

  904. #                     return

  905.                 

  906. #                 contact_wxid = m.get('wxid')

  907. #                 member_nickname = m.get("nickName")

  908. #                 group_add_contacts_history = await gewe_service.get_group_add_contacts_history_async(wxid, chatroom_id, contact_wxid)

  909. 

  910. #                 if group_add_contacts_history:

  911. #                     sorted_history = sorted(group_add_contacts_history, key=lambda x: x.addTime, reverse=True)

  912. 

  913. #                     # 已经邀请过两次，不再邀请

  914. #                     if len(sorted_history) == 2:

  915. #                         logger.info(f'{nickname}-{wxid}在{chatroom_nickname}-{chatroom_id} 群的 {member_nickname}-{contact_wxid} 已经邀请过2次，不再邀请')

  916. #                         return

  917.                     

  918. #                     # 当天邀请过，不再邀请

  919. #                     if len(sorted_history) > 0:

  920. #                         last_add_time = sorted_history[0].addTime

  921. 

  922. #                         def is_add_time_more_than_one_day(addTime: int) -> bool:

  923. #                             """

  924. #                             判断 addTime 是否与当前时间相隔大于 3600 × 24 秒

  925. #                             :param addTime: Unix 时间戳

  926. #                             :return: 如果 addTime 与当前时间相隔大于 3600 × 24 秒，返回 True；否则返回 False

  927. #                             """

  928. #                             # 获取当前时间的时间戳

  929. #                             current_time = time.time()

  930. 

  931. #                             # 计算时间戳差值

  932. #                             time_difference = abs(current_time - addTime)

  933. 

  934. #                             # 检查是否大于 3600 × 24 秒

  935. #                             return time_difference > 3600 * 24

  936. 

  937. #                         is_more_than_one_day = is_add_time_more_than_one_day(last_add_time)

  938. #                         if not is_more_than_one_day:

  939. #                             logger.info(f'{nickname}-{wxid}在{chatroom_nickname}-{chatroom_id} 群的 {member_nickname}-{contact_wxid} 已经当天邀请，不再邀请')

  940. #                             return 

  941. 

  942. #                 ret, msg, data = await gewe_service.add_group_member_as_friend_async(token_id, app_id, chatroom_id, m.get('wxid'),

  943. #                                                                                     f'我是群聊"{chatroom_nickname}"群的{nickname}')

  944. #                 if ret != 200:

  945. #                     logger.warning(f'群好友邀请失败原因：{ret} {msg} {data}')

  946. #                     if '操作过于频繁' in data.get('msg'):

  947. #                         await gewe_service.save_wx_expection_async(wxid, "addGroupMemberAsFriend", msg, today_seconds_remaining())

  948. #                         logger.warning(f'{nickname}-{wxid} 操作过于频繁，本次群好友邀请任务未完成跳过。')

  949. #                         return 

  950.                     

  951. #                 history = AddGroupContactsHistory.model_validate({

  952. #                     "chatroomId": chatroom_id,

  953. #                     "wxid": wxid,

  954. #                     "contactWixd": contact_wxid,

  955. #                     "addTime": int(time.time())

  956. #                 })

  957. #                 await gewe_service.save_group_add_contacts_history_async(wxid, chatroom_id, contact_wxid, history)

  958. #                 wixd_add_contacts_from_chatrooms_times[wxid] += 1

  959. #                 logger.info(f'{nickname} 向 {chatroom_nickname}-{chatroom_id} 群的  {m.get("nickName")}-{m.get("wxid")} 发送好友邀请  {msg}')

  960. #                 # 推送到kafka

  961. #                 k_message = wx_add_contacts_from_chatroom_message(history.wxid, history.chatroomId, history.contactWixd, history.addTime)

  962. #                 await kafka_service.send_message_async(k_message)

  963. #                 # await asyncio.sleep(random.uniform(1.5, 3))

  964. #                 await asyncio.sleep(random.uniform(30, 60))

  965. #             # 任务状态推送到kafka

  966. #             task_status = await gewe_service.wx_add_contacts_from_chatroom_task_status_async(wxid, chatroom_id)

  967. #             wx_add_contacts_from_chatroom_task_status_message(wxid, chatroom_id, task_status)

  968. #             await kafka_service.send_message_async(k_message)

  969. #             # 下一个群

  970. #             await asyncio.sleep(random.uniform(1.5, 3))

  971. 

  972. #     async def task():

  973. #         try:

  974. #             now = datetime.now()

  975. #             if now.hour < 8:

  976. #                 logger.info(f"定时群成员定时添好友任务不启动,当前时间为 {now.strftime('%Y-%m-%d %H:%M:%S')}，早于8点")

  977. #                 return

  978. 

  979. #             logger.info('定时群成员定时添好友任务开始')

  980. #             redis_service = RedisService()

  981. #             await redis_service.init(**redis_config)

  982. #             gewe_service = await GeWeService.get_instance(redis_service, gewe_config['api_url'])

  983. #             KAFKA_BOOTSTRAP_SERVERS = kafka_config['bootstrap_servers']

  984. #             KAFKA_TOPIC = kafka_config['topic']

  985. #             KAFKA_GROUP_ID = kafka_config['group_id']

  986. 

  987. #             kafka_service = KafkaService(KAFKA_BOOTSTRAP_SERVERS, KAFKA_TOPIC, KAFKA_TOPIC, KAFKA_GROUP_ID)

  988. #             await kafka_service.start_producer()

  989. #             global_config = await gewe_service.get_global_config_from_cache_async()

  990. #             scheduled_task_add_contacts_from_chatrooms_config = global_config.get('scheduledTaskAddContactsFromChatrooms', {})

  991. 

  992. #             oneday_add_contacts_total = 90

  993. #             once_add_contacts_total = 30

  994. #             # oneday_times=3

  995. #             if scheduled_task_add_contacts_from_chatrooms_config:

  996. #                 oneday_add_contacts_total = scheduled_task_add_contacts_from_chatrooms_config.get('oneDayAddContactsTotal', 90)

  997. #                 once_add_contacts_total = scheduled_task_add_contacts_from_chatrooms_config.get('onceAddContactsTotal', 30)

  998. #                 # oneday_times=scheduled_task_add_contacts_from_chatrooms_config.get('oneDayTimes',3)

  999. 

  1000. #             login_keys = []

  1001. #             async for key in redis_service.client.scan_iter(match='__AI_OPS_WX__:LOGININFO:*'):

  1002. #                 login_keys.append(key)

  1003. 

  1004. #             # 使用 asyncio.gather 并发处理每个 login_key

  1005. #             await asyncio.gather(*[process_login_key(redis_service, gewe_service, kafka_service, k, gewe_config, oneday_add_contacts_total, once_add_contacts_total) for k in login_keys])

  1006. 

  1007. #         except Exception as e:

  1008. #             # 获取当前的堆栈跟踪

  1009. #             tb = sys.exc_info()[2]

  1010. #             # 为异常附加堆栈跟踪

  1011. #             e = e.with_traceback(tb)

  1012. #             # 输出详细的错误信息

  1013. #             logger.error(f"任务执行过程中发生异常: {e}\n异常类型: {type(e).__name__}\n异常信息: {str(e)}\n堆栈跟踪: {traceback.format_exc()}")

  1014. #         finally:

  1015. #             await kafka_service.stop_producer()

  1016. 

  1017. #     loop = asyncio.get_event_loop()

  1018. #     if loop.is_closed():

  1019. #         loop = asyncio.new_event_loop()

  1020. #         asyncio.set_event_loop(loop)

  1021. #     loop.run_until_complete(task())  # 在现有事件循环中运行任务

  1022. 

  1023. 

  1024. 

  1025. 

  1026. 

  1027. 

  1028. REDIS_KEY_PATTERN = "friend_add_limit:{date}"

  1029. REDIS_LAST_RUN_KEY = "last_run_time:add_friends_task"

  1030. 

  1031. @celery_app.task(name='tasks.add_friends_task', bind=True, acks_late=True)

  1032. def add_friends_task(self,redis_config):

  1033.     """

  1034.     限制每天最多 15 个，每 2 小时最多 8 个

  1035.     """

  1036.     async def task():

  1037.         redis_service = RedisService()

  1038.         await redis_service.init(**redis_config)

  1039.         today_str = datetime.now().strftime("%Y%m%d")

  1040.         redis_key = REDIS_KEY_PATTERN.format(date=today_str)

  1041. 

  1042.         # 获取当前总添加数量

  1043.         total_added = await redis_service.get_hash_field(redis_key, "total") or 0

  1044.         last_2h_added =await redis_service.get_hash_field(redis_key, "last_2h") or 0

  1045. 

  1046.         total_added = int(total_added)

  1047.         last_2h_added = int(last_2h_added)

  1048. 

  1049.         logger.info(f"当前添加好友总数: {total_added}, 过去2小时添加: {last_2h_added}")

  1050. 

  1051.         # 判断是否超过限制

  1052.         if total_added >= 15:

  1053.             logger.warning("今日好友添加已达上限！")

  1054.             return

  1055. 

  1056.         if last_2h_added >= 8:

  1057.             logger.warning("过去2小时添加已达上限！")

  1058.             return

  1059. 

  1060.         # 计算本次要添加的好友数量 (控制每天 5-15 个)

  1061.         max_add = min(15 - total_added, 8 - last_2h_added)

  1062.         if max_add <= 0:

  1063.             return

  1064. 

  1065.         num_to_add = min(max_add, 1)  # 每次最多加 1 个

  1066.         logger.info(f"本次添加 {num_to_add} 位好友")

  1067. 

  1068.         # TODO: 调用好友添加逻辑 (接口 or 业务逻辑)

  1069.         # success = add_friends(num_to_add)

  1070. 

  1071.         success = num_to_add  # 假设成功添加 num_to_add 个

  1072. 

  1073.         # 更新 Redis 计数

  1074.         if success > 0:

  1075.             await redis_service.increment_hash_field(redis_key, "total", success)

  1076.             await redis_service.increment_hash_field(redis_key, "last_2h", success)

  1077. 

  1078.             # 设置 Redis 过期时间 (每日记录存 1 天, 2 小时记录存 2 小时)

  1079.             await redis_service.expire(redis_key, 86400)  # 24小时

  1080.             await redis_service.expire_field(redis_key, "last_2h", 7200)  # 2小时

  1081. 

  1082.         logger.info(f"成功添加 {success} 位好友, 今日总数 {total_added + success}")

  1083. 

  1084.         # 生成一个新的随机时间（5-15 分钟之间）

  1085.         # next_interval = random.randint(10, 20)

  1086. 

  1087.         # # 计算新的执行时间

  1088.         # next_run_time = datetime.datetime.now() + timedelta(seconds=next_interval)

  1089. 

  1090.         # # 重新注册 RedBeat 任务，确保下次执行时间不同

  1091.         # redbeat_entry = RedBeatSchedulerEntry(

  1092.         #     name="redbeat:add_friends_task",

  1093.         #     task="tasks.add_friends_task",

  1094.         #     schedule=celery.schedules.schedule(timedelta(seconds=next_interval)),

  1095.         #     args=[redis_config],

  1096.         #     app=celery_app

  1097.         # )

  1098. 

  1099.         # # 设置任务的下次执行时间

  1100.         # redbeat_entry.last_run_at = next_run_time

  1101.         # redbeat_entry.save()

  1102. 

  1103.         # logger.info(f"下次任务将在 {next_run_time} 执行（间隔 {next_interval} 秒）")

  1104. 

  1105.     loop = asyncio.get_event_loop()

  1106.     

  1107.     if loop.is_closed():

  1108.         loop = asyncio.new_event_loop()

  1109.         asyncio.set_event_loop(loop)

  1110. 

  1111.     loop.run_until_complete(task())  # 在现有事件循环中运行任务

  1112. 

  1113. def today_seconds_remaining()->int:

  1114.     current_time = datetime.now()

  1115. 

  1116.     # 计算当天的结束时间（23:59:59）

  1117.     end_of_day = datetime(current_time.year, current_time.month, current_time.day, 23, 59, 59)

  1118. 

  1119.     # 计算时间差

  1120.     time_difference = end_of_day - current_time

  1121. 

  1122.     # 将时间差转换为秒数

  1123.     time_difference_seconds = int(time_difference.total_seconds())

  1124. 

  1125.     return time_difference_seconds

  1126. 

  1127. 

  1128. @celery_app.task(name='tasks.random_scheduled_task', bind=True, acks_late=True)

  1129. def random_scheduled_task(self,):

  1130.     print(f"Task executed at {datetime.now()}")

  1131.     # 随机生成下次执行时间（例如：10-60秒内的随机时间）

  1132.     next_run_in = random.randint(10, 60)

  1133.     print(f"Next execution will be in {next_run_in} seconds")

  1134.     

  1135.     # 设置下次执行时间

  1136.     entry = RedBeatSchedulerEntry(

  1137.         name='random-task',

  1138.         task='tasks.random_scheduled_task',

  1139.         schedule=timedelta(seconds=next_run_in),

  1140.         app=celery_app

  1141.     )

  1142.     entry.save()

  1143.     return f"Scheduled next run in {next_run_in} seconds"