49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import etcd
|
|
|
|
from helper import check_otp, get_next_id
|
|
from flask import Flask
|
|
from flask_restful import Resource, Api, reqparse
|
|
from decouple import config
|
|
from enums import VmStatus
|
|
|
|
app = Flask(__name__)
|
|
api = Api(app)
|
|
|
|
etcd_client = etcd.Client(host=config("ETCD_HOST"), port=int(config("ETCD_PORT")))
|
|
etcd_lock = etcd.Lock(etcd_client, 'etcd_lock')
|
|
|
|
createvm_argparser = reqparse.RequestParser()
|
|
createvm_argparser.add_argument("name", type=str, required=True)
|
|
createvm_argparser.add_argument("realm", type=str, required=True)
|
|
createvm_argparser.add_argument("seed", type=str, required=True)
|
|
createvm_argparser.add_argument("specs", type=dict, required=True)
|
|
|
|
|
|
class CreateVM(Resource):
|
|
def post(self):
|
|
createvm_args = createvm_argparser.parse_args()
|
|
name, realm, seed, specs = createvm_args.name, createvm_args.realm,\
|
|
createvm_args.seed, createvm_args.specs
|
|
|
|
if check_otp(name, realm, seed) == 200:
|
|
# User is good
|
|
|
|
# Block until lock is acquired. There is no timeout associated with this lock.
|
|
with etcd_lock:
|
|
next_vm_id = get_next_id(etcd_client, "/v1/vm/")
|
|
|
|
vm_entry = {"owner": name,
|
|
"specs": specs,
|
|
"status": VmStatus.REQUESTED_NEW}
|
|
|
|
etcd_client.write(f"/v1/vm/{next_vm_id}", vm_entry)
|
|
|
|
return {'message': "VM Creation Queued"}, 200
|
|
else:
|
|
return {'message': 'Invalid Credentials'}, 400
|
|
|
|
|
|
api.add_resource(CreateVM, '/vm/create')
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|