refactoring lib classes file
This commit is contained in:
Thiago Macedo
parent
505187fcaf
commit
6a4654a278
3 changed files with 161 additions and 159 deletions
160
lib/__init__.py
Normal file
160
lib/__init__.py
Normal file
|
|
@ -0,0 +1,160 @@
|
||||||
|
import os, time
|
||||||
|
import subprocess
|
||||||
|
|
||||||
|
from yaml import load, dump, Loader, Dumper
|
||||||
|
|
||||||
|
CONFIG_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||||
|
|
||||||
|
def shell(cmd):
|
||||||
|
# cmd = cmd.split(' ')
|
||||||
|
p = subprocess.Popen(cmd,
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stderr=subprocess.STDOUT,
|
||||||
|
shell=True)
|
||||||
|
# p.communicate()
|
||||||
|
return p.stdout.read()
|
||||||
|
|
||||||
|
|
||||||
|
class Usage(object):
|
||||||
|
"""
|
||||||
|
Apps Usage Control.
|
||||||
|
Will store and inform about timers run time and limits.
|
||||||
|
"""
|
||||||
|
timer = None
|
||||||
|
file = None
|
||||||
|
|
||||||
|
def __init__(self, timer):
|
||||||
|
super(Usage, self).__init__()
|
||||||
|
self.timer = timer
|
||||||
|
self.file = '%s/usage/%s' % (CONFIG_PATH, self.timer.name)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def current(self):
|
||||||
|
'''Current usage of this timer in minutes'''
|
||||||
|
if not os.path.isfile(self.file):
|
||||||
|
return 0
|
||||||
|
with open(self.file, 'r') as f:
|
||||||
|
current = f.read()
|
||||||
|
return int(current) if current else 0
|
||||||
|
|
||||||
|
def increment(self, time):
|
||||||
|
'''Increment this timer usage'''
|
||||||
|
usage = self.current + int(time)
|
||||||
|
with open(self.file, 'w') as f:
|
||||||
|
f.write(str(usage))
|
||||||
|
|
||||||
|
def release(self):
|
||||||
|
'''Remove the timer usage to restart the counter'''
|
||||||
|
os.remove(self.file)
|
||||||
|
|
||||||
|
def isOffLimit(self):
|
||||||
|
'''Is the timer usage off its limits'''
|
||||||
|
time_limit = self.timer.timeLimit
|
||||||
|
if (time_limit < 0):
|
||||||
|
return False
|
||||||
|
if self.current > time_limit:
|
||||||
|
return True
|
||||||
|
|
||||||
|
def isOffInterval(self):
|
||||||
|
'''Is the timer usage expired'''
|
||||||
|
limit_interval = self.timer.limitInterval
|
||||||
|
if (limit_interval < 0):
|
||||||
|
return False
|
||||||
|
if not os.path.exists(self.file):
|
||||||
|
return False
|
||||||
|
usage_started = os.path.getctime(self.file)
|
||||||
|
usage_expiry = usage_started + (limit_interval * 60 * 60)
|
||||||
|
return time.time() >= usage_expiry
|
||||||
|
|
||||||
|
|
||||||
|
class Timer(object):
|
||||||
|
"""
|
||||||
|
Apps Timer Setup.
|
||||||
|
Will manage a timer and proxy its configuration
|
||||||
|
"""
|
||||||
|
item = None
|
||||||
|
name = None
|
||||||
|
usage = None
|
||||||
|
|
||||||
|
def __init__(self, name, item):
|
||||||
|
super(Timer, self).__init__()
|
||||||
|
self.name = name
|
||||||
|
self.item = item
|
||||||
|
self.usage = Usage(self)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def timeLimit(self):
|
||||||
|
time_limit = self.item.get('time-limit', -1)
|
||||||
|
return int(time_limit)
|
||||||
|
@property
|
||||||
|
def limitInterval(self):
|
||||||
|
limit_interval = self.item.get('limit-interval', -1)
|
||||||
|
return float(limit_interval)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def apps(self):
|
||||||
|
item_apps = self.item.get('apps', [])
|
||||||
|
if not isinstance(item_apps, list):
|
||||||
|
item_apps = [item_apps]
|
||||||
|
apps = [app.strip() for app in item_apps]
|
||||||
|
return apps
|
||||||
|
|
||||||
|
def isRunning(self):
|
||||||
|
running = False
|
||||||
|
for app in self.apps:
|
||||||
|
cmd = 'ps aux | grep -v grep | grep "%s"' % app
|
||||||
|
res = shell(cmd)
|
||||||
|
if len(res):
|
||||||
|
running = True
|
||||||
|
return running
|
||||||
|
|
||||||
|
def block(self):
|
||||||
|
for app in self.apps:
|
||||||
|
cmd = 'killall "%s"' % app
|
||||||
|
shell(cmd)
|
||||||
|
|
||||||
|
|
||||||
|
class Config(object):
|
||||||
|
"""
|
||||||
|
Config Parser.
|
||||||
|
Read and normalize the configuration objects
|
||||||
|
"""
|
||||||
|
data = None
|
||||||
|
timers = None
|
||||||
|
mtime = 0
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
super(Config, self).__init__()
|
||||||
|
self.file = '%s/config.yaml' % CONFIG_PATH
|
||||||
|
self.reload()
|
||||||
|
|
||||||
|
def hasChanges(self):
|
||||||
|
'''Check if the config file has changes'''
|
||||||
|
mtime = os.path.getmtime(self.file)
|
||||||
|
return self.mtime != mtime
|
||||||
|
|
||||||
|
def reload(self):
|
||||||
|
'''Reload config file data'''
|
||||||
|
self.mtime = os.path.getmtime(self.file)
|
||||||
|
self.data = self.read()
|
||||||
|
self.timers = self.getTimers()
|
||||||
|
|
||||||
|
def read(self):
|
||||||
|
'''Read the config file yaml object'''
|
||||||
|
with open(self.file, 'r') as f:
|
||||||
|
data = load(f, Loader=Loader)
|
||||||
|
return data
|
||||||
|
|
||||||
|
def getTimers(self):
|
||||||
|
'''Gets a generator of Timer objects'''
|
||||||
|
config_timers = self.data.get('timers', {})
|
||||||
|
timers = []
|
||||||
|
for name, item in config_timers.items():
|
||||||
|
timers.append(Timer(name, item))
|
||||||
|
return timers
|
||||||
|
|
||||||
|
@property
|
||||||
|
def checkInterval(self):
|
||||||
|
'''How often the usage check should happen'''
|
||||||
|
interval = self.data.get('check-interval', 1)
|
||||||
|
return int(interval)
|
||||||
BIN
lib/__pycache__/__init__.cpython-37.pyc
Normal file
BIN
lib/__pycache__/__init__.cpython-37.pyc
Normal file
Binary file not shown.
160
timer.py
160
timer.py
|
|
@ -1,164 +1,6 @@
|
||||||
import os, time
|
import os, time
|
||||||
import subprocess
|
|
||||||
|
|
||||||
from yaml import load, dump, Loader, Dumper
|
|
||||||
|
|
||||||
CONFIG_PATH = os.path.dirname(os.path.abspath(__file__))
|
|
||||||
|
|
||||||
def shell(cmd):
|
|
||||||
# cmd = cmd.split(' ')
|
|
||||||
p = subprocess.Popen(cmd,
|
|
||||||
stdout=subprocess.PIPE,
|
|
||||||
stderr=subprocess.STDOUT,
|
|
||||||
shell=True)
|
|
||||||
# p.communicate()
|
|
||||||
return p.stdout.read()
|
|
||||||
|
|
||||||
|
|
||||||
class Usage(object):
|
|
||||||
"""
|
|
||||||
Apps Usage Control.
|
|
||||||
Will store and inform about timers run time and limits.
|
|
||||||
"""
|
|
||||||
timer = None
|
|
||||||
file = None
|
|
||||||
|
|
||||||
def __init__(self, timer):
|
|
||||||
super(Usage, self).__init__()
|
|
||||||
self.timer = timer
|
|
||||||
self.file = '%s/usage/%s' % (CONFIG_PATH, self.timer.name)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def current(self):
|
|
||||||
'''Current usage of this timer in minutes'''
|
|
||||||
if not os.path.isfile(self.file):
|
|
||||||
return 0
|
|
||||||
with open(self.file, 'r') as f:
|
|
||||||
current = f.read()
|
|
||||||
return int(current) if current else 0
|
|
||||||
|
|
||||||
def increment(self, time):
|
|
||||||
'''Increment this timer usage'''
|
|
||||||
usage = self.current + int(time)
|
|
||||||
with open(self.file, 'w') as f:
|
|
||||||
f.write(str(usage))
|
|
||||||
|
|
||||||
def release(self):
|
|
||||||
'''Remove the timer usage to restart the counter'''
|
|
||||||
os.remove(self.file)
|
|
||||||
|
|
||||||
def isOffLimit(self):
|
|
||||||
'''Is the timer usage off its limits'''
|
|
||||||
time_limit = self.timer.timeLimit
|
|
||||||
if (time_limit < 0):
|
|
||||||
return False
|
|
||||||
if self.current > time_limit:
|
|
||||||
return True
|
|
||||||
|
|
||||||
def isOffInterval(self):
|
|
||||||
'''Is the timer usage expired'''
|
|
||||||
limit_interval = self.timer.limitInterval
|
|
||||||
if (limit_interval < 0):
|
|
||||||
return False
|
|
||||||
if not os.path.exists(self.file):
|
|
||||||
return False
|
|
||||||
usage_started = os.path.getctime(self.file)
|
|
||||||
usage_expiry = usage_started + (limit_interval * 60 * 60)
|
|
||||||
return time.time() >= usage_expiry
|
|
||||||
|
|
||||||
|
|
||||||
class Timer(object):
|
|
||||||
"""
|
|
||||||
Apps Timer Setup.
|
|
||||||
Will manage a timer and proxy its configuration
|
|
||||||
"""
|
|
||||||
item = None
|
|
||||||
name = None
|
|
||||||
usage = None
|
|
||||||
|
|
||||||
def __init__(self, name, item):
|
|
||||||
super(Timer, self).__init__()
|
|
||||||
self.name = name
|
|
||||||
self.item = item
|
|
||||||
self.usage = Usage(self)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def timeLimit(self):
|
|
||||||
time_limit = self.item.get('time-limit', -1)
|
|
||||||
return int(time_limit)
|
|
||||||
@property
|
|
||||||
def limitInterval(self):
|
|
||||||
limit_interval = self.item.get('limit-interval', -1)
|
|
||||||
return float(limit_interval)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def apps(self):
|
|
||||||
item_apps = self.item.get('apps', [])
|
|
||||||
if not isinstance(item_apps, list):
|
|
||||||
item_apps = [item_apps]
|
|
||||||
apps = [app.strip() for app in item_apps]
|
|
||||||
return apps
|
|
||||||
|
|
||||||
def isRunning(self):
|
|
||||||
running = False
|
|
||||||
for app in self.apps:
|
|
||||||
cmd = 'ps aux | grep -v grep | grep "%s"' % app
|
|
||||||
res = shell(cmd)
|
|
||||||
if len(res):
|
|
||||||
running = True
|
|
||||||
return running
|
|
||||||
|
|
||||||
def block(self):
|
|
||||||
for app in self.apps:
|
|
||||||
cmd = 'killall "%s"' % app
|
|
||||||
shell(cmd)
|
|
||||||
|
|
||||||
|
|
||||||
class Config(object):
|
|
||||||
"""
|
|
||||||
Config Parser.
|
|
||||||
Read and normalize the configuration objects
|
|
||||||
"""
|
|
||||||
data = None
|
|
||||||
timers = None
|
|
||||||
mtime = 0
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
super(Config, self).__init__()
|
|
||||||
self.file = '%s/config.yaml' % CONFIG_PATH
|
|
||||||
self.reload()
|
|
||||||
|
|
||||||
def hasChanges(self):
|
|
||||||
'''Check if the config file has changes'''
|
|
||||||
mtime = os.path.getmtime(self.file)
|
|
||||||
return self.mtime != mtime
|
|
||||||
|
|
||||||
def reload(self):
|
|
||||||
'''Reload config file data'''
|
|
||||||
self.mtime = os.path.getmtime(self.file)
|
|
||||||
self.data = self.read()
|
|
||||||
self.timers = self.getTimers()
|
|
||||||
|
|
||||||
def read(self):
|
|
||||||
'''Read the config file yaml object'''
|
|
||||||
with open(self.file, 'r') as f:
|
|
||||||
data = load(f, Loader=Loader)
|
|
||||||
return data
|
|
||||||
|
|
||||||
def getTimers(self):
|
|
||||||
'''Gets a generator of Timer objects'''
|
|
||||||
config_timers = self.data.get('timers', {})
|
|
||||||
timers = []
|
|
||||||
for name, item in config_timers.items():
|
|
||||||
timers.append(Timer(name, item))
|
|
||||||
return timers
|
|
||||||
|
|
||||||
@property
|
|
||||||
def checkInterval(self):
|
|
||||||
'''How often the usage check should happen'''
|
|
||||||
interval = self.data.get('check-interval', 1)
|
|
||||||
return int(interval)
|
|
||||||
|
|
||||||
|
from lib import Config
|
||||||
|
|
||||||
def check_timers(config):
|
def check_timers(config):
|
||||||
'''Will check every timer setup for it's usage and limits'''
|
'''Will check every timer setup for it's usage and limits'''
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue