ChidoriDiscordBotPy/main.py at main · terminator7/ChidoriDiscordBotPy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
from ChidoriDiscordClient import ChidoriDiscordBot
import discord
import asyncio
from dotenv import load_dotenv
import os
from discord import ui
from functools import partial

a_emoji = discord.PartialEmoji(name="A", id=1378874248562737254)
b_emoji = discord.PartialEmoji(name="B", id=1378874232703811594)
c_emoji = discord.PartialEmoji(name="C", id=1378874211543810058)
d_emoji = discord.PartialEmoji(name="D", id=1378874354900795453)

emoji_list = [a_emoji, b_emoji, c_emoji, d_emoji]

class Questionnaire(discord.ui.Modal, title="Questionnaire Response"):
    name = discord.ui.TextInput(label="Name")
    answer = discord.ui.TextInput(label="Answer", style=discord.TextStyle.paragraph)

    async def on_submit(self, interaction: discord.Interaction):
        print(f"{self.name.value}\n{self.answer.value}")
        await interaction.response.send_message(f"Thanks for your response, {self.name}!", ephemeral=True)

class QuestionnaireView(discord.ui.View):
    def __init__(self):
        super().__init__()

        for emoji in emoji_list:
            button = discord.ui.Button(style=discord.ButtonStyle.grey, emoji=emoji)
            button.callback = partial(self.button_callback, emoji=emoji)
            self.add_item(button)

    async def button_callback(self,interaction: discord.Interaction, emoji):
        if emoji != None:
            await interaction.response.send_message(f"{interaction.user.display_name} Pressure option {emoji.name}")


load_dotenv()
# --- Discord Information ---
DISCORD_TOKEN: str = str(os.getenv("SECRET_KEY"))

async def get_discord_ids(discord_client: ChidoriDiscordBot, discord_message: discord.Message) -> None:
    args = (discord_message.content.split(" "))[1:]
    server = discord_message.guild
    if server == None:
        print(f"Command in question: {discord_message.content.split(" ")[0]}\nError: Could not find server ID from some reason")
    else:
        members_message = ""
        async for member in server.fetch_members():
            members_message += f"{member.display_name}: {member.id}\n"
        channel = discord_message.channel
        await channel.send(members_message)


async def testing_command_handler(discord_client: ChidoriDiscordBot, discord_message: discord.Message) -> None:
    args = (discord_message.content.split(" "))[1:]
    print(args)
    channel_id = discord_message.channel
    await channel_id.send(f"{args}")

async def choose_choice(discord_client: ChidoriDiscordBot, discord_message: discord.Message)-> None:
    question_view = QuestionnaireView()
    channel_id = discord_message.channel
    await channel_id.send("Hello", view=question_view)

async def send_dm_to_person(discord_client: ChidoriDiscordBot, discord_message: discord.Message) -> None:
    args = (discord_message.content.split(" "))[1:]
    print(args)
    discord_user = discord_client.get_user(int(args[0]))
    message = "".join(f"{word} " for word in args[1:])
    if discord_user != None:
        try:
            await discord_user.send(message)
        except Exception as e:
            print(e)
    else:
        print("eat shit")


def main():
    intents = discord.Intents.default()
    intents.message_content = True

    discord_client = ChidoriDiscordBot(intents=discord.Intents.all())

    # -- Initialize --
    guild_id = str(os.getenv("GUILD_ID"))
    discord_client.add_server_id(int(guild_id))

    # -- Commands --
    discord_client.add_command("test", testing_command_handler)
    discord_client.add_command("members", get_discord_ids)
    discord_client.add_command("message", send_dm_to_person)
    discord_client.add_command("testing", choose_choice)
    discord_client.run(DISCORD_TOKEN)


if __name__ == "__main__":
    print(DISCORD_TOKEN)
    main()