0
0
mirror of https://gitflic.ru/project/maks1ms/ocab.git synced 2024-12-24 00:33:05 +03:00
This commit is contained in:
armatik 2023-07-09 16:15:06 +03:00
parent c5a9c6fd6f
commit 348c9b9619
6 changed files with 57 additions and 0 deletions

View 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

View File

36
src/TelegramBot/main.py Normal file
View 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
View File

@ -0,0 +1,7 @@
[Telegram]
token="....."
admin_password="Test_pass" # ONLY FOR ADDING CHAT TO ADMIN LIST!
[OpenAI]
token="....."
chat_model="....."