This vm_restart.py script is used to restart VMs or LXC containers on a Proxmox host.

It takes two command line arguments: the type of client (vm or lxc) and a list of client IDs. example: ./cron_scripts/power/vm_restart.py "lxc" "['110', '113', '114', '120']"
2025-07-01 10:03:52 -04:00
parent 7929f590a0
commit 536fbb4553
2 changed files with 161 additions and 0 deletions
@@ -0,0 +1,91 @@
+#!/usr/bin/python3
+import sys, ast
+from subprocess import check_output, run
+from time import sleep
+from datetime import datetime
+
+
+def get_time():
+    ts1 = str(datetime.now()).split('.')[0]
+    return ts1
+
+
+def call_shutdown(client_type, clientid):
+    if client_type == 'vm':
+        run(['/usr/sbin/qm', 'shutdown', clientid])
+    elif client_type == 'lxc':
+        run(['/usr/bin/lxc-stop', clientid])
+    else:
+        print('Unable to stop {clientid}. Invalid client type or client id')
+        print(f'client_type: {client_type}')
+        print(f'client id: {clientid}')
+
+
+def call_startup(client_type, clientid):
+    if client_type == 'vm':
+        run(['/usr/sbin/qm', 'start', clientid])
+    elif client_type == 'lxc':
+        run(['/usr/bin/lxc-start', clientid])
+    else:
+        print('invalid client type or client id')
+
+
+def get_status(client_type, clientid, vmstatus, status_wanted):
+    checker = None
+    if client_type == 'lxc':
+        checker = check_output(['lxc-info', clientid])
+    elif client_type == 'vm':
+        checker = check_output(['/usr/sbin/qm', 'status', clientid])
+    while vmstatus != status_wanted:
+        sleep(5)
+        vmoutput = checker
+        if (str.encode(status_wanted)) in vmoutput.upper():
+            vmstatus = status_wanted
+    return status_wanted
+
+
+def print_state(vmtype, vmhost, vmstatus):
+    print(f'{vmtype} {vmhost} is {vmstatus}')
+
+
+# This script is used to restart VMs or LXC containers on a Proxmox host.
+# It takes two command line arguments: the type of client (vm or lxc) and a list of client IDs.
+# example: ./cron_scripts/power/vm_restart.py "lxc" "['110', '113', '114', '120']"
+
+
+def main():
+    client_type = None
+    client_list = None
+    try:
+        client_type = sys.argv[1]
+        client_list = ast.literal_eval(sys.argv[2])
+    except Exception as e:
+        print(f'Unable to process cli arguments. {e}')
+    else:
+        print(f'{client_type} {client_list}')
+    timestamp = get_time()
+    print(f'Start time: {timestamp}')
+    for clientid in client_list:
+        try:
+            print(f'Shutting down {clientid}')
+            call_shutdown(client_type, clientid)
+            vresult = get_status(client_type, clientid, 'RUNNING', 'STOPPED')
+        except Exception as e:
+            print(f'Unable to stop container {clientid}: {e}')
+        else:
+            print_state(client_type.upper(), clientid, vresult)
+        try:
+            print(f'Starting {clientid}')
+            call_startup(client_type, clientid)
+            vresult = get_status(client_type, clientid, 'STOPPED', 'RUNNING')
+        except Exception as e:
+            print(f'Unable to start container {clientid}: {e}')
+        else:
+            print_state(client_type.upper(), clientid, vresult)
+    timestamp = get_time()
+    print(f'End time: {timestamp}')
+
+
+if __name__ == '__main__':
+    main()
+