mirror of
https://github.com/ComputerScienceHouse/proxstar.git
synced 2025-03-09 15:40:09 +00:00
split into 2 functions
This commit is contained in:
parent
4f5753aec5
commit
1ca123b612
1 changed files with 79 additions and 72 deletions
|
@ -566,86 +566,93 @@ def set_boot_order(vmid):
|
||||||
return '', 403
|
return '', 403
|
||||||
|
|
||||||
|
|
||||||
@app.route('/vm/create', methods=['GET', 'POST'])
|
@app.route('/vm/create', methods=['GET'])
|
||||||
@auth.oidc_auth
|
@auth.oidc_auth
|
||||||
def create():
|
def get_create():
|
||||||
user = User(session['userinfo']['preferred_username'])
|
user = User(session['userinfo']['preferred_username'])
|
||||||
proxmox = connect_proxmox()
|
proxmox = connect_proxmox()
|
||||||
if user.active or user.rtp:
|
if user.active or user.rtp:
|
||||||
if request.method == 'GET':
|
stored_isos = get_isos(proxmox, app.config['PROXMOX_ISO_STORAGE'])
|
||||||
stored_isos = get_isos(proxmox, app.config['PROXMOX_ISO_STORAGE'])
|
pools = get_pools(proxmox, db)
|
||||||
pools = get_pools(proxmox, db)
|
for pool in get_shared_pools(db, user.name, True):
|
||||||
for pool in get_shared_pools(db, user.name, True):
|
pools.append(pool.name)
|
||||||
pools.append(pool.name)
|
templates = get_templates(db)
|
||||||
templates = get_templates(db)
|
return render_template(
|
||||||
return render_template(
|
'create_vm.html',
|
||||||
'create_vm.html',
|
user=user,
|
||||||
user=user,
|
usage=user.usage,
|
||||||
usage=user.usage,
|
limits=user.limits,
|
||||||
limits=user.limits,
|
percents=user.usage_percent,
|
||||||
percents=user.usage_percent,
|
isos=stored_isos,
|
||||||
isos=stored_isos,
|
pools=pools,
|
||||||
pools=pools,
|
templates=templates,
|
||||||
templates=templates,
|
)
|
||||||
|
else:
|
||||||
|
return '', 403
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/vm/create', methods=['POST'])
|
||||||
|
@auth.oidc_auth
|
||||||
|
def create():
|
||||||
|
user = User(session['userinfo']['preferred_username'])
|
||||||
|
if user.active or user.rtp:
|
||||||
|
name = request.form['name'].lower()
|
||||||
|
cores = request.form['cores']
|
||||||
|
memory = request.form['mem']
|
||||||
|
disk = request.form['disk']
|
||||||
|
username = request.form['user']
|
||||||
|
## CHECK STUFF DEAR GOD
|
||||||
|
if int(cores) <= 0 or int(memory) <= 0 or int(disk) <= 0 or user == '':
|
||||||
|
return (
|
||||||
|
'VM creation with cores and/or mem and/or disk values that are less than 0',
|
||||||
|
400,
|
||||||
)
|
)
|
||||||
elif request.method == 'POST':
|
|
||||||
name = request.form['name'].lower()
|
|
||||||
cores = request.form['cores']
|
|
||||||
memory = request.form['mem']
|
|
||||||
disk = request.form['disk']
|
|
||||||
username = request.form['user']
|
|
||||||
## CHECK STUFF DEAR GOD
|
|
||||||
if int(cores) <= 0 or int(memory) <= 0 or int(disk) <= 0 or user == '':
|
|
||||||
return (
|
|
||||||
'VM creation with cores and/or mem and/or disk values that are less than 0',
|
|
||||||
400,
|
|
||||||
)
|
|
||||||
|
|
||||||
template = request.form['template']
|
template = request.form['template']
|
||||||
iso = request.form['iso']
|
iso = request.form['iso']
|
||||||
ssh_key = request.form['ssh_key']
|
ssh_key = request.form['ssh_key']
|
||||||
if iso != 'none':
|
if iso != 'none':
|
||||||
iso = '{}:iso/{}'.format(app.config['PROXMOX_ISO_STORAGE'], iso)
|
iso = '{}:iso/{}'.format(app.config['PROXMOX_ISO_STORAGE'], iso)
|
||||||
if not user.rtp:
|
if not user.rtp:
|
||||||
|
if template == 'none':
|
||||||
|
usage_check = user.check_usage(0, 0, disk)
|
||||||
|
else:
|
||||||
|
usage_check = user.check_usage(cores, memory, disk)
|
||||||
|
username = user.name
|
||||||
|
else:
|
||||||
|
usage_check = None
|
||||||
|
if usage_check:
|
||||||
|
return usage_check
|
||||||
|
else:
|
||||||
|
valid, available = (
|
||||||
|
check_hostname(starrs, name) if app.config['USE_STARRS'] else (True, True)
|
||||||
|
)
|
||||||
|
|
||||||
|
if valid and available:
|
||||||
if template == 'none':
|
if template == 'none':
|
||||||
usage_check = user.check_usage(0, 0, disk)
|
q.enqueue(
|
||||||
|
create_vm_task,
|
||||||
|
username,
|
||||||
|
name,
|
||||||
|
cores,
|
||||||
|
memory,
|
||||||
|
disk,
|
||||||
|
iso,
|
||||||
|
job_timeout=300,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
usage_check = user.check_usage(cores, memory, disk)
|
q.enqueue(
|
||||||
username = user.name
|
setup_template_task,
|
||||||
else:
|
template,
|
||||||
usage_check = None
|
name,
|
||||||
if usage_check:
|
username,
|
||||||
return usage_check
|
ssh_key,
|
||||||
else:
|
cores,
|
||||||
valid, available = (
|
memory,
|
||||||
check_hostname(starrs, name) if app.config['USE_STARRS'] else (True, True)
|
job_timeout=600,
|
||||||
)
|
)
|
||||||
|
return '', 200
|
||||||
if valid and available:
|
return '', 200
|
||||||
if template == 'none':
|
|
||||||
q.enqueue(
|
|
||||||
create_vm_task,
|
|
||||||
username,
|
|
||||||
name,
|
|
||||||
cores,
|
|
||||||
memory,
|
|
||||||
disk,
|
|
||||||
iso,
|
|
||||||
job_timeout=300,
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
q.enqueue(
|
|
||||||
setup_template_task,
|
|
||||||
template,
|
|
||||||
name,
|
|
||||||
username,
|
|
||||||
ssh_key,
|
|
||||||
cores,
|
|
||||||
memory,
|
|
||||||
job_timeout=600,
|
|
||||||
)
|
|
||||||
return '', 200
|
|
||||||
return '', 200
|
|
||||||
else:
|
else:
|
||||||
return '', 403
|
return '', 403
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue