Bladeren bron

Update agent_service: add staff whitelist for pushing

StrayWarrior 3 dagen geleden
bovenliggende
commit
f22417350a
1 gewijzigde bestanden met toevoegingen van 4 en 0 verwijderingen
  1. 4 0
      pqai_agent/agent_service.py

+ 4 - 0
pqai_agent/agent_service.py

@@ -415,8 +415,12 @@ class AgentService:
             return
 
         push_scan_threads = []
+        whitelist_staffs = apollo_config.get_json_value("agent_initiate_whitelist_staffs", [])
         for staff in self.user_relation_manager.list_staffs():
             staff_id = staff['third_party_user_id']
+            if staff_id not in whitelist_staffs:
+                logger.info(f"staff[{staff_id}] is not in whitelist, skip")
+                continue
             scan_thread = threading.Thread(target=PushScanThread(
                 staff_id, self, self.push_task_rmq_topic, self.push_task_producer).run)
             scan_thread.start()