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

fix vm_power parameter, add basic vm expire email

Browse source
This commit is contained in:
Jordan Rodgers 2018-01-24 21:47:18 -05:00
parent 3fd24e42ea
commit d0368e743f
3 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
proxstar/__init__.py
View file

@ -150,7 +150,7 @@ def vm_power(vmid, action):
proxmox, user) or 'rtp' in session['userinfo']['groups']: proxmox, user) or 'rtp' in session['userinfo']['groups']:
if action == 'start': if action == 'start':
config = get_vm_config(proxmox, vmid) config = get_vm_config(proxmox, vmid)
usage_check = check_user_usage(proxmox, user, config['cores'], usage_check = check_user_usage(proxmox, db, user, config['cores'],
config['memory'], 0) config['memory'], 0)
if usage_check: if usage_check:
return usage_check return usage_check

15
proxstar/mail.py
View file

@ -18,13 +18,14 @@ def send_email(toaddr, subject, body):
server.quit() server.quit()
def send_vm_expire_email(user, name, days): def send_vm_expire_email(user, vms):
toaddr = "{}@csh.rit.edu".format(user) toaddr = "{}@csh.rit.edu".format(user)
subject = 'Proxstar VM Expiration Notice' subject = 'Proxstar VM Expiration Notice'
if days != 1: body = "The following VMs in Proxstar are expiring soon:\n\n"
body = "{} is expiring in {} days. If you would like to keep this VM, please log into Proxstar and renew it.".format( for vm in vms:
name, days) if vm[1] != 1:
else: body += " - {} in {} days\n".format(vm[0], vm[1])
body = "{} is expiring in {} day. If you would like to keep this VM, please log into Proxstar and renew it.".format( else:
name, days) body += " - {} in {} day\n".format(vm[0], vm[1])
body += "\nPlease login to Proxstar and renew any VMs you would like to keep."
send_email(toaddr, subject, body) send_email(toaddr, subject, body)

5
proxstar/tasks.py
View file

@ -78,6 +78,7 @@ def process_expiring_vms_task():
starrs = connect_starrs() starrs = connect_starrs()
pools = get_pools(proxmox) pools = get_pools(proxmox)
for pool in pools: for pool in pools:
expiring_vms = []
vms = get_vms_for_user(proxmox, pool) vms = get_vms_for_user(proxmox, pool)
for vm in vms: for vm in vms:
vmid = vm['vmid'] vmid = vm['vmid']
@ -86,4 +87,6 @@ def process_expiring_vms_task():
days = (expire - datetime.date.today()).days days = (expire - datetime.date.today()).days
if days in [10, 7, 3, 1]: if days in [10, 7, 3, 1]:
name = get_vm_config(proxmox, vmid)['name'] name = get_vm_config(proxmox, vmid)['name']
send_vm_expire_email('com6056', name, days) expiring_vms.append([name, days])
if expiring_vms:
send_vm_expire_email('com6056', expiring_vms)