uncloud/ucloud-vm
2
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

move running_vm_statuses out of VMStatus

Browse source
This commit is contained in:
ahmadbilalkhalid 2019-07-25 16:21:38 +05:00
parent 4541ac13bf
commit c387bd3cfb
2 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
main.py
View file

@ -21,7 +21,7 @@ from typing import Union
from functools import wraps from functools import wraps
from decouple import config from decouple import config
from datetime import datetime from datetime import datetime
from ucloud_common.enums import VMStatus, RUNNING_VM_STATUES from ucloud_common.enums import VMStatus, RUNNING_VM_STATUSES
running_vms = [] running_vms = []
vnc_port_pool = list(range(0, 100)) vnc_port_pool = list(range(0, 100))
@ -67,7 +67,7 @@ def goodbye(host):
for vm in vms: for vm in vms:
vm.value["hostname"] = "" vm.value["hostname"] = ""
if vm.value["status"] in VMStatus.running_vm_statuses: if vm.value["status"] in RUNNING_VM_STATUSES:
vm.value["status"] = VMStatus.requested_start vm.value["status"] = VMStatus.requested_start
client.put(vm.key, vm.value, value_in_json=True) client.put(vm.key, vm.value, value_in_json=True)

9
ucloud_common/enums.py
View file

@ -23,9 +23,12 @@ class VMStatus(Enum):
running = "RUNNING" running = "RUNNING"
suspended = "SUSPENDED" suspended = "SUSPENDED"
running_vm_statuses = (requested_shutdown, requested_suspend,
requested_resume, requested_migrate,
running, suspended)
class HostStatus(Enum): class HostStatus(Enum):
alive = "ALIVE" alive = "ALIVE"
dead = "DEAD" dead = "DEAD"
RUNNING_VM_STATUSES = [VMStatus.requested_shutdown, VMStatus.requested_suspend,
VMStatus.requested_resume, VMStatus.requested_migrate,
VMStatus.running, VMStatus.suspended]