mirror of
https://github.com/ComputerScienceHouse/proxstar.git
synced 2025-03-09 15:40:09 +00:00
132 lines
4.4 KiB
Python
132 lines
4.4 KiB
Python
import os
|
|
import paramiko
|
|
from flask import Flask
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from proxstar.db 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)
|
|
for pool in pools:
|
|
expiring_vms = []
|
|
vms = get_vms_for_user(proxmox, 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, cores, memory):
|
|
with app.app_context():
|
|
proxmox = connect_proxmox()
|
|
starrs = connect_starrs()
|
|
db = connect_db()
|
|
vmid, mac = clone_vm(proxmox, template_id, name, user)
|
|
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'])
|
|
change_vm_cpu(proxmox, vmid, cores)
|
|
change_vm_mem(proxmox, vmid, memory)
|
|
time.sleep(60)
|
|
change_vm_power(proxmox, vmid, 'start')
|
|
client = paramiko.SSHClient()
|
|
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
retry = 0
|
|
while retry < 30:
|
|
try:
|
|
client.connect(ip, username='root', password='')
|
|
break
|
|
except:
|
|
retry += 1
|
|
time.sleep(3)
|
|
stdin, stdout, stderr = client.exec_command('ls')
|
|
for line in stdout:
|
|
print('... ' + line.strip('\n'))
|
|
client.close()
|