78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import glob
|
|
import re
|
|
from pprint import pprint
|
|
|
|
import jinja2
|
|
from bs4 import BeautifulSoup
|
|
|
|
from uptime_kuma_api import convert_from_socket, params_map_notification_provider
|
|
|
|
|
|
def deduplicate_list(l):
|
|
out = []
|
|
for i in l:
|
|
if i not in out:
|
|
out.append(i)
|
|
return out
|
|
|
|
|
|
def build_notification_providers():
|
|
providers = []
|
|
for path in glob.glob('uptime-kuma/server/notification-providers/*'):
|
|
with open(path) as f:
|
|
content = f.read()
|
|
match = re.search(r'class [^ ]+ extends NotificationProvider {', content)
|
|
if match:
|
|
match = re.search(r'name = "([^"]+)";', content)
|
|
name = match.group(1)
|
|
|
|
inputs = re.findall(r'notification\.([^ ,.;})\]]+)', content)
|
|
inputs = deduplicate_list(inputs)
|
|
inputs = [i.strip() for i in inputs]
|
|
|
|
providers.append({
|
|
"name": name,
|
|
"inputs": inputs,
|
|
})
|
|
return providers
|
|
|
|
|
|
def build_notification_provider_conditions():
|
|
conditions = {}
|
|
for path in glob.glob('uptime-kuma/src/components/notifications/*'):
|
|
if path.endswith("index.js"):
|
|
continue
|
|
with open(path) as f:
|
|
content = f.read()
|
|
match = re.search(r'<template>[\s\S]+</template>', content, re.MULTILINE)
|
|
html = match.group(0)
|
|
soup = BeautifulSoup(html, "html.parser")
|
|
inputs = soup.find_all("input")
|
|
for input in inputs:
|
|
condition = {}
|
|
attrs = input.attrs
|
|
v_model = attrs.get("v-model")
|
|
min_ = attrs.get("min")
|
|
max_ = attrs.get("max")
|
|
if min_:
|
|
condition["min"] = int(min_)
|
|
if max_:
|
|
condition["max"] = int(max_)
|
|
param_name = re.match(r'\$parent.notification.(.*)$', v_model).group(1)
|
|
if condition:
|
|
conditions[param_name] = condition
|
|
conditions = convert_from_socket(params_map_notification_provider, conditions)
|
|
return conditions
|
|
|
|
|
|
def write_to_file(template, destination, **kwargs):
|
|
env = jinja2.Environment(loader=jinja2.FileSystemLoader("./"))
|
|
template = env.get_template(template)
|
|
rendered = template.render(**kwargs)
|
|
with open(destination, "w") as f:
|
|
f.write(rendered)
|
|
|
|
conditions = build_notification_provider_conditions()
|
|
pprint(conditions)
|
|
# notification_providers = build_notification_providers()
|
|
# write_to_file("notification_providers.py.j2", "./../uptimekumaapi/notification_providers.py", notification_providers=notification_providers)
|