import requests import random import string # import json import datetime import time import os from classes.users import Users import classes.desinfection as des import classes.proverka as prov 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_задания""" desinfection = 'Дезинфекция подъездов (МОП)' proverka = 'Систематическая проверка подъездов МКД' proverka_monthly = 'Ежемесячная проверка подъездов МКД' # Получаем адреса addresses = get_addresses(headers['uid']) # Если адреса получили # {"filters":[{"value":{"distance":"50000","lat":55.81999969482422,"lon":37.380001068115234},"filter_attribute":10,"id":6,"name":"Локация","type":"distance"}]} if isinstance(addresses, list): session_t = session session_t.headers.update( {'referer': 'https://knd.mosreg.ru/executions', 'x-platform': 'android', 'accept': '*/*' } ) for address in addresses: # change '-' for '/' # need split enterance from address addr = address.replace("Чаиковского", "Чайковского") addr = address.replace("Маиданово", "Майданово") search = addr.replace("_", "/")[:-4] # print(search) # data = {'search': search, # 'filters': [{'value':{ # 'distance': 25000, # 'lat': 55.81999969482422, # 'lon': 37.380001068115234 # }, # 'filter_attribute': 10, # 'id': 6, # 'name': 'Локация', # 'type': 'distance'}]} data = {'search': search} tasks = session_t.post('https://knd.mosreg.ru//api/v1/actor/executions', headers=headers, data=data ) json_tasks = tasks.json() get_t = False # Если есть доступное задание с таким адресом exec_assigned = json_tasks['executions_assigned'] # Уже полученные задания # print(exec_assigned) exec_available = json_tasks['executions_available'] # Еще не принятые задания # print(exec_available) # print(address) 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') name_from_tasks = execution['name'] # print(name_from_tasks) if name_from_tasks == proverka or name_from_tasks == proverka_monthly: enterance = execution['dimensions'][2].get('entity_value_name') if address_from_tasks == 'Клин г, '+ addr.replace("_", "/")[:-4] and \ enterance[-1] == address[-2]: print('\n') print(name_from_tasks) print('\n') task = { 'id': execution['execution_id'], 'address': execution['dimensions'][0].get('entity_value_name'), 'enterance': execution['dimensions'][2].get('entity_value_name'), 'coord': coord } if name_from_tasks == desinfection: pass # des.assign_task(session, headers, task, address) elif name_from_tasks == proverka or name_from_tasks == proverka_monthly: prov.assign_task(session, headers, task, address) else: pass 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: get_task(session, headers) else: print(f"Отказ в авторизации для пользователя {user['email']}!") time.sleep(20) 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))