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

store ignored pools in db, check ignored pools when getting all pools

Browse source
This commit is contained in:
Jordan Rodgers 2018-02-02 15:18:50 -05:00
parent 51de0b69af
commit 8fbae0a071
5 changed files with 39 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

9
proxstar/db.py
View file

@ -2,7 +2,7 @@ import datetime
from sqlalchemy import exists
from dateutil.relativedelta import relativedelta
from proxstar.ldapdb import *
from proxstar.models import VM_Expiration, Usage_Limit, Pool_Cache, Base
from proxstar.models import VM_Expiration, Usage_Limit, Pool_Cache, Ignored_Pools, Base
def get_vm_expire(db, vmid, months):
@ -124,3 +124,10 @@ def get_pool_cache(db):
pool_dict['percents'] = pool.percents
pools.append(pool_dict)
return pools
def get_ignored_pools(db):
ignored_pools = []
for pool in db.query(Ignored_Pools).all():
ignored_pools.append(pool.id)
return ignored_pools