mirror of
https://gitflic.ru/project/maks1ms/ocab.git
synced 2024-12-24 00:33:05 +03:00
new code
This commit is contained in:
parent
c5a9c6fd6f
commit
348c9b9619
14
src/TelegramBot/MessageHandler.py
Normal file
14
src/TelegramBot/MessageHandler.py
Normal file
@ -0,0 +1,14 @@
|
||||
# Получение сообщений в чате, и запись их в базу данных
|
||||
from aiogram import types
|
||||
|
||||
from src.TelegramBot.main import dp
|
||||
|
||||
|
||||
@dp.message_handler()
|
||||
async def send(message: types.Message):
|
||||
# Получение сообщений в чате, и запись их в базу данных
|
||||
# Проверка на то, что сообщение не пустое и не отправлено в чате содержащим ChatType = 1 в базе данных chatlist
|
||||
if (message.chat.type == "group" or message.chat.type == "supergroup") and \
|
||||
message.text != '' and message.text != ' ': return None
|
||||
else:
|
||||
# Проверка статуса ChatType в базе данных chatlist
|
36
src/TelegramBot/main.py
Normal file
36
src/TelegramBot/main.py
Normal file
@ -0,0 +1,36 @@
|
||||
# Файл с инициализацией всех процессов телеграмм бота и запуском API OpenAI
|
||||
|
||||
# Импорт библиотек
|
||||
import os
|
||||
import openai
|
||||
import configparser
|
||||
import sqlite3
|
||||
|
||||
from aiogram import Bot, Dispatcher, executor, types
|
||||
from aiogram.contrib.fsm_storage.memory import MemoryStorage
|
||||
|
||||
from MessageHandler import *
|
||||
|
||||
|
||||
# Импорт переменных из файла .ini
|
||||
config = configparser.ConfigParser()
|
||||
config.read(os.path.join(os.path.dirname(__file__), 'config.ini'))
|
||||
|
||||
TOKEN = config['Telegram']['token']
|
||||
OPENAI_API_KEY = config['OpenAI']['api_key']
|
||||
|
||||
# Инициализация бота
|
||||
|
||||
bot = Bot(token=TOKEN)
|
||||
dp = Dispatcher(bot, storage=MemoryStorage())
|
||||
|
||||
# Инициализация API OpenAI
|
||||
|
||||
openai.api_key = OPENAI_API_KEY
|
||||
|
||||
# Инициализация базы данных OCAB_DB в папке DataBase/OCAB_DB.db
|
||||
database = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'DataBase/OCAB_DB.db'))
|
||||
cursor = database.cursor()
|
||||
|
||||
if __name__ == '__main__':
|
||||
executor.start_polling(dp, skip_updates=True)
|
7
src/config.ini
Normal file
7
src/config.ini
Normal file
@ -0,0 +1,7 @@
|
||||
[Telegram]
|
||||
token="....."
|
||||
admin_password="Test_pass" # ONLY FOR ADDING CHAT TO ADMIN LIST!
|
||||
|
||||
[OpenAI]
|
||||
token="....."
|
||||
chat_model="....."
|
Loading…
Reference in New Issue
Block a user