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

pve_cloud/cli/pvcli.py

@@ -0,0 +1,97 @@
+import argparse
+import yaml
+from proxmoxer import ProxmoxAPI
+import os
+import paramiko
+
+
+def connect_cluster(args):
+  # try load current dynamic inventory
+  inv_path = os.path.expanduser("~/.pve-cloud-dyn-inv.yaml")
+  if os.path.exists(inv_path):
+    with open(inv_path, "r") as file:
+      dynamic_inventory = yaml.safe_load(file)
+  else:
+    # initialize empty
+    dynamic_inventory = {}
+  
+  # init cloud domain if not there
+  if args.pve_cloud_domain not in dynamic_inventory:
+    dynamic_inventory[args.pve_cloud_domain] = {}
+
+  # connect to the passed host
+  proxmox = ProxmoxAPI(
+    args.pve_host, user="root", backend='ssh_paramiko'
+  )
+
+  # try get the cluster name
+  cluster_name = None
+  status_resp = proxmox.cluster.status.get()
+  for entry in status_resp:
+    if entry['id'] == "cluster":
+      cluster_name = entry['name']
+      break
+
+  if cluster_name is None:
+    raise Exception("Could not get cluster name")
+  
+  if cluster_name in dynamic_inventory[args.pve_cloud_domain] and not args.force:
+    print(f"cluster {cluster_name} already in dynamic inventory")
+    return
+  
+  # overwrite on force / create fresh
+  dynamic_inventory[args.pve_cloud_domain][cluster_name] = {}
+  
+  # not present => add and safe the dynamic inventory
+  cluster_hosts = proxmox.nodes.get()
+
+  for node in cluster_hosts:
+    node_name = node["node"]
+
+    if node["status"] == "offline":
+      print(f"skipping offline node {node_name}")
+      continue
+    
+    # get the main ip
+    ifaces = proxmox.nodes(node_name).network.get()
+    node_ip_address = None
+    for iface in ifaces:
+      if 'gateway' in iface:
+        if node_ip_address is not None:
+          raise Exception(f"found multiple ifaces with gateways for node {node_name}")
+        node_ip_address = iface.get("address")
+
+    if node_ip_address is None:
+      raise Exception(f"Could not find ip for node {node_name}")
+    
+    dynamic_inventory[args.pve_cloud_domain][cluster_name][node_name] = {
+      "ansible_user": "root",
+      "ansible_host": node_ip_address
+    }
+
+  with open(inv_path, "w") as file:
+    yaml.dump(dynamic_inventory, file)
+
+
+  
+def main():
+  parser = argparse.ArgumentParser(description="PVE general purpose cli for setting up.")
+
+  base_parser = argparse.ArgumentParser(add_help=False)
+
+  subparsers = parser.add_subparsers(dest="command", required=True)
+
+  connect_cluster_parser = subparsers.add_parser("connect-cluster", help="Add an entire pve cluster to this machine for use.", parents=[base_parser])
+  connect_cluster_parser.add_argument("--pve-host", type=str, help="PVE Host to connect to and add the entire cluster for the local machine.", required=True)
+  # todo: try and get the pve cloud domain from the cluster in case it is already initialized.
+  connect_cluster_parser.add_argument("--pve-cloud-domain", type=str, help="PVE Cloud domain the hosts are part of / should be initialized under.", required=True)
+  connect_cluster_parser.add_argument("--force", action="store_true", help="Will readd the cluster if set.")
+  connect_cluster_parser.set_defaults(func=connect_cluster)
+
+
+  args = parser.parse_args()
+  args.func(args)
+
+
+if __name__ == "__main__":
+  main()

pve_cloud/cli/pvclu.py

@@ -5,12 +5,12 @@ import socket
 
 
 def get_cloud_domain(target_pve):
-  with open(os.path.expanduser("~/.pve-inventory.yaml"), "r") as f:
+  with open(os.path.expanduser("~/.pve-cloud-dyn-inv.yaml"), "r") as f:
     pve_inventory = yaml.safe_load(f)
 
   for pve_cloud in pve_inventory:
     for pve_cluster in pve_inventory[pve_cloud]:
-      if pve_cluster == target_pve:
+      if pve_cluster + "." + pve_cloud == target_pve:
         return pve_cloud
   
   raise Exception(f"Could not identify cloud domain for {target_pve}")
@@ -21,12 +21,12 @@ def get_cld_domain_prsr(args):
 
 
 def get_online_pve_host(target_pve):
-  with open(os.path.expanduser("~/.pve-inventory.yaml"), "r") as f:
+  with open(os.path.expanduser("~/.pve-cloud-dyn-inv.yaml"), "r") as f:
     pve_inventory = yaml.safe_load(f)
 
   for pve_cloud in pve_inventory:
     for pve_cluster in pve_inventory[pve_cloud]:
-      if pve_cluster == target_pve:
+      if pve_cluster + "." + pve_cloud == target_pve:
         for pve_host in pve_inventory[pve_cloud][pve_cluster]:
           # check if host is available
           pve_host_ip = pve_inventory[pve_cloud][pve_cluster][pve_host]["ansible_host"]

pve_cloud/lib/inventory.py

@@ -0,0 +1,3 @@
+
+
+def get_pve_cloud_inventory()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-pve-cloud
-Version: 0.0.2
+Version: 0.0.4
 Author-email: Tobias Huebner <tobias.huebner@vmzberlin.com>
 License-Expression: GPL-3.0-or-later
 License-File: LICENSE
@@ -8,4 +8,5 @@ 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: proxmoxer==2.2.0
 Dynamic: license-file

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

@@ -1,4 +1,6 @@
-pve_cloud/cli/pvclu.py,sha256=xv833SHRvcUycwIsZ6gDKQkvvz-5igzwwdqGaN2NpF8,2454
+pve_cloud/cli/pvcli.py,sha256=inR4KMI63nWufPYoM9udxgh3CN10xyQlJw_kGzVw7DQ,3166
+pve_cloud/cli/pvclu.py,sha256=D5R5Gzo_iSImPTtEHVJWZoFr66uijeFbb-k6ODz_71o,2498
+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
 pve_cloud/orm/migrations/env.py,sha256=xtOgjF1KLmRUkG1-yb4eV4F2JzarDKFU1tdWJovNHDc,2200
@@ -9,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.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-py_pve_cloud-0.0.2.dist-info/METADATA,sha256=BiWtfqaXFEJLAbYO8j_F-7oZRc2CkKs8u84dBXB5yx4,330
-py_pve_cloud-0.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-py_pve_cloud-0.0.2.dist-info/entry_points.txt,sha256=xkrTpjvvswH0Vo_AQ8_BNkTbaUbrCBWYaavWQjjnykU,51
-py_pve_cloud-0.0.2.dist-info/top_level.txt,sha256=mpT7ttGRyZJVt_obhPLBHyIBcjKhUdJ-qVsMEVX5WJg,10
-py_pve_cloud-0.0.2.dist-info/RECORD,,
+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.0.2.dist-info → py_pve_cloud-0.0.4.dist-info}/entry_points.txt

@@ -1,2 +1,3 @@
 [console_scripts]
+pvcli = pve_cloud.cli.pvcli:main
 pvclu = pve_cloud.cli.pvclu:main