def get_root_rwr_servers_status(): """Check the status of the RWR root servers.""" from models import RwrRootServer, RwrRootServerStatus, Variable from disco.api.client import APIClient as DiscordAPIClient from flask import url_for from rwrs import db import rwr.constants import helpers import arrow click.echo('Pinging servers') hosts_to_ping = [server['host'] for group in rwr.constants.ROOT_RWR_SERVERS for server in group['servers']] rwr_root_servers = RwrRootServer.query.filter(RwrRootServer.host.in_(hosts_to_ping)).all() rwr_root_servers_by_host = {rwr_root_server.host: rwr_root_server for rwr_root_server in rwr_root_servers} servers_down_count_then = sum([1 for rwr_root_server in rwr_root_servers if rwr_root_server.status == RwrRootServerStatus.DOWN]) servers_down_count_now = 0 for host in hosts_to_ping: click.echo(host, nl=False) is_server_up = helpers.ping(host) if is_server_up: click.secho(' Up', fg='green') else: click.secho(' Down', fg='red') servers_down_count_now += 1 if host not in rwr_root_servers_by_host: rwr_root_server = RwrRootServer() rwr_root_server.host = host else: rwr_root_server = rwr_root_servers_by_host[host] rwr_root_server.status = RwrRootServerStatus.UP if is_server_up else RwrRootServerStatus.DOWN db.session.add(rwr_root_server) Variable.set_value('last_root_rwr_servers_check', arrow.utcnow().floor('minute')) click.echo('Persisting to database') db.session.commit() message = None if servers_down_count_then == 0 and servers_down_count_now > 0: with app.app_context(): message = ':warning: Online multiplayer is having issues right now. Are the devs aware? If not, poke **pasik** or **JackMayol**. Some details here: {}'.format(url_for('online_multiplayer_status', _external=True)) elif servers_down_count_then > 0 and servers_down_count_now == 0: message = ':white_check_mark: Outage update: online multiplayer is back up and running!' if message: click.echo('Sending Discord bot message') discord_api_client = DiscordAPIClient(app.config['DISCORD_BOT_TOKEN']) discord_api_client.channels_messages_create(app.config['DISCORD_BOT_CHANNEL_ID'], message) click.secho('Done', fg='green')
class DiscordInterface: TOKEN = None # FIXME a long token from Discord # the next two should be big decimal numbers; in Discord, you can right # click and Copy ID to get them GUILD = 'FIXME' HINT_CHANNEL = 'FIXME' # You also need to enable the "Server Members Intent" under the "Privileged # Gateway Intents" section of the "Bot" page of your application from the # Discord Developer Portal. Or you can comment out the code that # initializes `self.avatars` below. def __init__(self): self.client = None self.avatars = None if self.TOKEN and not settings.IS_TEST: self.client = APIClient(self.TOKEN) def get_avatars(self): if self.avatars is None: self.avatars = {} if self.client is not None: members = self.client.guilds_members_list(self.GUILD).values() for member in members: self.avatars[member.user.username] = member.user.avatar_url for member in members: self.avatars[member.name] = member.user.avatar_url return self.avatars # If you get an error code 50001 when trying to create a message, even # though you're sure your bot has all the permissions, it might be because # you need to "connect to and identify with a gateway at least once"?? # https://discord.com/developers/docs/resources/channel#create-message # I spent like four hours trying to find weird asynchronous ways to do this # right before each time I send a message, but it seems maybe you actually # just need to do this once and your bot can create messages forever? # disco.py's Client (not APIClient) wraps an APIClient and a GatewayClient, # the latter of which does this. So I believe you can fix this by running a # script like the following *once* on your local machine (it will, as # advertised, run forever; just kill it after a few seconds)? # from gevent import monkey # monkey.patch_all() # from disco.client import Client, ClientConfig # Client(ClientConfig({'token': TOKEN})).run_forever() def update_hint(self, hint): HintsConsumer.send_to_all( json.dumps({ 'id': hint.id, 'content': render_to_string('hint_list_entry.html', { 'hint': hint, 'now': timezone.localtime() }) })) embed = MessageEmbed() embed.author.url = hint.full_url() if hint.claimed_datetime: embed.color = 0xdddddd embed.timestamp = hint.claimed_datetime.isoformat() embed.author.name = 'Claimed by {}'.format(hint.claimer) if hint.claimer in self.get_avatars(): embed.author.icon_url = self.get_avatars()[hint.claimer] debug = 'claimed by {}'.format(hint.claimer) else: embed.color = 0xff00ff embed.author.name = 'U N C L A I M E D' claim_url = hint.full_url(claim=True) embed.title = 'Claim: ' + claim_url embed.url = claim_url debug = 'unclaimed' if self.client is None: message = hint.long_discord_message() logger.info('Hint, {}: {}\n{}'.format(debug, hint, message)) logger.info('Embed: {}'.format(embed.to_dict())) elif hint.discord_id: try: self.client.channels_messages_modify(self.HINT_CHANNEL, hint.discord_id, embed=embed) except Exception: dispatch_general_alert( 'Discord API failure: modify\n{}'.format( traceback.format_exc())) else: message = hint.long_discord_message() try: discord_id = self.client.channels_messages_create( self.HINT_CHANNEL, message, embed=embed).id except Exception: dispatch_general_alert( 'Discord API failure: create\n{}'.format( traceback.format_exc())) return hint.discord_id = discord_id hint.save(update_fields=('discord_id', )) def clear_hint(self, hint): HintsConsumer.send_to_all(json.dumps({'id': hint.id})) if self.client is None: logger.info('Hint done: {}'.format(hint)) elif hint.discord_id: # try: # self.client.channels_messages_delete( # self.HINT_CHANNEL, hint.discord_id) # except Exception: # dispatch_general_alert('Discord API failure: delete\n{}'.format( # traceback.format_exc())) # hint.discord_id = '' # hint.save(update_fields=('discord_id',)) # what DPPH did instead of deleting messages: # (nb. I tried to make these colors color-blind friendly) embed = MessageEmbed() if hint.status == 'ANS': embed.color = 0xaaffaa elif hint.status == 'REF': embed.color = 0xcc6600 # nothing for obsolete embed.author.name = '{} by {}'.format(hint.status, hint.claimer) embed.author.url = hint.full_url() embed.description = hint.response[:250] if hint.claimer in self.get_avatars(): embed.author.icon_url = self.get_avatars()[hint.claimer] debug = 'claimed by {}'.format(hint.claimer) try: self.client.channels_messages_modify( self.HINT_CHANNEL, hint.discord_id, content=hint.short_discord_message(), embed=embed) except Exception: dispatch_general_alert( 'Discord API failure: modify\n{}'.format( traceback.format_exc()))