From 99d8c5fb0b88c91f88a7a8b341610de59c86fd87 Mon Sep 17 00:00:00 2001 From: Noretsa Date: Fri, 26 Jun 2020 07:58:38 +0300 Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=20=D0=B1=D0=BE=D1=82=20=D0=B4=D0=BB=D1=8F=20=D0=BE=D0=B1?= =?UTF-8?q?=D1=80=D0=B0=D0=B1=D0=BE=D1=82=D0=BA=D0=B8=20=D0=BF=D1=80=D0=B8?= =?UTF-8?q?=D0=BD=D1=8F=D1=82=D1=8B=D1=85=20=D0=B7=D0=B0=D0=B4=D0=B0=D0=BD?= =?UTF-8?q?=D0=B8=D0=B9=20=D0=B8=20=D0=B2=20=D0=B0=D0=B4=D1=80=D0=B5=D1=81?= =?UTF-8?q?=D0=B5=20=D0=BC=D0=B5=D0=BD=D1=8F=D0=B5=D1=82=D1=81=D1=8F=20'-'?= =?UTF-8?q?=20=D0=BD=D0=B0=20'/'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- knd_bot.py | 44 ++++---- knd_bot_assigned.py | 267 ++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 286 insertions(+), 25 deletions(-) create mode 100644 knd_bot_assigned.py diff --git a/knd_bot.py b/knd_bot.py index a06665a..64b15f2 100644 --- a/knd_bot.py +++ b/knd_bot.py @@ -1,21 +1,13 @@ import requests import random import string -import json +# import json import datetime import time import os from classes.users import Users -# anwer_text = '{"question_chains":[{"id":6145,"questions":[{"id":6559,"question_components":[{"id":18850,"answer":[80634165],"answer_status":null,"end_time":1591866979,"start_time":1591866978},{"id":18851,"answer":{"location":[56.439337,36.557897]},"answer_status":null,"start_time":1591866873,"end_time":1591867165},{"id":18852,"answer":"1","answer_status":null,"start_time":1591867063,"end_time":1591867063}]}]},{"id":6147,"questions":[{"id":6561,"question_components":[{"id":18856,"answer":77,"answer_status":null,"end_time":1591867067,"start_time":1591867067},{"id":18857,"answer":[80639138],"answer_status":null,"end_time":1591867186,"start_time":1591867185}]}]},{"id":6147,"questions":[{"id":6561,"question_components":[{"id":18856,"answer":77,"answer_status":null,"end_time":1591867200,"start_time":1591867200},{"id":18857,"answer":[80639640],"answer_status":null,"end_time":1591867206,"start_time":1591867205}]}]},{"id":6147,"questions":[{"id":6561,"question_components":[{"id":18856,"answer":77,"answer_status":null,"end_time":1591867211,"start_time":1591867211},{"id":18857,"answer":[80640010],"answer_status":null,"end_time":1591867220,"start_time":1591867219}]}]},{"id":6147,"questions":[{"id":6561,"question_components":[{"id":18856,"answer":77,"answer_status":null,"end_time":1591867223,"start_time":1591867223},{"id":18857,"answer":[80640333],"answer_status":null,"end_time":1591867233,"start_time":1591867232}]}]}]}' - - - - - - - def get_questions(id_task, session, headers): url = 'https://knd.mosreg.ru//api/v1/executions/'+str(id_task)+'/questions' questions = session.get(url, headers=headers) @@ -23,7 +15,6 @@ def get_questions(id_task, session, headers): return json_quest['selected_chains'] - def send_image(session, headers, url, photo): """Отправляем фото на сервер. Сервер должен вернуть id и url.""" files = { @@ -72,11 +63,11 @@ def complete_task(session, headers, task, address): if len(data) == len(photos): photo_from_photos = photos.pop(0) img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) - count_send_photo = 0 - while img['id'] is None: - count_send_photo += 1 - print("Попытка № " + str(count_send_photo) + "отправить фото" + str(photo_from_photos)) - img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) + # count_send_photo = 0 + # while img['id'] is None: + # count_send_photo += 1 + # print("Попытка № " + str(count_send_photo) + "отправить фото" + str(photo_from_photos)) + # img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) first = {"id": 18850, "answer": [img['id']], 'answer_status': None, 'start_time': int(time.time())} time.sleep(2) first.update({'end_time': int(time.time())}) @@ -89,11 +80,11 @@ def complete_task(session, headers, task, address): #img = send_image(session, headers, task['id'], photos.pop(0)) photo_from_photos = photos.pop(0) img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) - count_send_photo = 0 - while img['id'] is None: - count_send_photo += 1 - print("Попытка № " + str(count_send_photo) + " отправить фото" + str(photo_from_photos)) - img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) + # count_send_photo = 0 + # while img['id'] is None: + # count_send_photo += 1 + # print("Попытка № " + str(count_send_photo) + " отправить фото" + str(photo_from_photos)) + # img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) first = {"id": 18856, "answer": 77, 'answer_status': None, 'start_time': int(time.time())} time.sleep(2) first.update({'end_time': int(time.time())}) @@ -130,7 +121,7 @@ def get_photo(address, email): return photos else: for images in os.listdir(main_path): - if images.endswith(".jpg") or images.endswith(".jpeg"): + if images.endswith(".jpg") or images.endswith(".jpeg") or images.endswith(".JPG"): photos.append(main_path + '/' + images) return photos @@ -151,8 +142,6 @@ def get_addresses(email): def assign_task(session, headers, task, address): -# complete_task(session, headers, task, address) - # DON'T START PROGRAM!!!! url = 'https://knd.mosreg.ru//api/v1/executions/'+str(task['id'])+'/assign' task_assign = session.put(url, headers=headers) if task_assign.status_code == 200: @@ -184,6 +173,8 @@ def get_task(session, headers): } ) for address in addresses: + # change '-' for '/' + address = address.replace("-", "/") # need split enterance from address search = address[:-4] # print(search) @@ -195,8 +186,10 @@ def get_task(session, headers): json_tasks = tasks.json() get_t = False # Если есть доступное задание с таким адресом - if len(json_tasks['executions_available']) > 0: - for n in json_tasks['executions_available']: + exec_assigned = json_tasks['executions_assigned'] + exec_available = json_tasks['executions_available'] + if len(exec_available) > 0: + for n in exec_available: coord = n['coord'] enterances = n['tasks'] for execution in enterances: @@ -265,6 +258,7 @@ def main(): if __name__ == "__main__": main() + k = input("Press ENTER for exit") def random_string(stringLength): diff --git a/knd_bot_assigned.py b/knd_bot_assigned.py new file mode 100644 index 0000000..bac7885 --- /dev/null +++ b/knd_bot_assigned.py @@ -0,0 +1,267 @@ +import requests +import random +import string +# import json +import datetime +import time +import os +from classes.users import Users + + +def get_questions(id_task, session, headers): + url = 'https://knd.mosreg.ru//api/v1/executions/'+str(id_task)+'/questions' + questions = session.get(url, headers=headers) + json_quest = questions.json() + return json_quest['selected_chains'] + + +def send_image(session, headers, url, photo): + """Отправляем фото на сервер. Сервер должен вернуть id и url.""" + files = { + 'image': (photo[-5:], open(photo, 'rb')), + } + # print(files) + img = session.post('https://knd.mosreg.ru//api/v1/executions/'+str(url)+'/images', headers=headers, files=files) + if img.status_code == 200: + print(f"Фото {photo} успешно отправлено") + return img.json() + else: + print(f"Не удалось загрузить фото на сервер по заданию {url}, статус ошибки {img.status_code}! Повторная попытка!") + second_img = send_image(session, headers, url, photo) + return second_img + + +def complete_task(session, headers, task, address): + photos = get_photo(address, headers['uid']) + if len(photos) > 0: + photos.sort() + print(task['id']) + id_task = task['id'] + questions = get_questions(id_task, session, headers) + data = [] + for answers in questions: + answer_chain = { + 'id': answers['id'], + 'questions': []#answers['questions'] + } + if answer_chain['id'] == 6146: # Цепочка без ответов + continue + for components in answers['questions']: + comp = [] + q_c = { + 'id': components['id'], + 'question_components': [] + } + if q_c['id'] == 6560: # Вопрос без ответа + continue + comp.append(q_c) + answer_chain.update({'questions': comp}) + data.append(answer_chain) + if len(data) == len(photos): + for chain in data: + for item in chain['questions']: + if len(data) == len(photos): + photo_from_photos = photos.pop(0) + img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) + # count_send_photo = 0 + # while img['id'] is None: + # count_send_photo += 1 + # print("Попытка № " + str(count_send_photo) + "отправить фото" + str(photo_from_photos)) + # img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) + first = {"id": 18850, "answer": [img['id']], 'answer_status': None, 'start_time': int(time.time())} + time.sleep(2) + first.update({'end_time': int(time.time())}) + second = {"id": 18851, "answer": {"location": task['coord']}, 'answer_status': None, 'start_time': int(time.time())} + time.sleep(2) + second.update({'end_time': int(time.time())}) + item['question_components'].append(first) + item['question_components'].append(second) + else: + #img = send_image(session, headers, task['id'], photos.pop(0)) + photo_from_photos = photos.pop(0) + img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) + # count_send_photo = 0 + # while img['id'] is None: + # count_send_photo += 1 + # print("Попытка № " + str(count_send_photo) + " отправить фото" + str(photo_from_photos)) + # img = send_image(session, headers, task['id'], photo_from_photos)#photos.pop(0)) + first = {"id": 18856, "answer": 77, 'answer_status': None, 'start_time': int(time.time())} + time.sleep(2) + first.update({'end_time': int(time.time())}) + second = {"id": 18857, "answer": [img['id']], 'answer_status': None, 'start_time': int(time.time())} + time.sleep(2) + second.update({'end_time': int(time.time())}) + item['question_components'].append(first) + item['question_components'].append(second) + ans = {"question_chains": data} + # print(ans) + session_t = session + session_t.headers.update( + { + 'referer': 'https://knd.mosreg.ru/executions'+str(task['id'])+'/solve', + 'x-platform': 'wb', + 'accept': '*/*' + } + ) + complete = session_t.post('https://knd.mosreg.ru//api/v1/executions/'+str(task['id'])+'/answers', headers=headers, json=ans) + if complete.status_code == 200: + print(f'Задание по адресу {address} выполнено успешно!') + else: + print(f'Задание по адресу {address} не выполненно! Статус ошибки {complete.status_code}') + else: + print(f"Несоответствие количества вопросов и количества фотографий по адресу: {address}") +# send_execution(task,session, headers) + + +def get_photo(address, email): + photos = [] + month = datetime.datetime.now().month + main_path = os.path.join(os.path.abspath('.'), 'data', email, str(month), address) + if not os.path.exists(main_path): + return photos + else: + for images in os.listdir(main_path): + if images.endswith(".jpg") or images.endswith(".jpeg") or images.endswith(".JPG"): + photos.append(main_path + '/' + images) + return photos + + +def get_addresses(email): + addresses = [] + month = datetime.datetime.now().month + main_path = os.path.join(os.path.abspath('.'), 'data') + path = os.walk(os.path.join(main_path, email, str(month))) + if not os.path.exists(os.path.join(main_path, email, str(month))): + return f"Не найдена папка с месяцем у пользователя {email}! Проверьте структуру каталога!" + for d, dirs, files in path: + addresses.append(dirs) + if len(addresses[0]) > 0: + return addresses[0] + else: + return f"Ни одного адреса для пользователя {email} не найдено! Добавьте фото и запустите программу снова!" + + +def assign_task(session, headers, task, address): + url = 'https://knd.mosreg.ru//api/v1/executions/'+str(task['id'])+'/assign' + task_assign = session.put(url, headers=headers) + if task_assign.status_code == 200: + print(f'Получили задани по адресу: {address}') + complete_task(session, headers, task, address) + else: + print(f"Не смогли получить задание {address} {task['id']}!") + + +def get_task(session, headers): + """Получаем все доступные задания. Если в переменную search добавить адрес, + то будет искать этот адрес. Он попадет в + json_tasks['executions_available'], + так что надо понять откуда берем задания. Отсюда можно возвращать + все параметры задания: + 1 координаты + 2 адрес + 3 подъезд + 4 id_задания""" + # Получаем адреса + addresses = get_addresses(headers['uid']) + # Если адреса получили + if isinstance(addresses, list): + session_t = session + session_t.headers.update( + {'referer': 'https://knd.mosreg.ru/executions', + 'x-platform': 'wb', + 'accept': '*/*' + } + ) + for address in addresses: + # change '-' for '/' + address = address.replace("-", "/") + # need split enterance from address + search = address[:-4] + # print(search) + data = {'search': search} + tasks = session_t.post('https://knd.mosreg.ru//api/v1/executions', + headers=headers, + data=data + ) + json_tasks = tasks.json() + get_t = False + # Если есть доступное задание с таким адресом + exec_assigned = json_tasks['executions_assigned'] + exec_available = json_tasks['executions_available'] + if len(exec_assigned) > 0: + for n in exec_assigned: + coord = n['coord'] + enterances = n['tasks'] + for execution in enterances: + address_from_tasks = execution['dimensions'][0].get('entity_value_name') + # print(address_from_tasks) + # print('Клин г, '+address[:-4]) + enterance = execution['dimensions'][2].get('entity_value_name') + # print(enterance) + # print(address[-2]) + if address_from_tasks == 'Клин г, '+address[:-4] and \ + enterance[-1] == address[-2]: + print(f'Начали выполнение задания по адресу: {address}') + task = { + 'id': execution['execution_id'], + 'address': execution['dimensions'][0].get('entity_value_name'), + 'enterance': execution['dimensions'][2].get('entity_value_name'), + 'coord': coord + } + # Принимаем задание + assign_task(session, headers, task, address) + get_t = True + if not get_t: + print(f"Нет доступных заданий для по адресу: {address}!") + # Если нет задания с таким адресом + else: + print(f"Нет доступных заданий для пользователя {headers['uid']} по адресу: {address}") + # Если адреса не получили выводим ошибку + else: + print(addresses) + + +def main(): + users = Users() + logins = users.get_passwords() + if isinstance(logins, list): + url = 'https://knd.mosreg.ru//api/v1/auth/sign_in' + for user in logins: + session = requests.Session() + login = { + 'email': user['email'], + 'password': user['password'] + } + response = session.post(url, data=login) + headers = { + 'client': response.headers.get('client'), + 'Access-token': response.headers.get('Access-token'), + 'uid': response.headers.get('uid'), + } + if response.status_code == 200: + # task = { + # 'id': 23799105, + # 'address': 'Решетниково рп, ОПМС-1 проезд, 13', + # 'enterance': 6, + # 'coord': [56.442503, 36.5596493] + # } + # address = 'Решетниково рп, ОПМС-1 проезд, 13, 6п' + # assign_task(session, headers, task, address) + # break + get_task(session, headers) + else: + print(f"Отказ в авторизации для пользователя {user['email']}!") + time.sleep(1) + else: + print(users.get_passwords()) + + +if __name__ == "__main__": + main() + k = input("Press ENTER for exit") + + +def random_string(stringLength): + letters = string.ascii_letters + digits = string.digits + return ''.join(random.choice(letters+digits) for i in range(stringLength))