mirror of
https://gitflic.ru/project/alt-gnome/karkas.git
synced 2024-12-24 16:44:46 +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