client.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702
  1. #!/usr/bin/env python3
  2. # encoding:utf-8
  3. import random
  4. import socketserver
  5. import ESL
  6. import time
  7. import uuid
  8. import threading
  9. import traceback
  10. import concurrent.futures
  11. from apscheduler.schedulers.background import BackgroundScheduler
  12. from src.core.callcenter import BizException
  13. from src.core.callcenter.cache import Cache
  14. from src.core.callcenter.api import MakeCallContext, DelayAction, CallInfo, DeviceInfo, NextCommand
  15. from src.core.callcenter.callback import Callback
  16. from src.core.callcenter.constant import SK, EMPTY, WaitingHangupMusicPath, SAAS_ID, HOLD_MUSIC_PATH
  17. from src.core.callcenter.esl.constant.esl_constant import BRIDGE_VARIABLES, BRIDGE, HANGUP, NORMAL_CLEARING, SIP_HEADER, \
  18. SPACE, SOFIA, \
  19. ORIGINATE, PARK, SET, EAVESDROP, SMF_ALEG, EXECUTE, PLAYBACK, PAUSE, TRANSFER, UUID_TRANSFER, UUID_BROADCAST, \
  20. UUID_BREAK, UUID_HOLD, \
  21. UUID_RECORD, UUID_SETVAR, UUID_GETVAR, UUID_KILL, ANSWER
  22. import src.core.callcenter.esl.utils.esl_event_util as EslEventUtil
  23. import src.core.callcenter.esl.handler as event_handler
  24. from src.core.callcenter.esl.constant.sip_header_constant import sipHeaderHoldMusic, profile1, profile2, sipHeaderCallId
  25. from src.core.callcenter.enumeration import CallCause, DeviceType, DelayActionEnum, HangupDir, CallType, NextType, \
  26. Direction, CdrType, BizErrorCode, WhiteTypeEnum
  27. from src.core.callcenter.esl.handler.default_esl_event_handler import DefaultEslEventHandler
  28. from src.core.callcenter.snowflake import Snowflake
  29. from src.core.datasource import SERVE_HOST
  30. from src.core.voip.constant import *
  31. from src.core.callcenter.data_handler import *
  32. from src.core.callcenter import registry
  33. class InboundClient:
  34. def __init__(self, agent, app):
  35. self.con = None
  36. self.thread_num = 12
  37. self.is_stopping = False
  38. self.app = app
  39. self.logger = app.logger
  40. self.bot_agent = agent
  41. self.cache = Cache(app)
  42. self.callback = Callback(app, self.thread_num)
  43. self.dataHandleServer = DataHandleServer(app)
  44. self.handler_table = self.scan_esl_event_handlers()
  45. self.default_event_handler = DefaultEslEventHandler(self, self.bot_agent)
  46. self.host, self.port, self.password = SERVE_HOST, '8021', '4918257983818884358'
  47. self.executors = {x: concurrent.futures.ThreadPoolExecutor(max_workers=1, thread_name_prefix="client-event-pool") for x in range(self.thread_num)}
  48. self.delay_action_executor = concurrent.futures.ThreadPoolExecutor(max_workers=5)
  49. self.delay_action_scheduler = BackgroundScheduler()
  50. self.delay_action_scheduler.add_job(self.submit_delay_action, 'interval', seconds=1, max_instances=1, name='delay_action_daemon')
  51. self.delay_action_scheduler.start()
  52. threading.Thread(target=self.start, args=()).start()
  53. def submit_delay_action(self):
  54. for name, member in DelayActionEnum.__members__.items():
  55. action_messages = self.cache.get_delay_message(name)
  56. for action_message in action_messages:
  57. self.delay_action_executor.submit(self.do_delay_action, name, action_message)
  58. def scan_esl_event_handlers(self):
  59. import inspect
  60. import importlib
  61. import pkgutil
  62. classes = []
  63. # 遍历包中的模块
  64. for module_info in pkgutil.iter_modules(event_handler.__path__):
  65. module = importlib.import_module(f"{event_handler.__name__}.{module_info.name}")
  66. for _name, _cls in inspect.getmembers(module, inspect.isclass):
  67. if hasattr(_cls, '_esl_event_name'):
  68. classes.append(_cls)
  69. handlers = {}
  70. for _cls in classes:
  71. items = handlers.get(_cls._esl_event_name, [])
  72. items.append(_cls(self, self.bot_agent))
  73. handlers[_cls._esl_event_name] = items
  74. return handlers
  75. def start(self):
  76. self.logger.info('inbound.start')
  77. self.con = ESL.ESLconnection(self.host, self.port, self.password)
  78. if self.con.connected():
  79. self.logger.info('inbound esl connected ... ')
  80. self.con.events('plain', 'all') #CHANNEL_ORIGINATE,CHANNEL_PROGRESS,CHANNEL_PROGRESS_MEDIA,CHANNEL_ANSWER,CHANNEL_HANGUP,CUSTOM,PLAYBACK_START,PLAYBACK_STOP,DETECTED_TONE
  81. while not self.is_stopping:
  82. e = self.con.recvEvent()
  83. # if e:
  84. # self.logger.info(json.loads(e.serialize('json')))
  85. event_name = e.getHeader("Event-Name")
  86. if event_name == "SERVER_DISCONNECTED":
  87. self.logger.info('come in SERVER_DISCONNECTED case')
  88. self.con.disconnect()
  89. time.sleep(3)
  90. self.start()
  91. else:
  92. registry.FLASK_ACTIVE_THREADS.set(threading.active_count())
  93. # threading.Thread(target=self.process_esl_event, args=(e,)).start()
  94. self.choose_thread_pool_executor(e).submit(self.process_esl_event, e)
  95. def choose_thread_pool_executor(self, e):
  96. call_id = EslEventUtil.getCallId(e)
  97. device_id = EslEventUtil.getUniqueId(e)
  98. wdh_device_id = EslEventUtil.getDeviceId(e)
  99. random_id = call_id if call_id else device_id
  100. if random_id:
  101. random_index = abs(mmh3.hash(random_id)) % len(self.executors)
  102. else:
  103. random_index = random.randint(0, len(self.executors) - 1) if self.executors else 0
  104. # self.logger.info('choose_thread_pool_executor.index=', random_index, call_id, device_id, wdh_device_id)
  105. return self.executors.get(random_index)
  106. def process_esl_event(self, e):
  107. # self.logger.info(json.loads(e.serialize('json')))
  108. start_time = time.time()
  109. event_name = EslEventUtil.getEventName(e)
  110. coreUUID = EslEventUtil.getCoreUuid(e)
  111. address = self.host + ':' + self.port
  112. # self.logger.info("process_esl_event.event_name=%s,coreUUID=%s", event_name, coreUUID)
  113. try:
  114. self.callback.callback_event(e)
  115. if event_name in self.handler_table:
  116. items = self.handler_table.get(event_name)
  117. for x in items:
  118. try:
  119. # self.logger.info("process_esl_event.handle.%s", x.__class__.__name__)
  120. x.handle(address, e, coreUUID)
  121. except:
  122. traceback.print_exc()
  123. else:
  124. self.default_event_handler.handle(address, e, coreUUID)
  125. except:
  126. traceback.print_exc()
  127. finally:
  128. latency = (time.time() - start_time)
  129. registry.ESL_EVENT_LATENCY.labels(event_name).observe(latency)
  130. def do_delay_action(self, action, message):
  131. delay_action = DelayAction.from_json(message)
  132. flag = self.cache.lock_delay_action(delay_action.uuid)
  133. self.logger.info("do_delay_action::action=%s, flag=%s", action, flag)
  134. if not flag:
  135. self.logger.info("异步延迟执行操作重复 action:%s msg:%s", action, message)
  136. return
  137. delay_action_enum = DelayActionEnum.get_by_code(action)
  138. if not delay_action_enum:
  139. self.logger.info("异步延迟执行 delayActionEnum为空 action:%s msg:%s", action, message)
  140. return
  141. if DelayActionEnum.CALL_TIMEOUT_HANGUP == delay_action_enum:
  142. self.exec_when_call_timeout(delay_action.call_id, delay_action.device_id)
  143. elif DelayActionEnum.PLAY_TIMEOUT_HANGUP == delay_action_enum:
  144. self.exec_when_play_timeout(delay_action.call_id)
  145. elif DelayActionEnum.ACD_TIMEOUT_PLAY == delay_action_enum:
  146. self.exec_when_acd_timeout(delay_action.call_id)
  147. def exec_when_call_timeout(self, call_id, device_id):
  148. call_info = self.cache.get_call_info(call_id)
  149. if not call_info or not (device_id in call_info.device_list):
  150. return
  151. device_info = call_info.device_info_map.get(device_id)
  152. if device_info and device_info.answer_time is None:
  153. self.logger.info("call:%s deviceId:%s execWhenCallTimeOut", call_id, device_id)
  154. device_info.hangup_cause = CallCause.CALL_TIMEOUT.name
  155. call_info.next_commands = []
  156. if device_info.device_type <= DeviceType.ROBOT.code:
  157. call_info.hangup_dir = HangupDir.PLATFORM_HANGUP.code
  158. call_info.hangup_code = CallCause.CALL_TIMEOUT.code
  159. if device_info.device_type.code == DeviceType.CUSTOMER.code:
  160. call_info.user_no_answer_end_call = True
  161. # if not device_info.end_time and device_info.device_type.code == DeviceType.CUSTOMER.code:
  162. # channel = self.show_channel(device_id)
  163. # if channel:
  164. # delay_action = DelayAction(call_id=call_id, device_id=device_id)
  165. # self.cache.add_delay_message(DelayActionEnum.CALL_TIMEOUT_DECR, delay_action, timeouts=20)
  166. self.cache.add_call_info(call_info)
  167. self.hangup_call(call_id, device_id, CallCause.CALL_TIMEOUT)
  168. # self.dataHandleServer.update_record(call_id, status= 0)
  169. def exec_when_play_timeout(self, call_id):
  170. call_info = self.cache.get_call_info(call_id)
  171. if not call_info or not call_info.next_commands:
  172. return
  173. self.logger.info("播放结束音乐失败,进行挂机 callId:%s", call_id)
  174. next_types = [x.next_type for x in call_info.next_commands]
  175. if NextType.NEXT_HANGUP.code in next_types:
  176. for device_id in call_info.device_list:
  177. self.hangup_call(call_id, device_id, CallCause.PLAY_TIMEOUT)
  178. def exec_when_acd_timeout(self, call_id):
  179. call_info = self.cache.get_call_info(call_id)
  180. if not call_info:
  181. self.logger.info("exec_when_acd_timeout callInfo为空 callId: %s", call_id)
  182. return
  183. device_list = [v for k,v in call_info.device_info_map.items() if v.device_type == DeviceType.CUSTOMER]
  184. if device_list and len(device_list) == 1:
  185. device_id = device_list[0].device_id
  186. self.break0(device_id)
  187. if not WaitingHangupMusicPath:
  188. self.hangup_call(call_id, device_id, CallCause.WAITING_TIMEOUT)
  189. return
  190. self.hold_play(device_id, WaitingHangupMusicPath)
  191. self.play_timeout(call_id, timeout=30)
  192. next_command = NextCommand(device_id = device_id, next_type=NextType.NEXT_HANGUP.code)
  193. call_info.next_commands = [next_command]
  194. self.cache.add_call_info(call_info)
  195. # self.dataHandleServer.update_record(call_id, status= 0)
  196. self.logger.info("waitingTimeOut 开始播放结束音乐 callId:%s customerDeviceId:%s playFile:%s", call_id,
  197. device_id, WaitingHangupMusicPath)
  198. def make_call(self, context: MakeCallContext):
  199. # self.logger.info("拨打测试context:%s", context.__dict__)
  200. called = context.get_called()
  201. params = {'gateway': context.route_gateway_name, 'called': called, 'realm': context.get_realm()}
  202. builder = [
  203. '{', context.get_sip_header(), '}'
  204. ]
  205. self.call_timeout(context.call_id, context.device_id, context.timeout)
  206. if context.device_type == DeviceType.CUSTOMER.code:
  207. profile = self.expression(profile1, params)
  208. builder.append(f"{SOFIA}{SK}{profile}{SK}{called}{PARK}")
  209. else:
  210. profile = self.expression(profile2, params)
  211. builder.append(f"{profile}{PARK}")
  212. # self.logger.info("拨打测试builder:%s", builder)
  213. cmd = "".join(builder)
  214. # self.logger.info("makeCall::%s", cmd)
  215. self.bgapi(ORIGINATE, arg=cmd, desc="make_call")
  216. # self.con.bgapi(ORIGINATE, cmd)
  217. def call_timeout(self, call_id, device_id, timeout):
  218. """呼叫超时主动挂机"""
  219. delay_action = DelayAction(call_id=call_id, device_id=device_id)
  220. self.cache.add_delay_message(DelayActionEnum.CALL_TIMEOUT_HANGUP.code, delay_action, timeouts=timeout)
  221. def send_args(self, device_id, name, arg, con=None):
  222. msg = ESL.ESLevent("sendmsg", device_id)
  223. msg.addHeader("call-command", EXECUTE)
  224. msg.addHeader("execute-app-name", name)
  225. msg.addHeader("execute-app-arg", arg)
  226. if con:
  227. con.sendEvent(msg)
  228. else:
  229. self.con.sendEvent(msg)
  230. def bridge_call(self, call_id, device_id1, device_id2):
  231. """桥接电话"""
  232. self.multi_set_var(device_id1, BRIDGE_VARIABLES)
  233. self.multi_set_var(device_id2, BRIDGE_VARIABLES)
  234. self.bgapi(BRIDGE, arg=device_id1 + SPACE + device_id2, desc="bridge_call")
  235. def transfer_call(self, _from, _to):
  236. """转接"""
  237. builder = [
  238. _from,
  239. " -both 'set:hangup_after_bridge=false,set:park_after_bridge=true,park:' inline "
  240. ]
  241. command = ''.join(builder)
  242. self.bgapi(command, desc="transfer_call")
  243. def answer(self, device_id):
  244. """应答"""
  245. self.con.bgapi('uuid_phone_event', device_id + ' talk')
  246. def hangup_call(self, call_id, device_id, case_enum=CallCause.DEFAULT):
  247. """挂机"""
  248. msg = ESL.ESLevent("sendmsg", device_id)
  249. msg.addHeader("call-command", EXECUTE)
  250. msg.addHeader("execute-app-name", HANGUP)
  251. msg.addHeader("execute-app-arg", NORMAL_CLEARING)
  252. # msg.addHeader("async", "true")
  253. _con = None
  254. try:
  255. _con = ESL.ESLconnection(self.host, self.port, self.password)
  256. if _con.connected():
  257. arg = ''.join([EslEventUtil.SIP_H_P_LIBRA_HANGUP_CAUSE, "=", case_enum.description])
  258. self.send_args(device_id, SET, arg, con=_con)
  259. e = _con.sendMSG(msg)
  260. # e = _con.api(UUID_KILL, device_id)
  261. self.logger.info("hangup_call挂机 hangup call: %s, device: %s, arg: %s, res:%s" % (call_id, device_id, arg, e))
  262. # self.logger.info("hangup_call挂机 hangup call: %s, device: %s, arg: %s, res:%s" % (call_id, device_id, arg, e.serialize('json')))
  263. except:
  264. traceback.print_exc()
  265. finally:
  266. if _con:
  267. _con.disconnect()
  268. def kill_call(self, call_id, device_id, case_enum=CallCause.DEFAULT):
  269. if not device_id:
  270. return
  271. self.logger.info("kill_call:%s, device:%s, case_enum:%s", call_id, device_id, case_enum)
  272. self.api(UUID_KILL, device_id, desc="kill_call")
  273. def broadcast(self, uuid, path, smf):
  274. builder = [
  275. UUID_BROADCAST,
  276. uuid,
  277. path,
  278. smf
  279. ]
  280. command = ' '.join(builder)
  281. self.bgapi(command, desc="broadcast")
  282. # self.con.bgapi(command, EMPTY)
  283. def break0(self, uuid, all=False, sync=True):
  284. builder = [
  285. UUID_BREAK,
  286. uuid
  287. ]
  288. if all:
  289. builder.append("all")
  290. command = ' '.join(builder)
  291. if sync:
  292. self.api(command, desc="break0")
  293. else:
  294. self.bgapi(command, desc="break0")
  295. def hold(self, smf, uuid, display):
  296. builder = [
  297. UUID_HOLD,
  298. smf,
  299. uuid,
  300. display
  301. ]
  302. if display:
  303. builder.append("all")
  304. else:
  305. builder.append(EMPTY)
  306. command = ' '.join(builder)
  307. self.con.bgapi(command, EMPTY)
  308. def get_var(self, uuid, var):
  309. builder = [
  310. UUID_GETVAR,
  311. uuid,
  312. var
  313. ]
  314. command = ' '.join(builder)
  315. self.con.bgapi(command, EMPTY)
  316. def set_var(self, uuid, var, val):
  317. builder = [
  318. UUID_SETVAR,
  319. uuid,
  320. var,
  321. val
  322. ]
  323. command = ' '.join(builder)
  324. self.bgapi(command, desc ="set_var")
  325. def multi_set_var(self, uuid, params):
  326. builder = [
  327. "uuid_setvar_multi " + uuid + " "
  328. ]
  329. builder1 = []
  330. for k, v in params.items():
  331. builder1.append(k + "="+v)
  332. builder.append(';'.join(builder1))
  333. command = ''.join(builder)
  334. self.bgapi(command, desc="multi_set_var")
  335. def record(self, uuid, action, path, limit):
  336. builder = [
  337. UUID_RECORD,
  338. uuid,
  339. action,
  340. path,
  341. str(limit)
  342. ]
  343. command = ' '.join(builder)
  344. self.bgapi(command, desc="record")
  345. def transfer(self, uuid, smf, dest, dialplan, context):
  346. builder = [
  347. UUID_TRANSFER,
  348. uuid,
  349. smf,
  350. dest,
  351. dialplan,
  352. context
  353. ]
  354. command = ' '.join(builder)
  355. self.con.bgapi(command, EMPTY)
  356. def insert(self, device_id):
  357. """强插"""
  358. builder = [
  359. device_id,
  360. " -both 'set:hangup_after_bridge=false,set:park_after_bridge=true,park:' inline "
  361. ]
  362. arg = ''.join(builder)
  363. self.con.api(TRANSFER, arg)
  364. def bridge_break(self, call_id, device_id):
  365. """拆线"""
  366. builder = [
  367. device_id,
  368. f" -both 'set:hangup_after_bridge=false,set:park_after_bridge=true,set:{SIP_HEADER}{sipHeaderHoldMusic}={call_id},park:' inline "
  369. ]
  370. arg = ''.join(builder)
  371. self.api(TRANSFER, arg, "bridge_break")
  372. def play_file(self, call_id, device_id, file, sync):
  373. """放音"""
  374. if sync:
  375. return self.hold_play(device_id, file)
  376. else:
  377. msg = ESL.ESLevent("sendmsg", device_id)
  378. msg.addHeader("call-command", EXECUTE)
  379. msg.addHeader("execute-app-name", PLAYBACK)
  380. msg.addHeader("execute-app-arg", file)
  381. msg.addHeader("async", "true")
  382. self.con.sendEvent(msg)
  383. def stop_play(self, device_id):
  384. """关闭播放音乐"""
  385. builder = [
  386. device_id,
  387. " on"
  388. ]
  389. arg = ''.join(builder)
  390. self.con.api(PAUSE, arg)
  391. def hold_play(self, device_id, play):
  392. """向a-leg插播tts音乐(无限播放)"""
  393. builder = [
  394. device_id,
  395. " playback::",
  396. play,
  397. " ",
  398. SMF_ALEG
  399. ]
  400. arg = ''.join(builder)
  401. self.api(UUID_BROADCAST, arg, "hold_play")
  402. def api(self, command, arg=EMPTY, desc=""):
  403. _con = None
  404. try:
  405. _con = ESL.ESLconnection(self.host, self.port, self.password)
  406. if _con.connected():
  407. e = _con.api(command, arg)
  408. self.logger.info('%s success, arg=%s, event=%s' % (desc, command + ' ' + arg, e.serialize('json')))
  409. except:
  410. traceback.print_exc()
  411. finally:
  412. if _con:
  413. _con.disconnect()
  414. def bgapi(self, command, arg=EMPTY, desc=""):
  415. _con = None
  416. try:
  417. _con = ESL.ESLconnection(self.host, self.port, self.password)
  418. if _con.connected():
  419. e = _con.bgapi(command, arg)
  420. self.logger.info('%s success, arg=%s, event=%s' % (desc, command + ' '+arg, e.serialize('json')))
  421. except:
  422. traceback.print_exc()
  423. finally:
  424. if _con:
  425. _con.disconnect()
  426. def play_timeout(self, call_id, timeout):
  427. """播放超时主动挂机"""
  428. delay_action = DelayAction(call_id=call_id)
  429. self.cache.add_delay_message(DelayActionEnum.PLAY_TIMEOUT_HANGUP.code, delay_action, timeouts=timeout)
  430. def listen(self, device_id1, device_id2, aleg=True, bleg=True):
  431. """监听"""
  432. if aleg:
  433. self.send_args(device_id1, SET, "eavesdrop_bridge_aleg=true")
  434. if bleg:
  435. self.send_args(device_id1, SET, "eavesdrop_bridge_bleg=true")
  436. self.send_args(device_id1, EAVESDROP, device_id2)
  437. def show_channel(self, device_id):
  438. msg = self.con.api("show", " channels like " + device_id + " as json")
  439. self.logger.info('show_channel::', msg)
  440. return msg
  441. def expression(self, template, params):
  442. for key, value in params.items():
  443. template = template.replace("#{["+key+"]}", str(value))
  444. return template
  445. def stop(self):
  446. for k, v in self.executors.items():
  447. v.shutdown()
  448. self.con.disconnect()
  449. self.is_stopping = True
  450. self.callback.stop()
  451. class OutboundClient:
  452. def __init__(self, agent, app):
  453. self.app = app
  454. self.logger = app.logger
  455. server_thread = threading.Thread(target=self.start, args=('0.0.0.0', 8084, agent))
  456. server_thread.daemon = True # 设置守护线程
  457. server_thread.start()
  458. class ESLRequestHandler(socketserver.BaseRequestHandler):
  459. def setup(self):
  460. try:
  461. self.server.logger.info('%s connected!', self.client_address)
  462. fd = self.request.fileno()
  463. con = ESL.ESLconnection(fd)
  464. self.server.logger.info('Connected: %s', con.connected())
  465. if con.connected():
  466. info = con.getInfo()
  467. self.server.logger.info(json.loads(info.serialize('json')))
  468. event_name = info.getHeader("Event-Name")
  469. self.server.logger.info('Event-Name: %s', event_name)
  470. device_id = info.getHeader("unique-id")
  471. caller_number = info.getHeader("Caller-Caller-ID-Number") # 获取来电号码
  472. call_id = 'C' + str(Snowflake().next_id())
  473. new_device_id = 'D' + str(Snowflake().next_id())
  474. kwargs = json.loads(info.serialize('json'))
  475. kwargs['variable_sip_h_P-LIBRA-CallId'] = call_id
  476. kwargs['variable_sip_h_P-LIBRA-DeviceId'] = device_id
  477. bucket_call_type = 2
  478. destination = None
  479. service_category = 0
  480. bucket = self.server.get_bucket(call_id)
  481. whitelist = self.server.get_whitelist()
  482. in_whitelist_type = self.in_whitelist(caller_number, whitelist)
  483. self.server.logger.info('call incoming call_id=%s, caller_number=%s, device_id=%s, new_device_id=%s, in_whitelist=%s, bucket=%s', call_id, caller_number, device_id, new_device_id, in_whitelist_type, bucket.name)
  484. # 检查白名单
  485. if in_whitelist_type:
  486. if WhiteTypeEnum.AI == in_whitelist_type:
  487. bucket_call_type = 0
  488. service_category = 1
  489. destination = self.bridge_ai(con, bucket_call_type, call_id, device_id, new_device_id, **kwargs)
  490. else:
  491. bucket_call_type = 0
  492. self.transfer_custom(con, bucket_call_type, call_id, device_id, new_device_id, **kwargs)
  493. else:
  494. # 自然分流
  495. if bucket and bucket.name == 'AI':
  496. bucket_call_type = 1
  497. service_category = 1
  498. destination = self.bridge_ai(con, bucket_call_type, call_id, device_id, new_device_id, **kwargs)
  499. else:
  500. self.transfer_custom(con, bucket_call_type, call_id, device_id, new_device_id, **kwargs)
  501. registry.CALL_INCOMING_REQUESTS.labels(f"{bucket_call_type}").inc()
  502. self.server.dataHandleServer.create_record(call_id, caller_number, bucket_call_type, service_category=service_category, user_id=destination if bucket_call_type == 1 else None , user_name= f"机器人{destination}" if bucket_call_type ==1 else None)
  503. try:
  504. con.disconnect()
  505. self.server.logger.info("connection disconnected !!!")
  506. except:
  507. traceback.print_exc()
  508. else:
  509. self.server.logger.info("Failed to connect to FreeSWITCH")
  510. except:
  511. traceback.print_exc()
  512. finally:
  513. try:
  514. if self.request.fileno() != -1:
  515. self.request.close()
  516. except OSError:
  517. # Ignore the error if socket is already closed
  518. pass
  519. def transfer_custom(self, con, bucket_call_type, call_id, device_id, new_device_id, **kwargs):
  520. self.answer(con, call_id, device_id)
  521. self.build_call_info(CallType.INCOMING_AGENT_CALL.code, call_id, device_id, new_device_id, destination=None, bucket_type=bucket_call_type, **kwargs)
  522. self.server.agent.acd_service.transfer_to_agent(call_id, device_id)
  523. def bridge_ai(self, con, bucket_call_type, call_id, device_id, new_device_id, **kwargs):
  524. destination = self.server.agent.register(**kwargs)
  525. self.server.logger.info("call_id=%s, device_id=%s, destination=%s, new_device_id=%s" % (call_id, device_id, destination, new_device_id))
  526. self.build_call_info(CallType.INCOMING_BOT_CALL.code, call_id, device_id, new_device_id, str(destination), bucket_type=bucket_call_type, **kwargs)
  527. self.server.cache.add_device_user_part(device_id, destination)
  528. con.execute("bridge", "{sip_h_P-LIBRA-CallId=%s,sip_h_P-LIBRA-DeviceId=%s,origination_uuid=%s}user/%s" % (call_id, new_device_id, new_device_id, destination), device_id)
  529. return destination
  530. def in_whitelist(self, caller_number, whitelist):
  531. for x in whitelist:
  532. phone, _type = x
  533. if caller_number in phone or phone in caller_number:
  534. return WhiteTypeEnum.get_by_code(_type)
  535. return None
  536. def answer(self, con, call_id, device_id, timeouts=30):
  537. con.execute("answer", "", device_id)
  538. # con.execute("bgapi", f"uuid_setvar {device_id} {sipHeaderCallId} {call_id}", device_id)
  539. con.execute("playback", HOLD_MUSIC_PATH, device_id)
  540. delay_action = DelayAction(call_id=call_id)
  541. self.server.cache.add_delay_message(DelayActionEnum.ACD_TIMEOUT_PLAY.name, delay_action, timeouts)
  542. def build_call_info(self, call_type, call_id, device_id, new_device_id, destination, bucket_type, **kwargs):
  543. caller = kwargs.get('Channel-Caller-ID-Number')
  544. called = destination
  545. now = datetime.now().timestamp()
  546. call_info = CallInfo(call_id=call_id, agent_key=destination,
  547. caller=caller, called=called, direction=Direction.INBOUND.code,
  548. call_type=call_type, call_time=now,
  549. uuid1=call_id, uuid2=device_id, saas_id=SAAS_ID, bucket_type=bucket_type,
  550. core_uuid=None, cti_flow_id=None, conference=None, group_id=None, hidden_customer=0, caller_display=None, called_display=None, number_location=None, agent_name=None, login_type=None, ivr_id=None, task_id=None, media_host=None, sip_server=None, client_host=None, record=None, record2=None, record_time=None, answer_flag=None, wait_time=None, answer_count=0, hangup_dir=None, sdk_hangup=0, hangup_code=None, answer_time=None, end_time=None, talk_time=None, first_queue_time=None, queue_start_time=None, queue_end_time=None, overflow_count=0, cdr_notify_url=None, queue_level=None, transfer_agent=None,device_list=[], device_info_map = {}, follow_data = {}, process_data = {}, next_commands=[], call_details=[])
  551. device_custom = DeviceInfo(device_id=device_id, call_time=now,
  552. call_id=call_id, device_type=DeviceType.CUSTOMER.code,
  553. agent_key=destination, cdr_type=CdrType.OUTBOUND.code,
  554. cti_flow_id=None, conference=None, agent_name=None, from_agent=None, caller=None, called=None, display=None, called_location=None, caller_location=None, ring_start_time=None, ring_end_time=None, answer_time=None, bridge_time=None, end_time=None, talk_time=None, sip_protocol=None, channel_name=None, hangup_cause=None, ring_cause=None, sip_status=None, record=None, record_time=None, record_start_time=None, state=None, apparent_number=None, caller_display=None)
  555. call_info.device_list.append(device_id)
  556. # call_info.next_commands.append(NextCommand(device_id, NextType.NEXT_CALL_BRIDGE.code, new_device_id))
  557. call_info.device_info_map = {device_id: device_custom}
  558. if CallType.INCOMING_BOT_CALL.code == call_type:
  559. device_bot = DeviceInfo(device_id=new_device_id, call_time=now,
  560. call_id=call_id, device_type=DeviceType.ROBOT.code,
  561. agent_key=destination, cdr_type=CdrType.INBOUND.code,
  562. cti_flow_id=None, conference=None, agent_name=None, from_agent=None, caller=None, called=None, display=None, called_location=None, caller_location=None, ring_start_time=None, ring_end_time=None, answer_time=None, bridge_time=None, end_time=None, talk_time=None, sip_protocol=None, channel_name=None, hangup_cause=None, ring_cause=None, sip_status=None, record=None, record_time=None, record_start_time=None, state=None, apparent_number=None, caller_display=None)
  563. call_info.device_list.append(new_device_id)
  564. call_info.device_info_map[new_device_id] = device_bot
  565. self.server.cache.add_call_info(call_info)
  566. return call_info
  567. class CustomTCPServer(socketserver.TCPServer):
  568. def __init__(self, server_address, RequestHandlerClass, agent, app):
  569. super().__init__(server_address, RequestHandlerClass)
  570. self.agent = agent
  571. self.app = app
  572. self.cache = Cache(app)
  573. self.logger = app.logger
  574. self.whitelist = []
  575. self.buckets = []
  576. # self.update_whitelist() # 初始化加载白名单
  577. # self.update_bucket() #初始化分流
  578. self.dataHandleServer = DataHandleServer(app)
  579. self.cache_job_scheduler = BackgroundScheduler()
  580. self.cache_job_scheduler.add_job(self.update_cache_job, run_date=datetime.now())
  581. self.cache_job_scheduler.add_job(self.update_cache_job, 'interval', seconds=60, max_instances=1, name='cache_job_daemon')
  582. self.cache_job_scheduler.add_job(self.update_whitelist, run_date=datetime.now())
  583. self.cache_job_scheduler.add_job(self.update_whitelist, 'interval', seconds=600, max_instances=1,name='cache_job_whiteList')
  584. self.cache_job_scheduler.start()
  585. def update_cache_job(self):
  586. # self.logger.info("begin flush cache job")
  587. # self.update_whitelist()
  588. self.update_bucket()
  589. # self.logger.info("end flush cache job")
  590. def update_whitelist(self):
  591. with self.app.app_context():
  592. phones = Whitelist.query.filter_by(del_flag=0).all()
  593. self.whitelist = [(phone.phone, phone.type) for phone in phones]
  594. self.logger.info("Whitelist updated: %s", self.whitelist)
  595. def get_whitelist(self):
  596. return self.whitelist
  597. def load_agent_monitor(self):
  598. with self.app.app_context():
  599. agents = AgentMonitor.query.filter_by(check_state=0, service_state=2).all()
  600. agent_nums = [agent.agent_num for agent in agents]
  601. return agent_nums
  602. def get_bucket(self, custom_uuid=None):
  603. custom_uuid == custom_uuid if custom_uuid else str(uuid.uuid4())
  604. random_id = abs(mmh3.hash(custom_uuid))
  605. if len(self.buckets) <= 0:
  606. raise BizException(BizErrorCode.RECORD_NOT_EXIST_ERROR)
  607. for bucket in self.buckets:
  608. num = (random_id % 100 + 100) %100
  609. if bucket.lower <= num < bucket.upper:
  610. return bucket
  611. return self.buckets[0]
  612. def update_bucket(self):
  613. with self.app.app_context():
  614. buckets = Bucket.query.filter_by(eid='001').all()
  615. self.buckets = buckets
  616. self.logger.info("bucket updated: %s", self.buckets)
  617. def start(self, HOST='0.0.0.0', PORT=8084, agent=None):
  618. # HOST, PORT = "0.0.0.0", 8084
  619. # 创建一个 TCP 服务器
  620. with self.CustomTCPServer((HOST, PORT), self.ESLRequestHandler, agent, self.app) as server:
  621. print(f"ESL server listening on {HOST}:{PORT}")
  622. server.serve_forever()