Убрал использование БД на потом; почистил лишние комментарии; переименовал главный файл в просто «bot.py»
This commit is contained in:
parent
0c1f7efb82
commit
e2a06a6219
1 changed files with 4 additions and 51 deletions
|
@ -1,29 +1,19 @@
|
||||||
''' Интеграция Дискорда и ГитЛаба
|
''' Интеграция Дискорда и ГитЛаба
|
||||||
© Стешенко Артём и Зажигин Богдан 2023—2023 '''
|
© Стешенко Артём и Зажигин Богдан 2023—2023 '''
|
||||||
|
|
||||||
from os import environ, remove
|
from os import environ
|
||||||
# Получение переменных среды и удаление файлов —
|
# Получение переменных среды и удаление файлов —
|
||||||
# https://docs.python.org/3.12/library/os.html
|
# https://docs.python.org/3.12/library/os.html
|
||||||
import msgpack
|
|
||||||
# Бинарная сериализация, создание машинной базы данных —
|
|
||||||
# https://github.com/msgpack/msgpack-python/blob/main/README.md
|
|
||||||
import gitlab, discord
|
import gitlab, discord
|
||||||
# Обертка АПИ ГитЛаба — https://python-gitlab.readthedocs.io/en/latest и
|
# Обертка АПИ ГитЛаба — https://python-gitlab.readthedocs.io/en/latest и
|
||||||
# АПИ Дискорда — https://discordpy.readthedocs.io/en/latest
|
# АПИ Дискорда — https://discordpy.readthedocs.io/en/latest
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
try:
|
|
||||||
with open("database.msgpack", "rb") as database:
|
|
||||||
database_content = database.read()
|
|
||||||
database_spisok = msgpack.unpackb(database_content, strict_map_key=False)
|
|
||||||
except:
|
|
||||||
database_spisok = dict()
|
database_spisok = dict()
|
||||||
|
|
||||||
|
|
||||||
gitlab_instance = gitlab.Gitlab(url = 'https://gitlab.megu.one', private_token = environ.get("TOKEN_GITLAB")) # определение адреса и токена экземляра ГитЛаба
|
gitlab_instance = gitlab.Gitlab(url = 'https://gitlab.megu.one', private_token = environ.get("TOKEN_GITLAB")) # определение адреса и токена экземляра ГитЛаба
|
||||||
'''project = gitlab_instance.projects.get(13)''' # определение проекта в котором нужно создавать задачи
|
|
||||||
# TODO: Привязка экзепмляра и проекта к Дискорд Каналу в БД, а не в коде
|
|
||||||
|
|
||||||
|
|
||||||
intents = discord.Intents.default() # использовать требования по умолчанию
|
intents = discord.Intents.default() # использовать требования по умолчанию
|
||||||
|
@ -43,7 +33,7 @@ async def on_message(message): # обработка каждого сообще
|
||||||
issue_text = message.content.replace("/issue ","") # получение текста команды «issue»
|
issue_text = message.content.replace("/issue ","") # получение текста команды «issue»
|
||||||
if command('/issue'): # команда создания задачи на ГитЛабе
|
if command('/issue'): # команда создания задачи на ГитЛабе
|
||||||
if database_spisok.get(int(message.channel.id)):
|
if database_spisok.get(int(message.channel.id)):
|
||||||
project = gitlab_instance.projects.get(database_spisok.get(int(message.channel.id)))
|
project = gitlab_instance.projects.get(str(database_spisok.get(int(message.channel.id))))
|
||||||
if project.issues.create({'title': issue_text}):
|
if project.issues.create({'title': issue_text}):
|
||||||
await reply("Задача «" + issue_text + "» создана успешно, ^w^")
|
await reply("Задача «" + issue_text + "» создана успешно, ^w^")
|
||||||
else:
|
else:
|
||||||
|
@ -54,7 +44,6 @@ async def on_message(message): # обработка каждого сообще
|
||||||
if command('/project'):
|
if command('/project'):
|
||||||
try:
|
try:
|
||||||
database_spisok[int(message.channel.id)] = int(message.content.replace("/project ",""))
|
database_spisok[int(message.channel.id)] = int(message.content.replace("/project ",""))
|
||||||
open("database.msgpack", "wb").write(msgpack.packb(database_spisok))
|
|
||||||
await reply("данные сохранены ^w^")
|
await reply("данные сохранены ^w^")
|
||||||
except:
|
except:
|
||||||
await reply("не удалось сохранить id")
|
await reply("не удалось сохранить id")
|
||||||
|
@ -77,40 +66,4 @@ async def on_message(message): # обработка каждого сообще
|
||||||
await reply("====================================")
|
await reply("====================================")
|
||||||
await reply("список комманд которые я выполняю:\n/issue - создание задачи на gitlab\n/project - подключение id канала discord с id канала gitlab\n/remove - удаление id\n/show - показ id (к каждому каналу discord подключён отдельный id gitlab)\n/speak - я расскажу немного о себе (что сейчас и делаю)")
|
await reply("список комманд которые я выполняю:\n/issue - создание задачи на gitlab\n/project - подключение id канала discord с id канала gitlab\n/remove - удаление id\n/show - показ id (к каждому каналу discord подключён отдельный id gitlab)\n/speak - я расскажу немного о себе (что сейчас и делаю)")
|
||||||
|
|
||||||
'''if command('/save'):
|
discord_bot.run(str(environ.get("TOKEN_DISCORD"))) # авторизация бота по токену из среды и запуск
|
||||||
open("database.msgpack", "wb").write(msgpack.packb(database_spisok))'''
|
|
||||||
|
|
||||||
discord_bot.run(environ.get("TOKEN_DISCORD")) # авторизация бота по токену из среды и запуск
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: Регистрировать команды бота в Команды Приложения —
|
|
||||||
# https://discordpy.readthedocs.io/en/latest/interactions/api.html#application-commands:
|
|
||||||
#=======================================================================
|
|
||||||
|
|
||||||
'''tree_commands = discord.app_commands.CommandTree(discord_bot) # Объявление дерева команд бота
|
|
||||||
if message.content.startswith('/project'):
|
|
||||||
slovar.update({message.channel.id: message.content.replace("/project ","")})
|
|
||||||
await reply(slovar)
|
|
||||||
slovar = dict()
|
|
||||||
command_issue_extras = dict()
|
|
||||||
@tree_commands.command(name="issue", description="создать задачу на GitLab", nsfw=False, auto_locale_strings=False)
|
|
||||||
async def issue(interaction):
|
|
||||||
await interaction.response.send_message(f"Pong", ephemeral=True)
|
|
||||||
add_command(*command_issue, guild=None, guilds=None, override=True)
|
|
||||||
asyncio.run(sync(*command_issue, guild=None))'''
|
|
||||||
|
|
||||||
#=======================================================================
|
|
||||||
|
|
||||||
'''database_file = open("database.msgpack", "w")
|
|
||||||
database = msgpack.unpackb(database_file.read()) # \31123\123123\123132\213
|
|
||||||
# {1: a}, {2: b}
|
|
||||||
Ctrl+c = SINGal TERMinate
|
|
||||||
|
|
||||||
database[3] = "c"
|
|
||||||
# {1: a}, {2: b}, {3, c}
|
|
||||||
|
|
||||||
# /save
|
|
||||||
database_file.write(
|
|
||||||
msgpack.packb(database, use_bin_type=True) # \31123\123123\123132\213 + \123123123123
|
|
||||||
)
|
|
||||||
database_file.close()'''
|
|
Loading…
Add table
Reference in a new issue