LumeWeb
/
skynet-webportal
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #934 from SkynetLabs/parameterize-health-check-script 

add option to pass channel and notify role to health check script
This commit is contained in:
Ivaylo Novakov 2021-06-30 15:47:31 +02:00 committed by GitHub
parent 8918b1f1dd af80a84094
commit bdf96cc9c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
setup-scripts/bot_utils.py
View File

@ -10,9 +10,6 @@ import urllib, json, os, traceback, discord, sys, re, subprocess, requests, io
# sc_precision is the number of hastings per siacoin # sc_precision is the number of hastings per siacoin
sc_precision = 10 ** 24 sc_precision = 10 ** 24
CHANNEL_NAME = "skynet-portal-health-check"
ROLE_NAME = "skynet-prod"
# Environment variable globals # Environment variable globals
api_endpoint, port, portal_name, bot_token, password = None, None, None, None, None api_endpoint, port, portal_name, bot_token, password = None, None, None, None, None
discord_client = None discord_client = None
@ -50,16 +47,19 @@ def setup():
load_dotenv(dotenv_path=env_path, override=True) load_dotenv(dotenv_path=env_path, override=True)
global bot_token global bot_token
bot_token = os.environ["DISCORD_BOT_TOKEN"] bot_token = os.getenv("DISCORD_BOT_TOKEN")
global bot_channel
bot_channel = os.getenv("DISCORD_BOT_CHANNEL", "skynet-server-health")
global bot_notify_role
bot_notify_role = os.getenv("DISCORD_BOT_NOTIFY_ROLE", "skynet-prod")
global portal_name global portal_name
portal_name = os.getenv("SKYNET_SERVER_API") portal_name = os.getenv("SKYNET_SERVER_API")
# Get a port or use default
global port global port
port = os.getenv("API_PORT") port = os.getenv("API_PORT", "9980")
if not port:
port = "9980"
global api_endpoint global api_endpoint
api_endpoint = "http://{}:{}".format(get_api_ip(), port) api_endpoint = "http://{}:{}".format(get_api_ip(), port)
@ -80,17 +80,17 @@ async def send_msg(client, msg, force_notify=False, file=None):
chan = None chan = None
for c in guild.channels: for c in guild.channels:
if c.name == CHANNEL_NAME: if c.name == bot_channel:
chan = c chan = c
break break
if chan is None: if chan is None:
print("Can't find channel {}".format(CHANNEL_NAME)) print("Can't find channel {}".format(bot_channel))
# Get the prod team role # Get the prod team role
role = None role = None
for r in guild.roles: for r in guild.roles:
if r.name == ROLE_NAME: if r.name == bot_notify_role:
role = r role = r
break break
@ -113,7 +113,7 @@ async def send_msg(client, msg, force_notify=False, file=None):
io.BytesIO(file.encode()), filename=filename io.BytesIO(file.encode()), filename=filename
) # wrap text into discord file wrapper ) # wrap text into discord file wrapper
if force_notify: if force_notify and role:
msg = "{} /cc {}".format(msg, role.mention) msg = "{} /cc {}".format(msg, role.mention)
await chan.send(msg, file=file) await chan.send(msg, file=file)