import paramiko

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect('140.44.4.11', username='root', key_filename=r'C:\Users\Snarf\.ssh\id_ed25519_snarfnet', timeout=10)

def run(cmd):
    _, o, e = c.exec_command(cmd)
    return o.read().decode().strip()

pod = run("kubectl get pod -n monitoring -l app.kubernetes.io/name=uptime-kuma -o jsonpath='{.items[0].metadata.name}'")

sql = """
INSERT INTO `group` (id, name, public, active, weight, status_page_id)
VALUES (1, 'Core Infrastructure', 1, 1, 1000, 1)
ON DUPLICATE KEY UPDATE name=VALUES(name);

INSERT INTO monitor_group (monitor_id, group_id, weight)
SELECT id, 1, 1000 FROM monitor WHERE active=1;
"""

print(run(f"kubectl exec -n monitoring {pod} -- mariadb --socket=/app/data/run/mariadb.sock -u root kuma -e \"{sql}\""))
print(run(f"kubectl exec -n monitoring {pod} -- mariadb --socket=/app/data/run/mariadb.sock -u root kuma -e 'SELECT count(id) FROM monitor_group;'"))

c.close()