diff --git a/enums.py b/enums.py deleted file mode 100644 index 0b21b42..0000000 --- a/enums.py +++ /dev/null @@ -1,6 +0,0 @@ -from enum import Enum - - -class VmStatus(Enum): - REQUESTED_NEW = "REQUESTED_NEW" - SCHEDULED_DEPLOY = "SCHEDULED_DEPLOY" diff --git a/main.py b/main.py index 5724b32..41e53b2 100644 --- a/main.py +++ b/main.py @@ -5,7 +5,6 @@ from helper import check_otp from flask import Flask from flask_restful import Resource, Api, reqparse from decouple import config -from enums import VmStatus from uuid import uuid4 app = Flask(__name__) @@ -35,7 +34,7 @@ class CreateVM(Resource): vm_entry = {"owner": name, "specs": specs, "hostname": "", - "status": VmStatus.REQUESTED_NEW.value} + "status": "REQUESTED_NEW"} etcd_client.put(f"/v1/vm/{uuid4().hex}", json.dumps(vm_entry))