proxstar/proxstar/tasks.py

import os
import paramiko
from flask import Flask
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from proxstar.db import *
from proxstar.util import *
from proxstar.mail import *
from proxstar.starrs import *
from proxstar.proxmox import *

app = Flask(__name__)
if os.path.exists(
        os.path.join(
            app.config.get('ROOT_DIR', os.getcwd()), "config.local.py")):
    config = os.path.join(
        app.config.get('ROOT_DIR', os.getcwd()), "config.local.py")
else:
    config = os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config.py")
app.config.from_pyfile(config)


def connect_db():
    engine = create_engine(app.config['SQLALCHEMY_DATABASE_URI'])
    Base.metadata.bind = engine
    DBSession = sessionmaker(bind=engine)
    db = DBSession()
    return db


def connect_starrs():
    starrs = psycopg2.connect(
        "dbname='{}' user='{}' host='{}' password='{}'".format(
            app.config['STARRS_DB_NAME'], app.config['STARRS_DB_USER'],
            app.config['STARRS_DB_HOST'], app.config['STARRS_DB_PASS']))
    return starrs


def create_vm_task(user, name, cores, memory, disk, iso):
    with app.app_context():
        proxmox = connect_proxmox()
        db = connect_db()
        starrs = connect_starrs()
        vmid, mac = create_vm(proxmox, user, name, cores, memory, disk, iso)
        register_starrs(starrs, name, app.config['STARRS_USER'], mac,
                        get_next_ip(starrs, app.config['STARRS_IP_RANGE']))
        get_vm_expire(db, vmid, app.config['VM_EXPIRE_MONTHS'])


def delete_vm_task(vmid):
    with app.app_context():
        proxmox = connect_proxmox()
        db = connect_db()
        starrs = connect_starrs()
        vmname = get_vm_config(proxmox, vmid)['name']
        delete_vm(proxmox, vmid)
        delete_starrs(starrs, vmname)
        delete_vm_expire(db, vmid)


def process_expired_vms_task():
    with app.app_context():
        proxmox = connect_proxmox()
        starrs = connect_starrs()
        expired_vms = get_expired_vms()
        print(expired_vms)

    #    for vmid in expired_vms:
    #        vmname = get_vm_config(proxmox, vmid)['name']
    #        delete_vm(proxmox, starrs, vmid)
    #        delete_starrs(starrs, vmname)
    #        delete_vm_expire(vmid)


def process_expiring_vms_task():
    with app.app_context():
        proxmox = connect_proxmox()
        db = connect_db()
        starrs = connect_starrs()
        pools = get_pools(proxmox, db)
        for pool in pools:
            expiring_vms = []
            vms = get_vms_for_user(proxmox, db, pool)
            for vm in vms:
                vmid = vm['vmid']
                expire = get_vm_expire(db, vmid,
                                       app.config['VM_EXPIRE_MONTHS'])
                days = (expire - datetime.date.today()).days
                if days in [10, 7, 3, 1, 0]:
                    name = get_vm_config(proxmox, vmid)['name']
                    expiring_vms.append([name, days])
                    if days == 0:
                        #change_vm_power(proxmox, vmid, 'stop')
                        pass
            if expiring_vms:
                send_vm_expire_email('com6056', expiring_vms)


def generate_pool_cache_task():
    with app.app_context():
        proxmox = connect_proxmox()
        db = connect_db()
        pools = get_vms_for_rtp(proxmox, db)
        store_pool_cache(db, pools)


def setup_template(template_id, name, user, password, cores, memory):
    with app.app_context():
        proxmox = connect_proxmox()
        starrs = connect_starrs()
        db = connect_db()
        template = get_template(db, template_id)
        print('clone')
        vmid, mac = clone_vm(proxmox, template_id, name, user)
        print('starrs')
        ip = get_next_ip(starrs, app.config['STARRS_IP_RANGE'])
        register_starrs(starrs, name, app.config['STARRS_USER'], mac, ip)
        get_vm_expire(db, vmid, app.config['VM_EXPIRE_MONTHS'])
        print('set cpu mem')
        change_vm_cpu(proxmox, vmid, cores)
        change_vm_mem(proxmox, vmid, memory)
        print('wait to start')
        time.sleep(90)
        print('start')
        change_vm_power(proxmox, vmid, 'start')
        print('wait after start')
        time.sleep(20)
        client = paramiko.SSHClient()
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        retry = 0
        while retry < 30:
            try:
                print('try ssh')
                print(ip)
                print(template['username'])
                print(template['password'])
                client.connect(
                    ip,
                    username=template['username'],
                    password=template['password'])
                break
            except:
                print('ssh failed')
                retry += 1
                time.sleep(3)
        print('start commands')
        stdin, stdout, stderr = client.exec_command("useradd {}".format(user))
        exit_status = stdout.channel.recv_exit_status()
        root_password = gen_password(32)
        stdin, stdout, stderr = client.exec_command(
            "echo '{}' | passwd root --stdin".format(root_password))
        exit_status = stdout.channel.recv_exit_status()
        stdin, stdout, stderr = client.exec_command(
            "echo '{}' | passwd '{}' --stdin".format(password, user))
        exit_status = stdout.channel.recv_exit_status()
        stdin, stdout, stderr = client.exec_command(
            "echo '{} ALL=(ALL:ALL) ALL' | sudo EDITOR='tee -a' visudo".format(
                user))
        exit_status = stdout.channel.recv_exit_status()
        print('done')
        client.close()