diff --git a/main.py b/main.py index 28d0f23..c413fbd 100644 --- a/main.py +++ b/main.py @@ -21,7 +21,7 @@ from typing import Union from functools import wraps from decouple import config from datetime import datetime -from ucloud_common.enums import VMStatus +from ucloud_common.enums import VMStatus, RUNNING_VM_STATUES running_vms = [] vnc_port_pool = list(range(0, 100)) @@ -67,7 +67,7 @@ def goodbye(host): for vm in vms: vm.value["hostname"] = "" - if vm.value["status"] in VMStatus.running_vm_statuses: + if vm.value["status"] in RUNNING_VM_STATUES: vm.value["status"] = VMStatus.requested_start client.put(vm.key, vm.value, value_in_json=True) diff --git a/ucloud_common/enums.py b/ucloud_common/enums.py index a1c0e17..18fe8ac 100644 --- a/ucloud_common/enums.py +++ b/ucloud_common/enums.py @@ -23,9 +23,12 @@ class VMStatus(Enum): running = "RUNNING" suspended = "SUSPENDED" - running_vm_statuses = [requested_shutdown, requested_suspend, - requested_resume, requested_migrate, - running, suspended] + class HostStatus(Enum): alive = "ALIVE" - dead = "DEAD" \ No newline at end of file + dead = "DEAD" + + +RUNNING_VM_STATUES = [VMStatus.requested_shutdown, VMStatus.requested_suspend, + VMStatus.requested_resume, VMStatus.requested_migrate, + VMStatus.running, VMStatus.suspended] \ No newline at end of file