mirror of
https://github.com/iFargle/headscale-webui.git
synced 2025-12-13 02:19:47 +01:00
87 lines
3.5 KiB
Python
87 lines
3.5 KiB
Python
|
|
||
|
|
import logging, sys, pytz, os, headscale
|
||
|
|
from datetime import datetime, timedelta, date
|
||
|
|
from dateutil import parser
|
||
|
|
|
||
|
|
log = logging.getLogger('server.helper')
|
||
|
|
handler = logging.StreamHandler(sys.stderr)
|
||
|
|
handler.setFormatter(logging.Formatter('%(name)s - %(levelname)s - %(message)s'))
|
||
|
|
log.addHandler(handler)
|
||
|
|
log.setLevel(logging.INFO)
|
||
|
|
|
||
|
|
def pretty_print_duration(duration):
|
||
|
|
days, seconds = duration.days, duration.seconds
|
||
|
|
hours = (days * 24 + seconds // 3600)
|
||
|
|
mins = ((seconds % 3600) // 60)
|
||
|
|
secs = (seconds % 60)
|
||
|
|
if days > 0: return str(days ) + " days ago" if days > 1 else str(days ) + " day ago"
|
||
|
|
elif hours > 0: return str(hours) + " hours ago" if hours > 1 else str(hours) + " hour ago"
|
||
|
|
elif mins > 0: return str(mins ) + " minutes ago" if mins > 1 else str(mins ) + " minute ago"
|
||
|
|
else: return str(secs ) + " seconds ago" if secs >= 1 or secs == 0 else str(secs ) + " second ago"
|
||
|
|
|
||
|
|
def text_color_duration(duration):
|
||
|
|
days, seconds = duration.days, duration.seconds
|
||
|
|
hours = (days * 24 + seconds // 3600)
|
||
|
|
mins = ((seconds % 3600) // 60)
|
||
|
|
secs = (seconds % 60)
|
||
|
|
if days > 30: return "grey-text "
|
||
|
|
elif days > 14: return "red-text text-darken-2 "
|
||
|
|
elif days > 5: return "deep-orange-text text-lighten-1"
|
||
|
|
elif days > 1: return "deep-orange-text text-lighten-1"
|
||
|
|
elif hours > 12: return "orange-text "
|
||
|
|
elif hours > 1: return "orange-text text-lighten-2"
|
||
|
|
elif hours == 1: return "yellow-text "
|
||
|
|
elif mins > 15: return "yellow-text text-lighten-2"
|
||
|
|
elif mins > 5: return "green-text text-lighten-3"
|
||
|
|
elif secs > 30: return "green-text text-lighten-2"
|
||
|
|
else: return "green-text "
|
||
|
|
|
||
|
|
def key_test():
|
||
|
|
api_key = headscale.get_api_key()
|
||
|
|
url = headscale.get_url()
|
||
|
|
|
||
|
|
# Test the API key. If the test fails, return a failure.
|
||
|
|
# AKA, if headscale returns Unauthorized, fail:
|
||
|
|
status = headscale.test_api_key(url, api_key)
|
||
|
|
if status != 200: return False
|
||
|
|
else:
|
||
|
|
# Check if the key needs to be renewed
|
||
|
|
headscale.renew_api_key(url, api_key)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def get_color(id, type = ""):
|
||
|
|
# Define the colors... Seems like a good number to start with
|
||
|
|
if type == "text":
|
||
|
|
colors = [
|
||
|
|
"red-text text-lighten-1",
|
||
|
|
"teal-text text-lighten-1",
|
||
|
|
"blue-text text-lighten-1",
|
||
|
|
"blue-grey-text text-lighten-1",
|
||
|
|
"indigo-text text-lighten-2",
|
||
|
|
"green-text text-lighten-1",
|
||
|
|
"deep-orange-text text-lighten-1",
|
||
|
|
"yellow-text text-lighten-2",
|
||
|
|
"purple-text text-lighten-2",
|
||
|
|
"indigo-text text-lighten-2",
|
||
|
|
"brown-text text-lighten-1",
|
||
|
|
"grey-text text-lighten-1",
|
||
|
|
]
|
||
|
|
index = id % len(colors)
|
||
|
|
return colors[index]
|
||
|
|
else:
|
||
|
|
colors = [
|
||
|
|
"red lighten-1",
|
||
|
|
"teal lighten-1",
|
||
|
|
"blue lighten-1",
|
||
|
|
"blue-grey lighten-1",
|
||
|
|
"indigo lighten-2",
|
||
|
|
"green lighten-1",
|
||
|
|
"deep-orange lighten-1",
|
||
|
|
"yellow lighten-2",
|
||
|
|
"purple lighten-2",
|
||
|
|
"indigo lighten-2",
|
||
|
|
"brown lighten-1",
|
||
|
|
"grey lighten-1",
|
||
|
|
]
|
||
|
|
index = id % len(colors)
|
||
|
|
return colors[index]
|