python Исходник Гайд Ticket System для Discord бота - Discord.py

[wlux.net]

Имеет +- чистый код, комментарии по коду, нетипичный функционал

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

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

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

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()

 

[SpaceRaketka]

А есть ли в этом смысл, если в дс шопе есть готовый тикет тул?

 

[user.none]

код не плохой, но можно было и перевести хотя бы на sqlite