import requests import xmpp from xmpp import cli import config from BotClass import JabberBot import json #раскомментировать, когда перейдем на разные пароли import hashlib from datetime import datetime import time import threading #test import jsonAndRequest as jsreq global tasklist def log_handler(): while True: with open('log.txt', 'a') as log_file: log_file.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} - Function is still running...\n") time.sleep(3600) botname="service_request_booked" operating_status='211' JID=botname+"@"+config.JSERVER PORT=config.PORT PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest() print(JID, PASSWORD) #таким образом хранится список jid, от которых можно получать сообщения этому боту listen_to=["799999999990@ej.sharix-app.org"] #тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки proceed_to=["service_request_assigned@ej.sharix-app.org","service_request_pending@ej.sharix-app.org", "service_request_forcemajeure@ej.sharix-app.org"] #максимальное значение попыток обработать заказ idle_value=10 def log_handler(): while True: with open(f'{botname}.txt', 'a') as log_file: log_file.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} - {botname} is still running...\n") time.sleep(3600) # обработчик входящих сообщений def message_handler(conn, mess): text = mess.getBody()#текст сообщения боту user = mess.getFrom()#отправитель сообщения if (str(user).split("/")[0]) in listen_to: # print (str(user).split("/")[0]) if text is not None: try : orderObj = jsreq.jsonToOrder(text) tasklist.append(orderObj) bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n") except: bot.proceed_status(proceed_to[-1], text) bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n") def service_request_booked_wait(period,localtask): bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ ожидает\n") time.sleep(period) #ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных localtask.title=localtask.title+"1" #обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется bot.proceed_status(proceed_to[0],localtask) bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ подождал и возвращен в очередь\n") def service_request_booked_process(localtask): bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ в обработчике\n") if not localtask.provider: answer,provider,service = set_provider(localtask.id) # localtask.service =service.id localtask.service = service['id'] if set_provider(localtask.id): bot.bot_log(botname+" "+str(datetime.now())+" "+"Водитель успешно установлен\n") else: bot.bot_log(botname+" "+str(datetime.now())+" "+"'Ошибка водитель не был установлен'\n") localtask.provider = provider current_time = datetime.now() # if localtask.asap or (current_time - datetime.fromisoformat(localtask.time_start_predicted) < timedelta(hours=3)): if localtask.asap: localtask.status = '212' print(jsreq.change_to_task(localtask,token)) bot.proceed_status(proceed_to[0], localtask) bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n") else: localtask.status = '220' print(jsreq.change_to_task(localtask,token)) bot.proceed_status(proceed_to[1], localtask) bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на pending\n") # Проверка срочности заказа # if order.asap: # #Заглушка назначение исполнителя системой # if set_provider(order.ticket, order.provider) == "success": # bot.proceed_status(proceed_to[0], localtask) # else: # bot.bot_log(botname+" "+str(datetime.now())+" "+"Ошибка назначения исполнителя\n") # else: # localtask.status = 220 # bot.proceed_status(proceed_to[0], localtask) # print(localtask) # bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n") # Проверка срочности заказа def get_balance(client): headers = {'Authorization': f'Token {token}'} # Передаем id пользователя и получаем его баланс response = requests.get(config.API_URL+f"/api/example/{client}", headers=headers) print(response.json()) ordertext=response.json() if not ordertext: print("Empty array") else: print(ordertext[0], "PROCESSING") try: balance = ordertext[0].get('balance') return balance except Exception as ex: print(ex) #set provider? def choose_provider(auth_token,geo): providers=requests.get(config.API_URL+"/dbsynce/api/provider/",headers=auth_token) provider=providers.json()[0]['id'] return provider def set_provider(localtask_id): token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"my/api/v1/auth/token/login/") headers = {'Authorization': f'Token {token}'} provider = choose_provider(headers,'geoposition') service = requests.get(config.API_URL+f"/dbsynce/api/service/?id_provider={provider}", headers=headers) service=service.json()[0] data = {'provider':int(provider)} # print(data) # print(str(config.API_URL+f"/dbsynce/api/orders/{localtask_id}/")) response = requests.patch(config.API_URL+f"/dbsynce/api/orders/{localtask_id}/", headers=headers, json=data) return response ,provider , service # if not ordertext: # print("Empty array") # else: # print(ordertext[0], "PROCESSING") # try: # message = ordertext[0].get('message') # return message # except Exception as ex: # print(ex) def service_request_booked(): log_thread = threading.Thread(target=log_handler) log_thread.daemon = True log_thread.start() while (len(tasklist)): localtask=tasklist.pop(0) if (str(localtask.status)!=str(operating_status)): # bot.proceed_status(proceed_to[-1],localtask) #временная проверка отправки print(localtask) localtask.status='239' jsreq.change_to_task(localtask,token) bot.proceed_status(proceed_to[-1],localtask) bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ не по адресу, перенаправляем на форсмажор\n") continue else: bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n") t1=threading.Thread(target=service_request_booked_process, args=([localtask])) t1.start() # elif (localtask.provider==T): # t1=threading.Thread(target=service_request_booked_wait, args=(5,localtask)) # t1.start() # if (localtask.title=="101"): # t1=threading.Thread(target=service_request_booked_process, args=(10,localtask)) # t1.start() #если никакие обработчики не подошли - отправляем обратно в очередь # bot.proceed_status(proceed_to[0],localtask) # print("Заказ возвращен в очередь\n") # print(tasklist) #Авторизация и запуск бота #пустой список заказов tasklist = [] # #надо инициализировать tasklist при запуске из API token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"my/api/v1/auth/token/login/") # #ВАЖНО! Надо еще фильтровать список по статусу обработки заявки tasklist_dbsynce=jsreq.requestGetList(token, config.API_URL+"/dbsynce/api/orders/") ###Должна быть проверка между двумя # # print (len(tasklist)) bot = JabberBot(JID, PASSWORD, PORT) bot.register_handler('message', message_handler) bot.start(service_request_booked) # print(set_provider('localtask'))