py-pve-cloud 0.0.4__py3-none-any.whl → 0.1.1__py3-none-any.whl

pve_cloud/cli/pvcli.py

@@ -2,7 +2,7 @@ import argparse
 import yaml
 from proxmoxer import ProxmoxAPI
 import os
-import paramiko
+
 
 
 def connect_cluster(args):

pve_cloud/cli/pvclu.py

@@ -2,6 +2,9 @@ import argparse
 import yaml
 import os 
 import socket
+import paramiko
+import dns.resolver
+import base64
 
 
 def get_cloud_domain(target_pve):
@@ -41,9 +44,65 @@ def get_online_pve_host(target_pve):
   raise Exception(f"Could not find online pve host for {target_pve}")
 
 
+def get_cloud_env(pve_host):
+  ssh = paramiko.SSHClient()
+  ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+  ssh.connect(pve_host, username="root")
+
+  # since we need root we cant use sftp and root via ssh is disabled
+  _, stdout, _ = ssh.exec_command("cat /etc/pve/cloud/cluster_vars.yaml")
+
+  cluster_vars = yaml.safe_load(stdout.read().decode('utf-8'))
+
+  _, stdout, _ = ssh.exec_command("cat /etc/pve/cloud/secrets/patroni.pass")
+
+  patroni_pass = stdout.read().decode('utf-8')
+
+  return cluster_vars, patroni_pass
+
+
 def get_online_pve_host_prsr(args):
   print(f"export PVE_ANSIBLE_HOST='{get_online_pve_host(args.target_pve)}'")
 
+
+def get_ssh_master_kubeconfig(cluster_vars, stack_name):
+  resolver = dns.resolver.Resolver()
+  resolver.nameservers = [cluster_vars['bind_master_ip'], cluster_vars['bind_slave_ip']]
+
+  ddns_answer = resolver.resolve(f"masters-{stack_name}.{cluster_vars['pve_cloud_domain']}")
+  ddns_ips = [rdata.to_text() for rdata in ddns_answer]
+
+  if not ddns_ips:
+    raise Exception("No master could be found via DNS!")
+
+  ssh = paramiko.SSHClient()
+  ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+  ssh.connect(ddns_ips[0], username="admin")
+
+  # since we need root we cant use sftp and root via ssh is disabled
+  _, stdout, _ = ssh.exec_command("sudo cat /etc/kubernetes/admin.conf")
+
+  return base64.b64encode(stdout.read().decode('utf-8').replace("https://127.0.0.1:6443", f"https://{ddns_ips[0]}:6443").encode('utf-8')).decode('utf-8')
+
+
+def export_envr(args):
+  ansible_host = get_online_pve_host(args.target_pve)
+  cloud_domain = get_cloud_domain(args.target_pve)
+  cluster_vars, patroni_pass = get_cloud_env(ansible_host)
+  print(f"export PVE_ANSIBLE_HOST='{ansible_host}'")
+  print(f"export PVE_CLOUD_DOMAIN='{cloud_domain}'")
+
+  # tf vars
+  print(f"export PG_CONN_STR=\"postgres://postgres:{patroni_pass}@{cluster_vars['pve_haproxy_floating_ip_internal']}:5000/tf_states?sslmode=disable\"")
+  print(f"export TF_VAR_pve_cloud_domain='{cloud_domain}'")
+  print(f"export TF_VAR_pve_host='{ansible_host}'")
+  print(f"export TF_VAR_cluster_proxy_ip='{cluster_vars['pve_haproxy_floating_ip_internal']}'")
+  print(f"export TF_VAR_pve_cloud_pg_cstr=\"postgresql+psycopg2://postgres:{patroni_pass}@{cluster_vars['pve_haproxy_floating_ip_internal']}:5000/pve_cloud?sslmode=disable\"")
+  print(f"export TF_VAR_master_b64_kubeconf='{get_ssh_master_kubeconfig(cluster_vars, args.stack_name)}'")
+
+
   
 def main():
   parser = argparse.ArgumentParser(description="PVE Cloud utility cli. Should be called with bash eval.")
@@ -56,6 +115,11 @@ def main():
   get_cld_domain_parser.add_argument("--target-pve", type=str, help="The target pve cluster to get the cloud domain of.", required=True)
   get_cld_domain_parser .set_defaults(func=get_cld_domain_prsr)
 
+  export_envr_parser = subparsers.add_parser("export-envrc", help="Export variables for k8s .envrc", parents=[base_parser])
+  export_envr_parser.add_argument("--target-pve", type=str, help="The target pve cluster.", required=True)
+  export_envr_parser.add_argument("--stack-name", type=str, help="Stack name of the deployment.", required=True)
+  export_envr_parser.set_defaults(func=export_envr)
+
   get_online_pve_host_parser = subparsers.add_parser("get-online-host", help="Gets the ip for the first online proxmox host in the cluster.", parents=[base_parser])
   get_online_pve_host_parser.add_argument("--target-pve", type=str, help="The target pve cluster to get the first online ip of.", required=True)
   get_online_pve_host_parser.set_defaults(func=get_online_pve_host_prsr)

{py_pve_cloud-0.0.4.dist-info → py_pve_cloud-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-pve-cloud
-Version: 0.0.4
+Version: 0.1.1
 Author-email: Tobias Huebner <tobias.huebner@vmzberlin.com>
 License-Expression: GPL-3.0-or-later
 License-File: LICENSE
@@ -8,5 +8,7 @@ Requires-Dist: PyYAML==6.0.2
 Requires-Dist: psycopg2-binary==2.9.10
 Requires-Dist: SQLAlchemy==2.0.43
 Requires-Dist: alembic==1.16.5
+Requires-Dist: paramiko==4.0.0
 Requires-Dist: proxmoxer==2.2.0
+Requires-Dist: dnspython==2.7.0
 Dynamic: license-file

{py_pve_cloud-0.0.4.dist-info → py_pve_cloud-0.1.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-pve_cloud/cli/pvcli.py,sha256=inR4KMI63nWufPYoM9udxgh3CN10xyQlJw_kGzVw7DQ,3166
-pve_cloud/cli/pvclu.py,sha256=D5R5Gzo_iSImPTtEHVJWZoFr66uijeFbb-k6ODz_71o,2498
+pve_cloud/cli/pvcli.py,sha256=sRMVhpUQzDF1sRP9d8u-aDUkCRp6KoneiqZX_mLCgGU,3151
+pve_cloud/cli/pvclu.py,sha256=Hszd6kVvWcD_Z4aMyVyzsDvCmV9myJTiRate1ABISJg,5276
 pve_cloud/lib/inventory.py,sha256=0W31j34jtxUa-3Cxe0HmHJqc_8CpvvxyDqd0sAZ5oCI,31
 pve_cloud/orm/alchemy.py,sha256=1PABOo2d-ceIwlSZPYx19aXZFszDXMh8YIXgkLT-554,5000
 pve_cloud/orm/alembic.ini,sha256=7140n-YUj06aAIHOHACm8U0xhUFUoBZ4Jw23KlYB9EA,4865
@@ -11,9 +11,9 @@ pve_cloud/orm/migrations/versions/7868bcd05006_migrate_old.py,sha256=rU8Bw2tYDyn
 pve_cloud/orm/migrations/versions/7dea8c4ee39f_init.py,sha256=iMDyHhtyvpSywMnLhiSEL3W12YSm6sPa18XRgzQcwDg,954
 pve_cloud/orm/migrations/versions/944a8fd5d5bc_ext_ctrl_plns.py,sha256=LnVAShLaU1asz1L7TYs7oI9SnLxPp2IOA6K83kHNkN0,1674
 pve_cloud/orm/migrations/versions/d9b711555be8_ext_control_plane.py,sha256=uBqv1r5pLX-RjqciKYx0zvWyygJMa5u58DTnVfIEAF0,1073
-py_pve_cloud-0.0.4.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-py_pve_cloud-0.0.4.dist-info/METADATA,sha256=EENqU0S2uq7SBI05ChHVDybFOBFEhcA55zEb4RWPJFo,362
-py_pve_cloud-0.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-py_pve_cloud-0.0.4.dist-info/entry_points.txt,sha256=VvncsKmTJ46irz-9wQZ4Zo1FgNBjRltGDBKR9ht18mE,84
-py_pve_cloud-0.0.4.dist-info/top_level.txt,sha256=mpT7ttGRyZJVt_obhPLBHyIBcjKhUdJ-qVsMEVX5WJg,10
-py_pve_cloud-0.0.4.dist-info/RECORD,,
+py_pve_cloud-0.1.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+py_pve_cloud-0.1.1.dist-info/METADATA,sha256=iOaf-gpm588QhGddRlPFmEgoXZa49o3sbtKWU5YG7nQ,425
+py_pve_cloud-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+py_pve_cloud-0.1.1.dist-info/entry_points.txt,sha256=VvncsKmTJ46irz-9wQZ4Zo1FgNBjRltGDBKR9ht18mE,84
+py_pve_cloud-0.1.1.dist-info/top_level.txt,sha256=mpT7ttGRyZJVt_obhPLBHyIBcjKhUdJ-qVsMEVX5WJg,10
+py_pve_cloud-0.1.1.dist-info/RECORD,,