zaboal/discord-gitlab
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Убрал использование БД на потом; почистил лишние комментарии; переименовал главный файл в просто «bot.py»

Browse source
This commit is contained in:
bogdan zažigin 2023-04-19 20:49:12 +03:00
parent 0c1f7efb82
commit e2a06a6219
Signed by: zaboal
GPG key ID: C9918166D559FE8A
1 changed files with 4 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
source/gitlab-discord_bot/gitlab-discord_bot.py → source/gitlab-discord_bot/bot.py
View file

@ -1,29 +1,19 @@
''' Интеграция Дискорда и ГитЛаба
© Стешенко Артём и Зажигин Богдан 2023—2023 '''
from os import environ, remove
from os import environ
# Получение переменных среды и удаление файлов —
# https://docs.python.org/3.12/library/os.html
import msgpack
# Бинарная сериализация, создание машинной базы данных —
# https://github.com/msgpack/msgpack-python/blob/main/README.md
import gitlab, discord
# Обертка АПИ ГитЛаба — https://python-gitlab.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")) # определение адреса и токена экземляра ГитЛаба
'''project = gitlab_instance.projects.get(13)''' # определение проекта в котором нужно создавать задачи
# TODO: Привязка экзепмляра и проекта к Дискорд Каналу в БД, а не в коде
intents = discord.Intents.default() # использовать требования по умолчанию
@ -43,7 +33,7 @@ async def on_message(message): # обработка каждого сообще
issue_text = message.content.replace("/issue ","") # получение текста команды «issue»
if command('/issue'): # команда создания задачи на ГитЛабе
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}):
await reply("Задача «" + issue_text + "» создана успешно, ^w^")
else:
@ -54,7 +44,6 @@ async def on_message(message): # обработка каждого сообще
if command('/project'):
try:
database_spisok[int(message.channel.id)] = int(message.content.replace("/project ",""))
open("database.msgpack", "wb").write(msgpack.packb(database_spisok))
await reply("данные сохранены ^w^")
except:
await reply("не удалось сохранить id")
@ -77,40 +66,4 @@ async def on_message(message): # обработка каждого сообще
await reply("====================================")
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'):
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()'''
discord_bot.run(str(environ.get("TOKEN_DISCORD"))) # авторизация бота по токену из среды и запуск