TotalFreedomMC/TotalFreedomBot
1
0
Fork 0
mirror of https://github.com/TotalFreedomMC/TotalFreedomBot.git synced 2024-12-22 15:44:57 +00:00
Code Issues Projects Releases Wiki Activity

Update server_commands.py

Browse source
This commit is contained in:
Elmon11 2021-01-07 20:18:25 +01:00 committed by GitHub
parent 813c190130
commit 16059bff13
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
commands/server_commands.py
View file

@ -1,16 +1,16 @@
import re
from datetime import datetime
import discord import discord
import requests import requests
import re
import time
from checks import *
from discord.ext import commands from discord.ext import commands
from datetime import datetime
from functions import * from checks import is_liaison, is_staff, is_senior, is_creative_designer, is_mod_or_has_perms, NoPermission
from functions import hit_endpoint, get_server_status, format_list_entry, read_json
from unicode import clipboard from unicode import clipboard
class ServerCommands(commands.Cog): class ServerCommands(commands.Cog, name="Server Commands"):
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
@ -30,7 +30,7 @@ class ServerCommands(commands.Cog):
@is_liaison() @is_liaison()
async def eventhost(self, ctx, user: discord.Member): async def eventhost(self, ctx, user: discord.Member):
"""Add or remove event host role - liaison only.""" """Add or remove event host role - liaison only."""
eventhostrole = ctx.guild.get_role(event_host) eventhostrole = ctx.guild.get_role(self.bot.event_host)
if eventhostrole in user.roles: if eventhostrole in user.roles:
await user.remove_roles(eventhostrole) await user.remove_roles(eventhostrole)
await ctx.send(f'```Succesfully took {eventhostrole.name} from {user.name}```') await ctx.send(f'```Succesfully took {eventhostrole.name} from {user.name}```')
@ -42,7 +42,7 @@ class ServerCommands(commands.Cog):
@is_creative_designer() @is_creative_designer()
async def masterbuilder(self, ctx, user: discord.Member): async def masterbuilder(self, ctx, user: discord.Member):
"""Add or remove master builder role - ECD only.""" """Add or remove master builder role - ECD only."""
master_builder_role = ctx.guild.get_role(master_builder) master_builder_role = ctx.guild.get_role(self.bot.master_builder)
if master_builder_role in user.roles: if master_builder_role in user.roles:
await user.remove_roles(master_builder_role) await user.remove_roles(master_builder_role)
await ctx.send(f'```Succesfully took {master_builder_role.name} from {user.name}```') await ctx.send(f'```Succesfully took {master_builder_role.name} from {user.name}```')
@ -54,7 +54,7 @@ class ServerCommands(commands.Cog):
@is_staff() @is_staff()
async def serverban(self, ctx, user: discord.Member): async def serverban(self, ctx, user: discord.Member):
"""Add or remove server banned role.""" """Add or remove server banned role."""
serverbannedrole = ctx.guild.get_role(server_banned) serverbannedrole = ctx.guild.get_role(self.bot.server_banned)
if serverbannedrole in user.roles: if serverbannedrole in user.roles:
await user.remove_roles(serverbannedrole) await user.remove_roles(serverbannedrole)
await ctx.send(f'Took Server Banned role from {user.name}') await ctx.send(f'Took Server Banned role from {user.name}')
@ -144,12 +144,12 @@ class ServerCommands(commands.Cog):
bytes(command, 'ascii') + b"\r\n") bytes(command, 'ascii') + b"\r\n")
elif args[0] == 'saconfig': elif args[0] == 'saconfig':
if args[1] not in ['add', 'remove']: if args[1] not in ['add', 'remove']:
raise no_permission(['IS_SENIOR_ADMIN']) raise NoPermission(['IS_SENIOR_ADMIN'])
else: else:
self.write_telnet_session(server, self.write_telnet_session(server,
bytes(command, 'ascii') + b"\r\n") bytes(command, 'ascii') + b"\r\n")
else: else:
raise no_permission(['IS_SENIOR_ADMIN']) raise NoPermission(['IS_SENIOR_ADMIN'])
except Exception as e: except Exception as e:
em.title = 'Command error' em.title = 'Command error'
em.colour = 0xFF0000 em.colour = 0xFF0000
@ -181,8 +181,7 @@ class ServerCommands(commands.Cog):
if 'error' in attempt.lower(): if 'error' in attempt.lower():
em.title = 'Command error' em.title = 'Command error'
em.colour = 0xFF0000 em.colour = 0xFF0000
em.description = '{attempt}' em.description = f'{attempt}'
print(f'Error while killing server: {e}')
await ctx.send(embed=em) await ctx.send(embed=em)
else: else:
em.title = 'Success' em.title = 'Success'
@ -280,9 +279,9 @@ class ServerCommands(commands.Cog):
order = ['Owners', 'Executives', 'Developers', 'Senior Admins', 'Admins', 'Master Builders', order = ['Owners', 'Executives', 'Developers', 'Senior Admins', 'Admins', 'Master Builders',
'Operators', 'Imposters'] 'Operators', 'Imposters']
sorted = [entry for rank in order for entry in entries if entry.name == rank] sorted_ranks = [entry for rank in order for entry in entries if entry.name == rank]
for x in sorted: for x in sorted_ranks:
em.add_field(name=f'{x.name} ({x.playercount})', value=x.value, inline=False) em.add_field(name=f'{x.name} ({x.playercount})', value=x.value, inline=False)
except Exception as e: except Exception as e:
@ -294,16 +293,15 @@ class ServerCommands(commands.Cog):
async def ip(self, ctx): async def ip(self, ctx):
"""Returns the server IP.""" """Returns the server IP."""
await ctx.send(embed=discord.Embed(description='play.totalfreedom.me', title='Server IP')) await ctx.send(embed=discord.Embed(description='play.totalfreedom.me', title='Server IP'))
# pass #discordSRV responds already.
@commands.command() @commands.command()
@is_staff() @is_staff()
async def archivereports(self, ctx): async def archivereports(self, ctx):
"""Archive all in-game reports older than 24 hours.""" """Archive all in-game reports older than 24 hours."""
count = 0 count = 0
reports_channel = self.bot.get_channel(reports_channel_id) reports_channel = self.bot.get_channel(self.bot.reports_channel_id)
archived_reports_channel = self.bot.get_channel( archived_reports_channel = self.bot.get_channel(
archived_reports_channel_id) self.bot.archived_reports_channel_id)
await ctx.channel.trigger_typing() await ctx.channel.trigger_typing()
async for report in reports_channel.history(limit=100): async for report in reports_channel.history(limit=100):
try: try:
@ -353,7 +351,8 @@ class ServerCommands(commands.Cog):
try: try:
server_uptime = \ server_uptime = \
re.match('([0-9]+ days? )?([0-9]+ hours )?([0-9]+ minutes )?([0-6]?[0-9] seconds)', server_uptime)[0] re.match('([0-9]+ days? )?([0-9]+ hours )?([0-9]+ minutes )?([0-6]?[0-9] seconds)', server_uptime)[
0]
server_tps = re.match('[0-2][0-9].?[0-9]*', server_tps)[0] server_tps = re.match('[0-2][0-9].?[0-9]*', server_tps)[0]
maximum_memory = re.match('[0-9]+,?[0-9]* MB', maximum_memory)[0] maximum_memory = re.match('[0-9]+,?[0-9]* MB', maximum_memory)[0]
allocated_memory = re.match('[0-9]+,?[0-9]* MB', allocated_memory)[0] allocated_memory = re.match('[0-9]+,?[0-9]* MB', allocated_memory)[0]
@ -375,7 +374,7 @@ class ServerCommands(commands.Cog):
@is_mod_or_has_perms() @is_mod_or_has_perms()
async def fixreports(self, ctx): async def fixreports(self, ctx):
await ctx.channel.trigger_typing() await ctx.channel.trigger_typing()
reports_channel = self.bot.get_channel(reports_channel_id) reports_channel = self.bot.get_channel(self.bot.reports_channel_id)
messages = await reports_channel.history(limit=500).flatten() messages = await reports_channel.history(limit=500).flatten()
fixed = 0 fixed = 0
for message in messages: for message in messages: