3 from datetime import datetime
4 from planetstack.config import Config
5 from util.logger import Logger, logging
6 from observer.steps import *
8 logger = Logger(level=logging.INFO)
10 class FailedDependency(Exception):
14 """ A PlanetStack Sync step.
17 psmodel Model name the step synchronizes
18 dependencies list of names of models that must be synchronized first if the current model depends on them
23 sync_config_dir = Config().sync_config_dir
25 sync_config_dir = '/etc/planetstack/sync'
26 prop_config_path = '/'.join(sync_config_dir,self.name,prop)
27 return open(prop_config_path).read().rstrip()
29 def __init__(self, **args):
30 """Initialize a sync step
32 name -- Name of the step
33 provides -- PlanetStack models sync'd by this step
36 self.driver = args.get('driver')
37 self.error_map = args.get('error_map')
40 self.soft_deadline = int(self.get_prop('soft_deadline_seconds'))
42 self.soft_deadline = 5 # 5 seconds
46 def fetch_pending(self):
48 #return Sliver.objects.filter(ip=None)
50 def check_dependencies(self, obj, failed):
51 for dep in self.dependencies:
52 peer_object = getattr(obj, dep.lower())
53 if (peer_object.pk==failed.pk):
54 raise FailedDependency
56 def call(self, failed=[]):
57 pending = self.fetch_pending()
61 self.check_dependencies(o,f) # Raises exception if failed
63 o.enacted = datetime.now() # Is this the same timezone? XXX
64 o.backend_status = "OK")
65 o.save(update_fields=['enacted'])
68 o.backend_status = self.error_map.map(str(e))
70 o.backend_status = str(e)
72 o.save(update_fields=['backend_status'])
74 logger.log_exc("sync step failed!")
79 def __call__(self, **args):
80 return self.call(**args)