431 lines
20 KiB
Python
431 lines
20 KiB
Python
import re
|
||
import cv2
|
||
import io
|
||
import numpy as np
|
||
|
||
from aiogram import F, types, Router, Bot
|
||
from decimal import Decimal
|
||
from datetime import datetime
|
||
|
||
from aiogram.exceptions import TelegramBadRequest
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.types import Message, CallbackQuery
|
||
from aiogram.fsm.state import StatesGroup, State
|
||
from qreader import QReader
|
||
|
||
from handlers.get_check_info import NalogRuPython
|
||
from keyboards.inline import expenses, to_start, fill_check, check_failed, input_check_back1, input_check_back2, \
|
||
input_check_back3, input_check_back4, input_check_back5, input_check_back6
|
||
|
||
expenses_router = Router()
|
||
qreader = QReader()
|
||
|
||
|
||
go_back = [
|
||
'input_check_back1',
|
||
'input_check_back2',
|
||
'input_check_back3',
|
||
'input_check_back4',
|
||
'input_check_back5',
|
||
'input_check_back6'
|
||
]
|
||
|
||
|
||
class ExpensesState(StatesGroup):
|
||
""" Состояния """
|
||
choose_expense = State()
|
||
choose_other = State()
|
||
load_check = State()
|
||
|
||
|
||
class InputCheckState(StatesGroup):
|
||
""" Состояния """
|
||
input_date = State()
|
||
input_time = State()
|
||
input_sum = State()
|
||
input_fn = State()
|
||
input_fd = State()
|
||
input_fp = State()
|
||
|
||
|
||
def check_info(ticket):
|
||
date = ticket["operation"]["date"]
|
||
parsed_date = datetime.fromisoformat(date)
|
||
formatted_date = parsed_date.strftime("%d.%m.%Y %H:%M")
|
||
total = Decimal(ticket["operation"]["sum"]) / Decimal(100)
|
||
nds = Decimal(ticket["ticket"]["document"]["receipt"]["nds18"]) / Decimal(100)
|
||
dif = total - nds
|
||
data = {
|
||
"date": formatted_date,
|
||
"fd": ticket["query"]["documentId"],
|
||
"total": f"{total:.2f}",
|
||
"nds": f"{nds:.2f}",
|
||
"sum": f"{dif:.2f}"
|
||
}
|
||
print(data)
|
||
|
||
formatted_text = (
|
||
f"<b>Статус отчёта:</b> подтверждён ✅\n"
|
||
f"<b>Дата чека:</b> {data['date']}\n"
|
||
f"<b>ФД:</b> {data['fd']}\n"
|
||
f"<b>Итого:</b> {data['total']}\n"
|
||
f"<b>НДС:</b> {data['nds']}\n"
|
||
f"<b>Сумма без НДС:</b> {data['sum']}\n"
|
||
)
|
||
|
||
return formatted_text
|
||
|
||
|
||
def check_fake_info(data):
|
||
formatted_text = (
|
||
f"<b>Статус отчёта:</b> не подтвержден ❌\n"
|
||
f"<b>Дата чека:</b> {data['date']} {data['time']}\n"
|
||
f"<b>ФД:</b> {data['fd']}\n"
|
||
f"<b>Итого:</b> {data['sum']}\n"
|
||
)
|
||
|
||
return formatted_text
|
||
|
||
|
||
async def edit_prev_msg(data, bot: Bot, message, state: FSMContext):
|
||
if data["delkeyboard"]:
|
||
prev_msg = data["delkeyboard"]
|
||
try:
|
||
await bot.edit_message_reply_markup(chat_id=message.chat.id, message_id=prev_msg, reply_markup=None)
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
|
||
async def cmd_clear(message: Message, bot: Bot, report_id) -> None:
|
||
await bot.delete_messages(message.chat.id, list(range(max(1, message.message_id - 90, report_id + 1), message.message_id + 1)))
|
||
|
||
|
||
@expenses_router.callback_query(F.data == 'var2')
|
||
async def choose_expense(call: CallbackQuery, state: FSMContext):
|
||
await state.set_state(ExpensesState.choose_expense)
|
||
try:
|
||
await call.message.edit_text('Выберите тип расхода', reply_markup=expenses.as_markup())
|
||
except Exception:
|
||
await call.message.answer('Выберите тип расхода', reply_markup=expenses.as_markup())
|
||
|
||
|
||
@expenses_router.callback_query(ExpensesState.choose_expense, F.data == 'ex5')
|
||
async def add_category(call: CallbackQuery, state: FSMContext):
|
||
await state.update_data(ex_type=call.data)
|
||
await state.set_state(ExpensesState.choose_other)
|
||
try:
|
||
await call.message.edit_text('Введите тип расхода')
|
||
except Exception:
|
||
await call.message.answer('Введите тип расхода')
|
||
|
||
|
||
@expenses_router.message(ExpensesState.choose_other)
|
||
async def handle_post(message: Message, state: FSMContext):
|
||
await state.update_data(ex_type=message.text)
|
||
await state.set_state(ExpensesState.load_check)
|
||
await message.answer('Отправьте фото чека, чтобы на нем было видно QR-код')
|
||
|
||
|
||
@expenses_router.message(ExpensesState.choose_expense)
|
||
async def handle_post(message: Message, state: FSMContext):
|
||
await state.set_state(ExpensesState.choose_expense)
|
||
try:
|
||
await message.edit_text('Выберите тип расхода', reply_markup=expenses.as_markup())
|
||
except Exception:
|
||
await message.answer('Выберите тип расхода', reply_markup=expenses.as_markup())
|
||
|
||
|
||
@expenses_router.callback_query(ExpensesState.choose_expense)
|
||
async def load_check(call: CallbackQuery, state: FSMContext):
|
||
await state.update_data(ex_type=call.data)
|
||
await state.set_state(ExpensesState.load_check)
|
||
try:
|
||
await call.message.edit_text('Отправьте фото чека, чтобы на нем было видно QR-код')
|
||
except Exception:
|
||
await call.message.answer('Отправьте фото чека, чтобы на нем было видно QR-код')
|
||
|
||
|
||
@expenses_router.callback_query(lambda call: call.data in go_back)
|
||
async def go_to_prev_check_input(call: CallbackQuery, state: FSMContext):
|
||
msg = None
|
||
if call.data == 'input_check_back1':
|
||
await state.set_state(ExpensesState.load_check)
|
||
try:
|
||
msg = await call.message.edit_text('Отправьте фото чека, чтобы на нем было видно QR-код')
|
||
except Exception:
|
||
msg = await call.message.answer('Отправьте фото чека, чтобы на нем было видно QR-код')
|
||
elif call.data == 'input_check_back2':
|
||
await state.set_state(InputCheckState.input_date)
|
||
try:
|
||
msg = await call.message.edit_text('Введите дату с чека в формате: ДД.ММ.ГГГГ (например: 02.02.2020)',
|
||
reply_markup=input_check_back1.as_markup())
|
||
except Exception:
|
||
msg = await call.message.answer('Введите дату с чека в формате: ДД.ММ.ГГГГ (например: 02.02.2020)',
|
||
reply_markup=input_check_back1.as_markup())
|
||
elif call.data == 'input_check_back3':
|
||
await state.set_state(InputCheckState.input_time)
|
||
try:
|
||
msg = await call.message.edit_text('Введите время с чека в формате: ЧЧ:ММ (например: 15:30)',
|
||
reply_markup=input_check_back2.as_markup())
|
||
except Exception:
|
||
msg = await call.message.answer('Введите время с чека в формате: ЧЧ:ММ (например: 15:30)',
|
||
reply_markup=input_check_back2.as_markup())
|
||
elif call.data == 'input_check_back4':
|
||
await state.set_state(InputCheckState.input_sum)
|
||
try:
|
||
msg = await call.message.edit_text('Введите сумму с чека например 157.00 (157 рублей 00 копеек)',
|
||
reply_markup=input_check_back3.as_markup())
|
||
except Exception:
|
||
msg = await call.message.answer('Введите сумму с чека например 157.00 (157 рублей 00 копеек)',
|
||
reply_markup=input_check_back3.as_markup())
|
||
elif call.data == 'input_check_back5':
|
||
await state.set_state(InputCheckState.input_fn)
|
||
try:
|
||
msg = await call.message.edit_text('Введите ФН', reply_markup=input_check_back4.as_markup())
|
||
except Exception:
|
||
msg = await call.message.answer('Введите ФН', reply_markup=input_check_back4.as_markup())
|
||
elif call.data == 'input_check_back6':
|
||
await state.set_state(InputCheckState.input_fd)
|
||
try:
|
||
msg = await call.message.edit_text('Введите ФД', reply_markup=input_check_back5.as_markup())
|
||
except Exception:
|
||
msg = await call.message.answer('Введите ФД', reply_markup=input_check_back5.as_markup())
|
||
if msg:
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.callback_query(ExpensesState.load_check, F.data == 'fill_check')
|
||
async def input_check_data(call: CallbackQuery, state: FSMContext):
|
||
await state.set_state(InputCheckState.input_date)
|
||
try:
|
||
msg = await call.message.edit_text('Введите дату с чека в формате: ДД.ММ.ГГГГ (например: 02.02.2020)',
|
||
reply_markup=input_check_back1.as_markup())
|
||
except Exception:
|
||
msg = await call.message.answer('Введите дату с чека в формате: ДД.ММ.ГГГГ (например: 02.02.2020)', reply_markup=input_check_back1.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.message(InputCheckState.input_date)
|
||
async def input_check_time(message: Message, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
await edit_prev_msg(data, bot, message, state)
|
||
await state.update_data(delkeyboard=None)
|
||
|
||
date_text = message.text.strip()
|
||
|
||
if not re.match(r"^\d{2}\.\d{2}\.\d{4}$", date_text):
|
||
msg = await message.reply("Неверный формат даты. Введите дату в формате: ДД.ММ.ГГГГ (например: 02.02.2020)", reply_markup=input_check_back1.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
try:
|
||
datetime.strptime(date_text, "%d.%m.%Y")
|
||
except ValueError:
|
||
msg = await message.reply("Некорректная дата. Проверьте правильность ввода и попробуйте снова.", reply_markup=input_check_back1.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
await state.set_state(InputCheckState.input_time)
|
||
await state.update_data(date=message.text)
|
||
msg = await message.answer('Введите время с чека в формате: ЧЧ:ММ (например: 15:30)', reply_markup=input_check_back2.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.message(InputCheckState.input_time)
|
||
async def input_check_sum(message: Message, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
await edit_prev_msg(data, bot, message, state)
|
||
await state.update_data(delkeyboard=None)
|
||
|
||
time_text = message.text.strip()
|
||
|
||
if not re.match(r"^\d{2}:\d{2}$", time_text):
|
||
msg = await message.reply("Неверный формат времени. Введите время в формате: ЧЧ:ММ (например: 15:30)", reply_markup=input_check_back2.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
try:
|
||
datetime.strptime(time_text, "%H:%M")
|
||
except ValueError:
|
||
msg = await message.reply("Некорректное время. Проверьте правильность ввода и попробуйте снова.", reply_markup=input_check_back2.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
await state.set_state(InputCheckState.input_sum)
|
||
await state.update_data(time=message.text)
|
||
msg = await message.answer('Введите сумму с чека например 157.00 (157 рублей 00 копеек)', reply_markup=input_check_back3.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.message(InputCheckState.input_sum)
|
||
async def input_check_fn(message: Message, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
await edit_prev_msg(data, bot, message, state)
|
||
await state.update_data(delkeyboard=None)
|
||
|
||
sum_text = message.text.strip()
|
||
|
||
try:
|
||
sum_value = float(sum_text)
|
||
if sum_value < 0:
|
||
raise ValueError("Сумма не может быть отрицательной. Повторите ввод")
|
||
formatted_sum = f"{sum_value:.2f}"
|
||
|
||
except ValueError:
|
||
msg = await message.reply(
|
||
"Некорректный формат суммы. Введите сумму в виде целого числа или с двумя знаками после точки (например: 157 или 157.00).", reply_markup=input_check_back3.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
await state.set_state(InputCheckState.input_fn)
|
||
await state.update_data(sum=formatted_sum)
|
||
msg = await message.answer('Введите ФН', reply_markup=input_check_back4.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.message(InputCheckState.input_fn)
|
||
async def input_check_fd(message: Message, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
await edit_prev_msg(data, bot, message, state)
|
||
await state.update_data(delkeyboard=None)
|
||
|
||
fn_text = message.text.strip()
|
||
|
||
if not (fn_text.isdigit() and len(fn_text) == 16):
|
||
msg = await message.reply("Некорректный формат ФН. Убедитесь, что он состоит ровно из 16 цифр.", reply_markup=input_check_back4.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
await state.set_state(InputCheckState.input_fd)
|
||
await state.update_data(fn=message.text)
|
||
msg = await message.answer('Введите ФД', reply_markup=input_check_back5.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.message(InputCheckState.input_fd)
|
||
async def input_check_fp(message: Message, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
await edit_prev_msg(data, bot, message, state)
|
||
await state.update_data(delkeyboard=None)
|
||
|
||
fd_text = message.text.strip()
|
||
|
||
if not fd_text.isdigit():
|
||
msg = await message.reply("Некорректный формат ФД. Убедитесь, что он состоит из цифр.", reply_markup=input_check_back5.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
await state.set_state(InputCheckState.input_fp)
|
||
await state.update_data(fd=message.text)
|
||
msg = await message.answer('Введите ФП', reply_markup=input_check_back6.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
|
||
|
||
@expenses_router.message(InputCheckState.input_fp)
|
||
async def input_check_load_data(message: Message, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
await edit_prev_msg(data, bot, message, state)
|
||
await state.update_data(delkeyboard=None)
|
||
|
||
fp_text = message.text.strip()
|
||
|
||
if not fp_text.isdigit():
|
||
msg = await message.reply("Некорректный формат ФП. Убедитесь, что он состоит из цифр.", reply_markup=input_check_back6.as_markup())
|
||
await state.update_data(delkeyboard=msg.message_id)
|
||
return
|
||
|
||
await state.update_data(fp=message.text)
|
||
await state.set_state(ExpensesState.load_check)
|
||
data = await state.get_data()
|
||
|
||
date_obj = datetime.strptime(data['date'], "%d.%m.%Y")
|
||
time_obj = datetime.strptime(data['time'], "%H:%M").time()
|
||
|
||
combined_datetime = datetime.combine(date_obj.date(), time_obj)
|
||
iso_format = combined_datetime.strftime("%Y%m%dT%H%M")
|
||
|
||
sum_total = data['sum']
|
||
fn = data['fn']
|
||
fd = data['fd']
|
||
fp = data['fp']
|
||
query = f"t={iso_format}&s={sum_total}&fn={fn}&i={fd}&fp={fp}&n=1"
|
||
try:
|
||
client = NalogRuPython()
|
||
ticket = client.get_ticket(query)
|
||
ans = check_info(ticket)
|
||
report_id = data["report"] if 'report' in data else message.message_id - 90
|
||
await cmd_clear(message, bot, report_id)
|
||
report = await message.answer(ans, parse_mode="HTML", reply_markup=to_start.as_markup())
|
||
await state.update_data(report=report.message_id)
|
||
|
||
except Exception as e:
|
||
try:
|
||
await message.edit_text("Не удалось получить информацию о чеке\nПроверьте введенные данные и попробуйте снова\nЛибо можете сформировать отчёт на основе введённых данных", reply_markup=check_failed.as_markup())
|
||
except Exception:
|
||
await message.answer(
|
||
"Не удалось получить информацию о чеке\nПроверьте введенные данные и попробуйте снова\nЛибо можете сформировать отчёт на основе введённых данных",
|
||
reply_markup=check_failed.as_markup())
|
||
print(e)
|
||
return
|
||
|
||
|
||
@expenses_router.callback_query(ExpensesState.load_check, F.data == 'make_report')
|
||
async def make_fake_report(call: CallbackQuery, state: FSMContext, bot: Bot):
|
||
data = await state.get_data()
|
||
ans = check_fake_info(data)
|
||
report_id = data["report"] if 'report' in data else call.message.message_id - 90
|
||
await cmd_clear(call.message, bot, report_id)
|
||
try:
|
||
report = await call.message.edit_text(ans, parse_mode="HTML", reply_markup=to_start.as_markup())
|
||
await state.update_data(report=report.message_id)
|
||
except Exception:
|
||
report = await call.message.answer(ans, parse_mode="HTML", reply_markup=to_start.as_markup())
|
||
await state.update_data(report=report.message_id)
|
||
|
||
|
||
@expenses_router.message(ExpensesState.load_check)
|
||
async def handle_post(message: Message, state: FSMContext):
|
||
if not message.photo:
|
||
await message.answer('Неподдерживаемый формат, пожалуйста, отправьте фото')
|
||
return
|
||
|
||
loading = await message.answer('Пожалуйста подождите, идет получение данных о чеке...')
|
||
check_photo = message.photo[-1]
|
||
bot = message.bot
|
||
photo_bytes = io.BytesIO()
|
||
await bot.download(check_photo.file_id, photo_bytes)
|
||
photo_bytes.seek(0)
|
||
nparr = np.frombuffer(photo_bytes.read(), np.uint8)
|
||
image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
|
||
reader = QReader()
|
||
decoded_text = reader.detect_and_decode(image=image)
|
||
|
||
if not decoded_text or not decoded_text[0]:
|
||
try:
|
||
await message.edit_text("QR-код не распознан.\nМожете заполнить данные о чеке самостоятельно", reply_markup=fill_check.as_markup())
|
||
except Exception:
|
||
await bot.delete_message(chat_id=loading.chat.id, message_id=loading.message_id)
|
||
await message.answer("QR-код не распознан.\nМожете заполнить данные о чеке самостоятельно", reply_markup=fill_check.as_markup())
|
||
finally:
|
||
return
|
||
else:
|
||
try:
|
||
client = NalogRuPython()
|
||
ticket = client.get_ticket(decoded_text[0])
|
||
ans = check_info(ticket)
|
||
await bot.delete_message(chat_id=loading.chat.id, message_id=loading.message_id)
|
||
data = await state.get_data()
|
||
report_id = data["report"] if 'report' in data else message.message_id - 90
|
||
await cmd_clear(message, bot, report_id)
|
||
report = await message.answer(ans, parse_mode="HTML", reply_markup=to_start.as_markup())
|
||
await state.update_data(report=report.message_id)
|
||
except Exception as e:
|
||
try:
|
||
await message.edit_text("Не удалось получить информацию о чеке\nМожете заполнить данные о чеке самостоятельно", reply_markup=fill_check.as_markup())
|
||
except Exception:
|
||
await bot.delete_message(chat_id=loading.chat.id, message_id=loading.message_id)
|
||
await message.answer("Не удалось получить информацию о чеке\nМожете заполнить данные о чеке самостоятельно", reply_markup=fill_check.as_markup())
|
||
finally:
|
||
print(e)
|
||
return
|