proxstar/app.py

import os
import time
import psycopg2
import subprocess
from db import *
from starrs import *
from proxmox import *
from flask import Flask, render_template, request, redirect, send_from_directory

app = Flask(__name__)

config = os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config.py")

app.config.from_pyfile(config)

app.config["GIT_REVISION"] = subprocess.check_output(
    ['git', 'rev-parse', '--short', 'HEAD']).decode('utf-8').rstrip()

user = 'proxstar'


@app.route("/")
def list_vms():
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    vms = get_vms_for_user(proxmox, user)
    for vm in vms:
        if 'name' not in vm:
            vms.remove(vm)
    vms = sorted(vms, key=lambda k: k['name'])
    return render_template('list_vms.html', username='com6056', vms=vms)


@app.route("/isos")
def isos():
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    isos = get_isos(proxmox, app.config['PROXMOX_ISO_STORAGE'])
    return ','.join(isos)


@app.route("/hostname/<string:name>")
def hostname(name):
    starrs = connect_starrs(
        app.config['STARRS_DB_NAME'], app.config['STARRS_DB_USER'],
        app.config['STARRS_DB_HOST'], app.config['STARRS_DB_PASS'])
    result = check_hostname(starrs, name)
    return str(result)


@app.route("/vm/<string:vmid>")
def vm_details(vmid):
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    starrs = connect_starrs(
        app.config['STARRS_DB_NAME'], app.config['STARRS_DB_USER'],
        app.config['STARRS_DB_HOST'], app.config['STARRS_DB_PASS'])
    if int(vmid) in get_user_allowed_vms(proxmox, user):
        vm = get_vm(proxmox, vmid)
        vm['vmid'] = vmid
        vm['config'] = get_vm_config(proxmox, vmid)
        vm['disks'] = get_vm_disks(proxmox, vmid, config=vm['config'])
        vm['iso'] = get_vm_iso(proxmox, vmid, config=vm['config'])
        vm['interfaces'] = []
        for interface in get_vm_interfaces(proxmox, vm['vmid'], config=vm['config']):
            vm['interfaces'].append([interface[0], get_ip_for_mac(starrs, interface[1])[0][3]])
        vm['expire'] = get_vm_expire(vmid, app.config['VM_EXPIRE_MONTHS']).strftime('%m/%d/%Y')
        return render_template('vm_details.html', username='com6056', vm=vm)
    else:
        return '', 403


@app.route("/vm/<string:vmid>/power/<string:action>", methods=['POST'])
def vm_power(vmid, action):
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    if int(vmid) in get_user_allowed_vms(proxmox, user):
        change_vm_power(proxmox, vmid, action)
        return '', 200
    else:
        return '', 403


@app.route("/vm/<string:vmid>/renew", methods=['POST'])
def vm_renew(vmid):
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    if int(vmid) in get_user_allowed_vms(proxmox, user):
        renew_vm_expire(vmid, app.config['VM_EXPIRE_MONTHS'])
        return '', 200
    else:
        return '', 403


@app.route("/vm/<string:vmid>/eject", methods=['POST'])
def iso_eject(vmid):
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    if int(vmid) in get_user_allowed_vms(proxmox, user):
        eject_vm_iso(proxmox, vmid)
        return '', 200
    else:
        return '', 403


@app.route("/vm/<string:vmid>/mount/<string:iso>", methods=['POST'])
def iso_mount(vmid, iso):
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    if int(vmid) in get_user_allowed_vms(proxmox, user):
        iso = "{}:iso/{}".format(app.config['PROXMOX_ISO_STORAGE'], iso)
        mount_vm_iso(proxmox, vmid, iso)
        return '', 200
    else:
        return '', 403


@app.route("/vm/<string:vmid>/delete", methods=['POST'])
def delete(vmid):
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    starrs = connect_starrs(
        app.config['STARRS_DB_NAME'], app.config['STARRS_DB_USER'],
        app.config['STARRS_DB_HOST'], app.config['STARRS_DB_PASS'])
    if int(vmid) in get_user_allowed_vms(proxmox, user):
        vmname = get_vm_config(proxmox, vmid)['name']
        delete_vm(proxmox, starrs, vmid)
        delete_starrs(starrs, vmname)
        delete_vm_expire(vmid)
        return '', 200
    else:
        return '', 403


@app.route("/vm/create", methods=['GET', 'POST'])
def create():
    proxmox = connect_proxmox(app.config['PROXMOX_HOST'],
                              app.config['PROXMOX_USER'],
                              app.config['PROXMOX_PASS'])
    starrs = connect_starrs(
        app.config['STARRS_DB_NAME'], app.config['STARRS_DB_USER'],
        app.config['STARRS_DB_HOST'], app.config['STARRS_DB_PASS'])
    if request.method == 'GET':
        usage = get_user_usage(proxmox, 'proxstar')
        limits = get_user_usage_limits(user)
        full_limits = check_user_limit(proxmox, user, usage, limits)
        percents = get_user_usage_percent(proxmox, usage, limits)
        isos = get_isos(proxmox, app.config['PROXMOX_ISO_STORAGE'])
        return render_template(
            'create.html',
            username='com6056',
            usage=usage,
            limits=limits,
            full_limits=full_limits,
            percents=percents,
            isos=isos)
    elif request.method == 'POST':
        name = request.form['name']
        cores = request.form['cores']
        memory = request.form['mem']
        disk = request.form['disk']
        iso = request.form['iso']
        if iso != 'none':
            iso = "{}:iso/{}".format(app.config['PROXMOX_ISO_STORAGE'], iso)
        usage_check = check_user_usage(proxmox, user, cores, memory, disk)
        if usage_check:
            return usage_check
        else:
            if not check_hostname(starrs, name):
                vmid, mac = create_vm(proxmox, starrs, user, name, cores, memory,
                                      disk, iso)
                register_starrs(starrs, name, user, mac,
                                get_next_ip(starrs,
                                            app.config['STARRS_IP_RANGE'])[0][0])
                get_vm_expire(vmid, app.config['VM_EXPIRE_MONTHS'])
                return vmid


@app.route('/novnc/<path:path>')
def send_novnc(path):
    return send_from_directory('static/novnc-pve/novnc', path)


if __name__ == "__main__":
    app.run(debug=True)