diff options
Diffstat (limited to 'lib/utils.py')
-rw-r--r-- | lib/utils.py | 29 |
1 files changed, 27 insertions, 2 deletions
diff --git a/lib/utils.py b/lib/utils.py index 9024c7e..447637a 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -1,5 +1,16 @@ import os import json +from myconfig import MyConfig + + +config = MyConfig() + +#Exceptions +class EpochError(Exception): + def __init__(self, value): + self.value = value + def __str__(self): + return repr(self.value) #Utils @@ -29,10 +40,24 @@ def store_status(outfile, obj): return def check_epoch(obj, passname): - pass + epochname = passname + 'Epoch' + if not epochname in obj: + return False + object_epoch = int(obj[epochname]) + config_epoch = int(config.getEpochs()[epochname]) + if object_epoch > config_epoch: + raise EpochError('Un-excepted larger epoch en-countered.\n' + \ + 'Please increace the epoch in myconfig.py\n' ) + + if object_epoch < config_epoch: + return False + if object_epoch == config_epoch: + return True + return None def sign_epoch(obj, passname): - pass + epochname = passname + 'Epoch' + obj[epochname] = config.getEpochs()[epochname] #test case if __name__ == '__main__': |