321 lines
13 KiB
Python
321 lines
13 KiB
Python
# import json
|
||
import datetime
|
||
import os
|
||
import random
|
||
import string
|
||
import time
|
||
|
||
import requests
|
||
|
||
import classes.desinfection as des
|
||
import classes.proverka as prov
|
||
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 = []
|
||
comp = []
|
||
for answers in questions:
|
||
answer_chain = {"id": answers["id"], "questions": []} # answers['questions']
|
||
if answer_chain["id"] == 6146: # Цепочка без ответов
|
||
continue
|
||
for components in answers["questions"]:
|
||
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 = "Систематическая проверка подъездов МКД"
|
||
# Получаем адреса
|
||
addresses = get_addresses(headers["uid"])
|
||
# Если адреса получили
|
||
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)
|
||
# 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"
|
||
] # Еще не принятые задания
|
||
# print(address)
|
||
if len(exec_available) > 0:
|
||
for n in exec_available:
|
||
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"]
|
||
if name_from_tasks == proverka:
|
||
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:
|
||
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(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))
|