Исходник Ticket System для Discord бота | Discord.py

Flowseal · 12 Апр 2021

Продолжаем парад селфкод функций для дискорд бота на пайтоне
На повествке дня система тикетов: имеет +- чистый код, комментарии по коду, нетипичный функционал

Возможности:

Создание тикета с уникальным id
Создание тикета с несколькими участниками
Открытие тикета
Закрытие тикета
Создание временного voice-канала тикета
Удаление тикета

Перед использованием создайте файл по пути files/servers.json с контентном '{}' (без апострофов)
Создайте категорию и установите её id с помощью команды tickets_category_set
Далее, настройте эту категорию, чтобы никто, кроме админов/проверяющих не имел доступ к этой категории, все нужные права для участников настроит бот
Поменяйте id админов в листе admin_ids или же сделайте проверку с определённой ролью
Поменяйте при необходимости значение логической переменной CAN_READ_IF_CLOSE

Python:

import os
import sys
import random
import asyncio

sys.path.append(os.path.abspath('../settings'))
from global_sets import *
from discord.ext import commands

'''
Перед использованием создайте файл по пути files/servers.json с контентном '{}' (без апострофов)
Создайте категорию и установите её id с помощью команды tickets_category_set
Далее, настройте эту категорию, чтобы никто, кроме админов/проверяющих не имел доступ к этой категории, все нужные права для участников настроит бот
Поменяйте id админов в листе admin_ids или же сделайте проверку с определённой ролью
Поменяйте при необходимости значение логической переменной CAN_READ_IF_CLOSE
'''

####  CUSTOM  ##    VARS    ## 

CAN_READ_IF_CLOSE = True
admin_ids = []

##    CUSTOM  ##    VARS  ####

def generate(length):
    letters = 'qwertyuiopasdfghjklzxcvbnm1234567890'
    name = ''.join(random.choice(letters) for i in range(length))
    return name

async def create_text_channel(category, channel_name):
        channel = await category.create_text_channel(channel_name)
        return channel

async def create_voice_channel(category, channel_name):
        channel = await category.create_voice_channel(channel_name)
        return channel

async def delete_channel(guild, channel_id):
        channel = guild.get_channel(channel_id)
        await channel.delete()

def is_ticket_message(ctx):
    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    channel_name = (ctx.message.channel.name)[-7:]
    guild_id = str(ctx.message.guild.id)

    if (guild_id in table):
        if ('tickets' in table[guild_id]):
            if (channel_name in table[guild_id]['tickets']):
                return True

    return False

def is_ticket_server(server):
    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    if not server.channel:
        return False

    channel_name = (server.channel.name)[-7:]
    guild_id = str(server.channel.guild.id)

    if (guild_id in table):
        if ('tickets' in table[guild_id]):
            if (channel_name in table[guild_id]['tickets']):
                return True

    return False



@client.command(aliases = ['tickets_category_set'])
async def __tickets_category_set(ctx, id:int):
    if (not (str(ctx.message.author.id) in admin_ids)):
        return False

    if (not right_channel(ctx)):
        return False

    category = client.get_channel(id)
    if (not category):
        return False

    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    guild_id = str(ctx.message.guild.id)

    if (not guild_id in table):
        table[guild_id] = {}

    table[guild_id]['tickets_category'] = id

    with open('files/servers.json', 'w') as f:
        json.dump(table, f)

    await ctx.message.delete()



@client.command(aliases = ['ticket'])
async def __ticket(ctx, *users:discord.User):

    # Проверка, в каком канале написана команда
    if (not right_channel(ctx)):
        return False

    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    ticket_id = generate(7)
    channel_name = ctx.message.author.name + '-' + ticket_id
    guild_id = str(ctx.message.guild.id)

    # Если в нашем файле нет ключа нашего сервера - записываем
    if (not guild_id in table):
        table[guild_id] = {}
        table[guild_id]['tickets'] = {}

        with open('files/servers.json', 'w') as f:
            json.dump(table, f)

        return False

    # Есть ли запись категории с тикетами
    if (not 'tickets_category' in table[guild_id]):
        print('Tickets category ID is invalid, update it (!tickets_category_set)')
        return False
    
    category_id = table[guild_id]['tickets_category']
    category = client.get_channel(category_id)

    # Существует ли категория с таким id
    if (not category):
        print('Tickets category ID is invalid, update it (!tickets_category_set)')
        return False

    if (not 'tickets' in table[guild_id]):
        table[guild_id]['tickets'] = {}

    # Если id тикета совпал с существующим
    while (ticket_id in table[guild_id]['tickets']):
        ticket_id = generate(7)
        channel_name = ctx.message.author.name + '-' + ticket_id

    # Создаём текстовый канал нашего тикета
    channel = await create_text_channel(category, channel_name)

    if (not channel):
        print('Cant create a text-channel' + channel_name)
        return False

    users_list = []

    for user in users:
        if (user.id != ctx.message.author.id):
            users_list.append(user.id)


    # Создаём новую запись о тикете в файле
    table[guild_id]['tickets'][ticket_id] = {}
    table[guild_id]['tickets'][ticket_id]['channel'] = channel.id
    table[guild_id]['tickets'][ticket_id]['author'] = ctx.message.author.id
    table[guild_id]['tickets'][ticket_id]['users'] = users_list
    table[guild_id]['tickets'][ticket_id]['closed'] = False

    with open('files/servers.json', 'w') as f:
        json.dump(table, f)

    # Создаём оверрайт-права
    overwrite = discord.PermissionOverwrite()
    overwrite.send_messages = True
    overwrite.read_messages = True
    overwrite.read_message_history = True
    overwrite.attach_files = True
    overwrite.view_channel = True

    # Задаём права автору сообщения и тегаем
    await channel.set_permissions(ctx.message.author, overwrite=overwrite)
    await channel.send(f'<@{ctx.message.author.id}>')

    await asyncio.sleep(1)

    # Задаём права доп. участникам тикета и тегаем
    for user in users:
        await channel.set_permissions(user, overwrite=overwrite)
        await channel.send(f'<@{user.id}>')
        await asyncio.sleep(1)

    # Саздём emb
    emb = discord.Embed( title = f'Тикет #{ticket_id}', description = '', colour = discord.Color.red() )
    
    emb.set_author(name = ctx.author.name, icon_url = ctx.author.avatar_url)

    emb.add_field( name = 'Команды', value = f'`{prefix}voice`', inline=False)
    emb.add_field( name = 'Админ-команды', value = f'`{prefix}close` `{prefix}open` `{prefix}delete`', inline=False)

    emb.set_thumbnail(url = "https://icons.iconarchive.com/icons/sonya/swarm/128/Ticket-icon.png")

    await channel.send(embed=emb)

    await ctx.send(f':tickets: Тикет {ticket_id} создан! (<#{channel.id}>)')

    

@client.command(aliases = ['voice'])
async def __voice(ctx):
    if not is_ticket_message(ctx):
        return False
    
    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    channel = await create_voice_channel(ctx.message.channel.category, ctx.message.channel.name)
    ticket_id = (ctx.message.channel.name)[-7:]
    guild_id = str(ctx.message.guild.id)

    # Создаём оверрайт-права
    overwrite = discord.PermissionOverwrite()
    overwrite.connect = True
    overwrite.speak = True
    overwrite.stream = True

    author = ctx.message.guild.get_member(table[guild_id]['tickets'][ticket_id]['author'])

    # Выдаём войс-права для автора
    await channel.set_permissions(author, overwrite=overwrite)

    # Выдаём войс-права сторонним пользователям
    for user_id in table[guild_id]['tickets'][ticket_id]['users']:
        user = ctx.message.guild.get_member(user_id)
        if (not user):
            continue

        await channel.set_permissions(user, overwrite=overwrite)


@client.event
async def on_voice_state_update(member, before, after):
    if (is_ticket_server(before)):
        if len(before.channel.members) == 0:
            await delete_channel(before.channel.guild, before.channel.id)

    

@client.command(aliases = ['close'])
async def __close(ctx):
    if not is_ticket_message(ctx):
        return False

    if (not (str(ctx.message.author.id) in admin_ids)):
        return False
    
    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    ticket_id = (ctx.message.channel.name)[-7:]
    guild_id = str(ctx.message.guild.id)
    channel_id = table[guild_id]['tickets'][ticket_id]['channel']
    channel = ctx.message.channel.guild.get_channel(channel_id)

    table[guild_id]['tickets'][ticket_id]['closed'] = True

    with open('files/servers.json', 'w') as f:
        json.dump(table, f)

    # Создаём оверрайт-права
    overwrite = discord.PermissionOverwrite()
    overwrite.send_messages = False
    overwrite.read_messages = CAN_READ_IF_CLOSE
    overwrite.read_message_history = CAN_READ_IF_CLOSE
    overwrite.attach_files = False
    overwrite.view_channel = False

    author = ctx.message.guild.get_member(table[guild_id]['tickets'][ticket_id]['author'])

    # Удаляем права с создателя
    await channel.set_permissions(author, overwrite=overwrite)

    await asyncio.sleep(1)

    # Удаляем права с юзеров
    for user_id in table[guild_id]['tickets'][ticket_id]['users']:
        user = ctx.message.guild.get_member(user_id)
        if (not user):
            continue

        await channel.set_permissions(user, overwrite=overwrite)


@client.command(aliases = ['open'])
async def __open(ctx):
    if not is_ticket_message(ctx):
        return False

    if (not (str(ctx.message.author.id) in admin_ids)):
        return False
    
    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    ticket_id = (ctx.message.channel.name)[-7:]
    guild_id = str(ctx.message.guild.id)
    channel_id = table[guild_id]['tickets'][ticket_id]['channel']
    channel = ctx.message.channel.guild.get_channel(channel_id)

    table[guild_id]['tickets'][ticket_id]['closed'] = False

    with open('files/servers.json', 'w') as f:
        json.dump(table, f)

    # Создаём оверрайт-права
    overwrite = discord.PermissionOverwrite()
    overwrite.send_messages = True
    overwrite.read_messages = True
    overwrite.read_message_history = True
    overwrite.attach_files = True
    overwrite.view_channel = True

    author = ctx.message.guild.get_member(table[guild_id]['tickets'][ticket_id]['author'])

    # Удаляем права с создателя
    await channel.set_permissions(author, overwrite=overwrite)

    await asyncio.sleep(1)

    # Удаляем права с юзеров
    for user_id in table[guild_id]['tickets'][ticket_id]['users']:
        user = ctx.message.guild.get_member(user_id)
        if (not user):
            continue

        await channel.set_permissions(user, overwrite=overwrite)


@client.command(aliases = ['delete'])
async def __delete(ctx):
    if not is_ticket_message(ctx):
        return False

    if (not (str(ctx.message.author.id) in admin_ids)):
        return False
    
    with open('files/servers.json', 'r') as f:
        table = json.load(f)

    ticket_id = (ctx.message.channel.name)[-7:]
    guild_id = str(ctx.message.guild.id)
    channel_id = table[guild_id]['tickets'][ticket_id]['channel']
    channel = ctx.message.channel.guild.get_channel(channel_id)

    # delete text channel & ticket
    await channel.delete()
    del table[guild_id]['tickets'][ticket_id]

    # delete voice_channel probably
    channel = discord.utils.get(ctx.message.guild.channels, name=ctx.message.channel.name)
    if (channel is not None):
        await channel.delete()

angro · 12 Апр 2021

для генерации id используй uuid4, если нужен уникальный идентификатор это стандартная практика.

Flowseal · 12 Апр 2021

SQL написал(а):
Какой уровень в радуге ?

90, голда 2

angro написал(а):
для генерации id используй uuid4, если нужен уникальный идентификатор это стандартная практика.

зачем

Neki_play · 20 Апр 2021

Спс, думаю пригодятся сурсы

ded78667 · 20 Апр 2021

angro написал(а):
для генерации id используй uuid4, если нужен уникальный идентификатор это стандартная практика.

message.id как лучший идентификатор тикета :negr:

angro · 20 Апр 2021

Думаю что лучшим будет итерируемое значение, хранящиеся например в фаиле

Flowseal · 21 Апр 2021

angro написал(а):
Думаю что лучшим будет итерируемое значение, хранящиеся например в фаиле

Ты знаешь, что такое итерируемое значение?)

angro · 21 Апр 2021

Имел в виду инкрементное значение, голова забита ...

whynotvoid · 27 Сен 2021

~~Пожалуйста простите за ап темы, я новичок и не очень понимаю, хотя очень интересно, поэтому не бейте плз.~~
У меня появилось пару вопросиков на которые я не могу найти ответ, для чего нужен servers.json и куда писать токен
просто интересно и бот реально крутой, те кто понимают в этом, подскажите пожалуйста(

krutiy11 · 31 Июл 2022

Выдаёт данную ошибку:

Command raised an exception: NameError: name 'right_channel' is not defined

что делать?