external/proxstar
1
0
Fork 0
mirror of https://github.com/ComputerScienceHouse/proxstar.git synced 2025-03-09 15:40:09 +00:00
Code Issues Releases Wiki Activity

change db to postgres, fix rtp functions, change how db is connected

Browse source
This commit is contained in:
Jordan Rodgers 2018-01-23 23:12:19 -05:00
parent af451f6176
commit 1d094b1e25
10 changed files with 113 additions and 113 deletions
Download patch file Download diff file Expand all files Collapse all files

41
proxstar/tasks.py
View file

@ -1,34 +1,59 @@
import os
from flask import Flask
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from proxstar.db import *
from proxstar.starrs import *
from proxstar.proxmox import *
from flask import Flask, current_app
app = Flask(__name__)
config = os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config.py")
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, starrs, user, name, cores, memory, disk,
iso)
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'])[0][0])
get_vm_expire(vmid, app.config['VM_EXPIRE_MONTHS'])
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, starrs, vmid)
delete_vm(proxmox, vmid)
delete_starrs(starrs, vmname)
delete_vm_expire(vmid)
delete_vm_expire(db, vmid)
def process_expired_vms_task():
@ -37,7 +62,7 @@ def process_expired_vms_task():
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)