Ilia Miheev
5b62112ab2
Бот вынесен из корня в src/bot_qr.py. Строки интерфейса вынесены в language.json, чтобы не хардкодить их в коде. В requirements.txt обновлены зависимости под новый запуск.
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
from dotenv import load_dotenv
|
|
from sys import exit as sexit
|
|
import os
|
|
import json
|
|
|
|
import telebot
|
|
import qrcode
|
|
|
|
# Загрузить переменные из файла .env и language.json
|
|
load_dotenv()
|
|
with open('./src/language.json', 'r', encoding='utf-8') as file:
|
|
language = json.load(file)
|
|
|
|
# Получить значение переменной
|
|
API_TOKEN = os.getenv("API_TOKEN")
|
|
|
|
if API_TOKEN is None or API_TOKEN == "":
|
|
print(language['no_token'])
|
|
sexit(1) # Завершаем программу с кодом ошибки 1
|
|
|
|
|
|
bot = telebot.TeleBot(API_TOKEN)
|
|
photoName = None
|
|
|
|
@bot.message_handler(commands=['start'])
|
|
def start(mess):
|
|
bot.send_message(mess.chat.id, language['start'], parse_mode='html')
|
|
|
|
|
|
def gen_qr(text, id):
|
|
global photoName
|
|
photoName = f"qrcode_{id}.png"
|
|
# Создаем экземпляр класса QRCode
|
|
qr = qrcode.QRCode(
|
|
version=1,
|
|
error_correction=qrcode.constants.ERROR_CORRECT_H,
|
|
box_size=10,
|
|
border=1,
|
|
)
|
|
# Добавляем данные в QR-код
|
|
qr.add_data(text)
|
|
qr.make(fit=True)
|
|
# Создаем изображение QR-кода
|
|
img = qr.make_image(fill_color="black", back_color="white")
|
|
# Сохраняем изображение QR-кода
|
|
img.save(photoName)
|
|
|
|
|
|
@bot.message_handler(content_types=['text'])
|
|
def generate_qr_code(mess):
|
|
gen_qr(mess.text, mess.from_user.id)
|
|
text = language['qr_info'].format(
|
|
text=mess.text,
|
|
words=len(mess.text.split()),
|
|
chars=len(mess.text)
|
|
)
|
|
with open(photoName, 'rb') as qr_photo:
|
|
bot.send_photo(mess.chat.id, qr_photo)
|
|
bot.send_message(mess.chat.id, text, parse_mode='html')
|
|
os.remove(photoName)
|
|
|
|
print(language['bot_started'])
|
|
bot.polling()
|
|
print(language['bot_stopped'])
|