proxstar/cron.py

47 lines
1.3 KiB
Python

import os
from db import *
from starrs import *
from proxmox import *
from flask import Flask, current_app
app = Flask(__name__)
config = os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config.py")
app.config.from_pyfile(config)
def process_expired_vms():
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 get_rrd_graphs():
proxmox = connect_proxmox()
pools = get_pools(proxmox)
for pool in pools:
vms = proxmox.pools(pool).get()['members']
for vm in vms:
vm_dir = "rrd/{}".format(vm['vmid'])
if not os.path.exists(vm_dir):
os.makedirs(vm_dir)
sources = [
'cpu', 'mem', 'netin', 'netout', 'diskread', 'diskwrite'
]
for source in sources:
image = get_rrd_for_vm(proxmox, vm['vmid'], source, 'day')
with open("rrd/{}/{}.png".format(vm['vmid'], source),
'wb') as f:
f.write(image.encode('raw_unicode_escape'))
with app.app_context():
process_expired_vms()
get_rrd_graphs()