Use localhost.

This commit is contained in:
Ivaylo Novakov 2020-09-04 16:39:39 +02:00
parent 59a77bfaf6
commit 62e27120cd
No known key found for this signature in database
GPG Key ID: 06B9354AB08BE9C6
1 changed files with 3 additions and 18 deletions

View File

@ -16,20 +16,6 @@ bot_token = setup()
client = discord.Client() client = discord.Client()
# get_hostname reads the HOSTNAME from the .env file passed as first argument
# to the script
async def get_hostname():
if len(sys.argv) > 1:
env_file = sys.argv[1]
with open(env_file, 'r') as file:
for line in file.read().split('\n'):
pair = line.split("=")
if pair[0] == "HOSTNAME":
return pair[1]
await send_msg(client, "HOSTNAME not found, cannot check health status", force_notify=True)
os.exit(0)
# exit_after kills the script if it hasn't exited on its own after `delay` seconds # exit_after kills the script if it hasn't exited on its own after `delay` seconds
async def exit_after(delay): async def exit_after(delay):
await asyncio.sleep(delay) await asyncio.sleep(delay)
@ -52,7 +38,7 @@ async def run_checks():
if len(trace) < 1900: if len(trace) < 1900:
await send_msg(client, "```\n{}\n```".format(trace), force_notify=False) await send_msg(client, "```\n{}\n```".format(trace), force_notify=False)
else: else:
await send_msg(client, "Failed to run the checks!", file=discord.File(io.BytesIO(trace), filename="failed_checks.log"), force_notify=True) await send_msg(client, "Failed to run the checks!", file=discord.File(io.BytesIO(trace.encode()), filename="failed_checks.log"), force_notify=True)
# check_health checks /health-check endpoint and reports recent issues # check_health checks /health-check endpoint and reports recent issues
@ -60,15 +46,14 @@ async def check_health():
print("\nChecking wallet/funds health...") print("\nChecking wallet/funds health...")
try: try:
hostname = await get_hostname() res = requests.get("http://localhost/health-check", verify=False)
res = requests.get("http://"+hostname+"/health-check")
except: # catch all exceptions except: # catch all exceptions
trace = traceback.format_exc() trace = traceback.format_exc()
print("[DEBUG] check_health() failed.") print("[DEBUG] check_health() failed.")
if len(trace) < 1900: if len(trace) < 1900:
await send_msg(client, "```\n{}\n```".format(trace), force_notify=False) await send_msg(client, "```\n{}\n```".format(trace), force_notify=False)
else: else:
await send_msg(client, "Failed to run the checks!", file=discord.File(io.BytesIO(trace), filename="failed_checks.log"), force_notify=True) await send_msg(client, "Failed to run the checks!", file=discord.File(io.BytesIO(trace.encode()), filename="failed_checks.log"), force_notify=True)
return return
# Get the number of hours to look back in the logs or use 1 as default. # Get the number of hours to look back in the logs or use 1 as default.