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

Отправил

Browse source
This commit is contained in:
bogdan zažigin 2023-04-12 20:46:15 +03:00
parent 3ee02c4411
commit 86ab62b440
Signed by: zaboal
GPG key ID: C9918166D559FE8A
2 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
source/gitlab-discord_bot/bot.py
View file

@ -54,11 +54,11 @@ async def on_message(message): # обработка каждого сообще
except: except:
await reply("не удалось сохранить ссылку на проект")''' await reply("не удалось сохранить ссылку на проект")'''
if command('/project'): if command('/setup'):
try: try:
content = message.content.replace("/project ","").split(" ") content = message.content.replace("/setup ","").split(" ")
setup = GitlabDiscordBinding(content[0], content[1], int(message.channel.id), int(message.author.id)) setup = GitlabDiscordBinding(content[0], content[1], int(message.channel.id), int(message.author.id))
database.gitlab_discord_bindings.append(setup) database.update(setup)
await reply(database) await reply(database)
except: except:
await reply("asdadadfa") await reply("asdadadfa")

10
source/gitlab-discord_bot/database.py
View file

@ -73,4 +73,12 @@ class GitlabDiscordDatabase:
def get_project(self, discord_channel_id: int): def get_project(self, discord_channel_id: int):
for binding in range(len(self.gitlab_discord_bindings)): for binding in range(len(self.gitlab_discord_bindings)):
if self.gitlab_discord_bindings[binding].discord_channel_id == discord_channel_id: if self.gitlab_discord_bindings[binding].discord_channel_id == discord_channel_id:
return self.gitlab_discord_bindings[binding].gitlab_project_url return self.gitlab_discord_bindings[binding].gitlab_project_url
def update(self, current_binding: GitlabDiscordBinding):
for binding in range(len(self.gitlab_discord_bindings)):
if self.gitlab_discord_bindings[binding].discord_channel_id == current_binding.discord_channel_id and self.gitlab_discord_bindings[binding].discord_user_id == current_binding.discord_user_id:
self.gitlab_discord_bindings[binding] = current_binding
elif self.gitlab_discord_bindings[binding].gitlab_project_url != current_binding.gitlab_project_url and self.gitlab_discord_bindings[binding].discord_user_gitlab_token != current_binding.discord_user_gitlab_token:
self.gitlab_discord_bindings.append(current_binding)