main.py 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493
  1. import os
  2. from dotenv import load_dotenv
  3. import discord
  4. from discord.ext import commands
  5. from discord.commands import Option
  6. from discord.commands import slash_command
  7. from datetime import datetime
  8. import configparser
  9. import mysql.connector
  10. intents = discord.Intents.default()
  11. intents.message_content = True
  12. intents.members = True
  13. intents.guilds = True
  14. intents.reactions = True
  15. client = discord.Client(intents=intents)
  16. #------#
  17. #Load .env file
  18. load_dotenv()
  19. token = os.getenv("TOKEN")
  20. if token is None:
  21. raise ValueError("TOKEN not found in .env file")
  22. debug_guilds_up = []
  23. server_token = os.getenv("SERVER").split(",")
  24. for i in range(len(server_token)):
  25. debug_guilds_up.append(int(server_token[i]))
  26. dbhost = os.getenv("HOST")
  27. if token is None:
  28. raise ValueError("HOST not found in .env file")
  29. dbname = os.getenv("NAME")
  30. if token is None:
  31. raise ValueError("NAME not found in .env file")
  32. dbpsswd = os.getenv("PASSWORD")
  33. if token is None:
  34. raise ValueError("PASSWORD not found in .env file")
  35. dbdb = os.getenv("DATABASE")
  36. if token is None:
  37. raise ValueError("DATABASE not found in .env file")
  38. #------#
  39. #ConfigParser
  40. config = configparser.RawConfigParser()
  41. configFilePath = r'config.cfg'
  42. config.read_file(open(configFilePath))
  43. label_rules = config.get('Reactionroles Rules', 'label_rules')
  44. role_rules = config.get('Reactionroles Rules', 'rules_role')
  45. channel_log = config.get('Logs', 'channel_log')
  46. channel_banlog = config.get('Logs', 'ban_log')
  47. #------#
  48. #SQL initialization
  49. conn = mysql.connector.connect(
  50. host=dbhost,
  51. user=dbname,
  52. password=dbpsswd
  53. )
  54. cursor = conn.cursor()
  55. conn.database = str(dbdb)
  56. cursor.execute("""
  57. CREATE TABLE IF NOT EXISTS User (
  58. id INT AUTO_INCREMENT PRIMARY KEY,
  59. discordid INTEGER(20)
  60. name VARCHAR(100),
  61. roles INTEGER(250)
  62. )
  63. """)
  64. cursor.execute("""
  65. CREATE TABLE IF NOT EXISTS Warns (
  66. id INT AUTO_INCREMENT PRIMARY KEY,
  67. userid INTEGER(20),
  68. username VARCHAR(100),
  69. moderatorname VARCHAR(100),
  70. reason VARCHAR(250),
  71. date TIMESTAMP DEFAULT CURRENT_TIMESTAMP
  72. )
  73. """)
  74. #------#
  75. #Initialize Bot
  76. bot = commands.Bot(
  77. command_prefix=commands.when_mentioned_or("!"),
  78. description="VicePD Bot",
  79. intents=intents,
  80. debug_guilds=debug_guilds_up if debug_guilds_up else None
  81. )
  82. async def load_extensions():
  83. for filename in os.listdir("cogs"):
  84. if filename.endswith(".py"):
  85. bot.load_extension(f"cogs.{filename[:-3]}")
  86. class Admin(commands.Cog):
  87. def __init__(self, bot):
  88. self.bot = bot
  89. #---------------------------------#
  90. #Bot Online Console
  91. @bot.event
  92. async def on_ready():
  93. print(f"{bot.user} ist online")
  94. if bot.guilds:
  95. channel = discord.utils.get(bot.guilds[0].channels, id=int(channel_log))
  96. if channel:
  97. await channel.send(f"{bot.user} ist online")
  98. await load_extensions()
  99. bot.add_view(PersistentRoleView()) #loading reactionrole memory
  100. #---------------------------------------------------------------------------------------#
  101. #DONT Touch anything above this line, unless you know what you are doing!#
  102. #---------------------------------------------------------------------------------------#
  103. #---------------------------------#
  104. ## Greet
  105. @bot.slash_command(description="Greet a User")
  106. async def greet(ctx, user: str = Option(discord.User, "The user, you want to greet")):
  107. await ctx.respond(f"Hello {user.mention}")
  108. #---------------------------------#
  109. #---------------------------------#
  110. ## Say
  111. @bot.slash_command(description="Let the bot send a message")
  112. async def say(
  113. ctx,
  114. text: str = Option(description="Input the text you want to send"),
  115. channel_input: discord.TextChannel = Option(description="Select the channel,where you want to send the message.")
  116. ):
  117. channel= discord.utils.get(ctx.guild.channels, id = int(channel_input[2:-1]))
  118. await channel.send(text)
  119. await ctx.respond("Message sent", ephemeral=True)
  120. #---------------------------------#
  121. #---------------------------------#
  122. ## Userinfo
  123. @bot.slash_command(name="userinfo", description="Show informations of a user from this server")
  124. async def userinfo(
  125. ctx,
  126. user: str = Option(discord.User, "Select User"),
  127. ):
  128. if user is None:
  129. user = ctx.author
  130. elif user not in ctx.guild.members:
  131. await ctx.respond("The selected user is not a member on this Server!", ephemeral=True)
  132. return
  133. elif user == bot.user:
  134. await ctx.respond(f"This is me - the {bot.user}", ephemeral=True)
  135. return
  136. embed = discord.Embed(
  137. title=f"Information about *{user.name}*",
  138. description=f"Here you see all details about {user.mention}",
  139. color=discord.Color.blue()
  140. )
  141. time = discord.utils.format_dt(user.created_at, "R")
  142. embed.add_field(name="Account creation date", value=time, inline=False)
  143. if len(user.roles) >= 2:
  144. embed.add_field(name="Roles", value=", ".join([role.mention for role in user.roles if role.name != "@everyone"]), inline=False)
  145. else:
  146. embed.add_field(name="Roles", value="User has no roles", inline=False)
  147. embed.add_field(name="Server join date", value=discord.utils.format_dt(user.joined_at, "R"), inline=False)
  148. embed.add_field(name="User ID", value=user.id)
  149. embed.set_thumbnail(url=user.display_avatar.url)
  150. embed.set_author(name="VicePD", icon_url="https://i.imgur.com/6QteFrg.png")
  151. embed.set_footer(text="VicePD - Bot | Made by BaumSplitter41")
  152. await ctx.respond(embed=embed)
  153. #---------------------------------#
  154. #_________________________________#
  155. #BAN SYSTEM
  156. #---------------------------------#
  157. ##Ban
  158. @bot.slash_command(name="ban", description="Ban a user from this Server")
  159. async def ban(
  160. ctx,
  161. user: Option(discord.User, description = "Select User", required=True), # type: ignore
  162. reason: Option(str, description = "Reason for the ban", default="No reason provided") # type: ignore
  163. ):
  164. if not ctx.author.guild_permissions.ban_members:
  165. await ctx.respond("Error: You don't have the permission to ban Members!", ephemeral=True)
  166. return
  167. if user == bot.user:
  168. await ctx.respond("Error: I can't ban myself!", ephemeral=True)
  169. return
  170. if user == ctx.author:
  171. await ctx.respond("Error: You can't ban yourself!", ephemeral=True)
  172. return
  173. channel= discord.utils.get(ctx.guild.channels, id = int(channel_banlog))
  174. embed = discord.Embed(
  175. title=f"Ban of **{user.name}**",
  176. description=f"User {user.mention} has been banned from the Server",
  177. color=discord.Color.red()
  178. )
  179. time = discord.utils.format_dt(datetime.now(), "f")
  180. embed.add_field(name="Ban Date", value=time, inline=False)
  181. embed.add_field(name="Moderator", value=f"{ctx.author}", inline=False)
  182. embed.add_field(name="Reason", value=reason, inline=False)
  183. embed.add_field(name="User ID", value=user.id)
  184. embed.set_thumbnail(url=user.display_avatar.url)
  185. embed.set_author(name="VicePD", icon_url="https://i.imgur.com/6QteFrg.png")
  186. embed.set_footer(text="VicePD - Bot | Made by BaumSplitter41")
  187. try:
  188. await ctx.guild.ban(user, reason=reason)
  189. await ctx.respond(f"User {user.mention} has been banned from this Server!", ephemeral=True)
  190. await channel.send(embed=embed)
  191. except discord.Forbidden:
  192. await ctx.respond("Error: I don't have permission to ban this user.", ephemeral=True)
  193. except discord.HTTPException as e:
  194. await ctx.respond(f"Error: Could not ban User {user.mention}. Reason: {e}", ephemeral=True)
  195. except Exception as e:
  196. await ctx.respond(f"Unexpected error: {e}", ephemeral=True)
  197. #---------------------------------#
  198. #Unban
  199. @bot.slash_command(name="unban", description="Unban a user from this Server")
  200. async def unban(
  201. ctx,
  202. user: Option(discord.User, description = "Insert User ID", required=True), # type: ignore
  203. reason: Option(str, description = "Reason for the unbanning", default="No reason provided") # type: ignore
  204. ):
  205. if not ctx.author.guild_permissions.ban_members:
  206. await ctx.respond("Error: You don't have the permission to unban Members!", ephemeral=True)
  207. return
  208. if user == bot.user:
  209. await ctx.respond("Error: I can't unban myself!", ephemeral=True)
  210. return
  211. if user == ctx.author:
  212. await ctx.respond("Error: You can't unban yourself!", ephemeral=True)
  213. return
  214. if user in ctx.guild.members:
  215. await ctx.respond("Error: This user is not banned!", ephemeral=True)
  216. return
  217. channel= discord.utils.get(ctx.guild.channels, id = int(channel_banlog))
  218. embed = discord.Embed(
  219. title=f"Unban of **{user.name}**",
  220. description=f"User {user.mention} was unbanned from this server.",
  221. color=discord.Color.green()
  222. )
  223. time = discord.utils.format_dt(datetime.now(), "f")
  224. embed.add_field(name="Unban Date", value=time, inline=False)
  225. embed.add_field(name="Moderator", value=f"{ctx.author}", inline=False)
  226. embed.add_field(name="Reason", value=reason, inline=False)
  227. embed.add_field(name="User ID", value=user.id)
  228. embed.set_thumbnail(url=user.display_avatar.url)
  229. embed.set_author(name="VicePD", icon_url="https://i.imgur.com/6QteFrg.png")
  230. embed.set_footer(text="VicePD - Bot | Made by BaumSplitter41")
  231. try:
  232. await ctx.guild.unban(user, reason=reason)
  233. await ctx.respond(f"User {user.mention} is now unbanned!", ephemeral=True)
  234. await channel.send(embed=embed)
  235. except discord.Forbidden:
  236. await ctx.respond("Error: I don't have permission to unban this user.", ephemeral=True)
  237. except discord.HTTPException as e:
  238. await ctx.respond(f"Error: Could not unban User {user.mention}. Reason: {e}", ephemeral=True)
  239. except Exception as e:
  240. await ctx.respond(f"Unexpected error: {e}", ephemeral=True)
  241. #---------------------------------#
  242. #_________________________________#
  243. #---------------------------------#
  244. #Kick
  245. @bot.slash_command(name="kick", description="Kick a user from this Server")
  246. async def kick(
  247. ctx,
  248. user: Option(discord.User, description = "Select User", required=True), # type: ignore
  249. reason: Option(str, description = "Reason for the ban", default="No reason provided") # type: ignore
  250. ):
  251. if not ctx.author.guild_permissions.kick_members:
  252. await ctx.respond("Error: You don't have the permission to kick Members!", ephemeral=True)
  253. return
  254. if user == bot.user:
  255. await ctx.respond("Error: I can't kick myself!", ephemeral=True)
  256. return
  257. if user == ctx.author:
  258. await ctx.respond("Error: You can't kick yourself!", ephemeral=True)
  259. return
  260. try:
  261. await ctx.guild.kick(user, reason=reason)
  262. await ctx.respond(f"User {user.mention} has been kick from this Server!", ephemeral=True)
  263. except discord.Forbidden:
  264. await ctx.respond("Error: I don't have permission to kick this user.", ephemeral=True)
  265. except discord.HTTPException as e:
  266. await ctx.respond(f"Error: Could not kick User {user.mention}. Reason: {e}", ephemeral=True)
  267. except Exception as e:
  268. await ctx.respond(f"Unexpected error: {e}", ephemeral=True)
  269. #---------------------------------#
  270. #---------------------------------#
  271. #Warn
  272. @bot.slash_command(name="warn", description="Warn a user from this Server")
  273. async def warn(
  274. ctx,
  275. user: Option(discord.User, description = "Select User", required=True), # type: ignore
  276. reason: Option(str, description = "Reason for the warn", default="No reason provided") # type: ignore
  277. ):
  278. if not ctx.author.guild_permissions.kick_members:
  279. await ctx.respond("Error: You don't have the permission to warn Members!", ephemeral=True)
  280. return
  281. if user == bot.user:
  282. await ctx.respond("Error: I can't warn myself!", ephemeral=True)
  283. return
  284. if user == ctx.author:
  285. await ctx.respond("Error: You can't warn yourself!", ephemeral=True)
  286. return
  287. cursor.execute("""
  288. INSERT INTO Warns (userid, username, moderatorname, reason)
  289. VALUES (%s, %s, %s, %s)
  290. """, (user.id, str(user), str(ctx.author), reason))
  291. conn.commit()
  292. await ctx.respond(f"User {user.mention} has been warned for: {reason}", ephemeral=True)
  293. #_________________________________#
  294. ## Reaction role system
  295. #---------------------------------#
  296. #reaction role verfiy
  297. class PersistentRoleView(discord.ui.View):
  298. def __init__(self):
  299. super().__init__(timeout=None)
  300. @discord.ui.button(
  301. label=label_rules,
  302. style=discord.ButtonStyle.success,
  303. emoji="✅",
  304. custom_id="persistent_view:role_verify"
  305. )
  306. async def verify_callback(self, button: discord.ui.Button, interaction: discord.Interaction):
  307. role = interaction.guild.get_role(int(role_rules))
  308. if role is None:
  309. await interaction.response.send_message("Fehler: Die konfigurierte Rolle wurde nicht gefunden.", ephemeral=True)
  310. return
  311. if role in interaction.user.roles:
  312. await interaction.user.remove_roles(role)
  313. await interaction.response.send_message(f"Rolle **{role.name}** wurde entfernt.", ephemeral=True)
  314. else:
  315. await interaction.user.add_roles(role)
  316. await interaction.response.send_message(f"Du hast die Rolle **{role.name}** erhalten!", ephemeral=True)
  317. @bot.slash_command(name="verify_message", description="Sendet die persistente Reaction-Role Nachricht")
  318. async def setup_rr(
  319. ctx: discord.ApplicationContext,
  320. channel: discord.TextChannel,
  321. title: str,
  322. description: str
  323. ):
  324. if not ctx.author.guild_permissions.administrator:
  325. await ctx.respond("Keine Berechtigung.", ephemeral=True)
  326. return
  327. embed = discord.Embed(
  328. title=title,
  329. description=description,
  330. color=discord.Color.red()
  331. )
  332. try:
  333. await channel.send(embed=embed, view=PersistentRoleView())
  334. await ctx.respond(f"Nachricht erfolgreich in {channel.mention} gesendet!", ephemeral=True)
  335. except discord.Forbidden:
  336. await ctx.respond("Ich darf in diesem Kanal nicht schreiben.", ephemeral=True)
  337. #---------------------------------#
  338. #_________________________________#
  339. #_________________________________#
  340. ## Help System
  341. #---------------------------------#
  342. #How to team
  343. @bot.slash_command(name="how_to_team", description= "Get Infos")
  344. async def how_to_team(
  345. ctx,
  346. ):
  347. server = ctx.guild
  348. embed = discord.Embed(
  349. title=f"__How to join the Team on {server.name}__",
  350. description=f"If you want to join the Serverteam open a ticket in #ticket.",
  351. color=discord.Color.yellow()
  352. )
  353. embed.set_thumbnail(url=server.icon)
  354. embed.set_author(name="VicePD", icon_url="https://i.imgur.com/6QteFrg.png")
  355. embed.set_footer(text="VicePD - Bot | Made by BaumSplitter41")
  356. await ctx.respond(embed=embed, ephemeral=True)
  357. #---------------------------------#
  358. #How to start
  359. @bot.slash_command(name="how_to_start", description= "Get Infos")
  360. async def how_to_start(
  361. ctx,
  362. ):
  363. server = ctx.guild
  364. embed = discord.Embed(
  365. title=f"__How to start__",
  366. description=f"Hallo {ctx.author.mention}, um auf unserem Server spielen zu können, ließ dir zuerst das Regelwerk in in #regelwerk durch. Um einem Department beizutreten wähle in #how-to-start eine Einweisungsrolle aus. Melde dich anschließend für eine Einweisung an. **Wichtig: gehe erst krz vor der Einweisung auf den Server!**",
  367. color=discord.Color.yellow()
  368. )
  369. embed.set_thumbnail(url=server.icon)
  370. embed.set_author(name="VicePD", icon_url="https://i.imgur.com/6QteFrg.png")
  371. embed.set_footer(text="VicePD - Bot | Made by BaumSplitter41")
  372. await ctx.respond(embed=embed, ephemeral=True)
  373. #---------------------------------#
  374. #Help_cache
  375. @bot.slash_command(name="help_cache", description= "Get Infos")
  376. async def help_cache(
  377. ctx,
  378. ):
  379. server = ctx.guild
  380. embed = discord.Embed(
  381. title=f"__How to clear your game cache__",
  382. description=f"Follow the follwing steps to clear your game cache:",
  383. color=discord.Color.yellow()
  384. )
  385. embed.add_field(name="Close Game", value="Close FiveM completely", inline=False)
  386. embed.add_field(name="Press keys", value="Win + R", inline=False)
  387. embed.add_field(name="Go to the folder", value="\Local\FiveM\FiveM.app\data", inline=False)
  388. embed.add_field(name="Delete the folders", value="cache, server-cache, server-cache-priv", inline=False)
  389. embed.add_field(name="Restart Game", value="Restart the game and download the resources again.", inline=False)
  390. embed.set_thumbnail(url=server.icon)
  391. embed.set_author(name="VicePD", icon_url="https://i.imgur.com/6QteFrg.png")
  392. embed.set_footer(text="VicePD - Bot | Made by BaumSplitter41")
  393. await ctx.respond(embed=embed, ephemeral=True)
  394. #---------------------------------#
  395. #_________________________________#
  396. #---------------------------------#
  397. #Run function
  398. bot.run(token)
  399. #---------------------------------#