def __init__(self, module_object, params=None, dataset=None): BaseMsg.__init__(self, 'task') self.__module = module_object self.__params = params if dataset != None and not issubclass(type(dataset), BaseDataSet): exc_msg = 'expected object of type DataSet or None for dataset param.' # TODO(ali01): logging raise ValueError, exc_msg self.__dataset = dataset
def __init__(self, refs=None): BaseMsg.__init__(self, 'refs') if refs is None: refs = [] if not isinstance(refs, list): raise TypeError, 'refs parameter must be of type list' for ref in refs: if not isinstance(ref, list) or len(ref) != 2: raise TypeError, 'each element of refs must be a list of size 2' self.__refs = refs
class WidgetResponseMsg(BaseMsg): def __init__(self, msg, q, **kwargs): self.msg = msg self.q = q self.id = self.msg.id self.body = BaseMsg().get_body(self.msg) self.widget_id = self.body.get('WidgetId', None) self.widget_maker_url = kwargs.get('widget_maker_url', 'http://localhost:5000/add') def __repr__(self): return '<WidgetResponseMsg widget_id:{0.widget_id}>'.format(self) def process(self): data = {'widget_id': self.widget_id} headers = {'Content-type': 'application/json', 'Accept': 'text/plain'} widget = requests.post(self.widget_maker_url, data=json.dumps(data), headers=headers) try: widget.raise_for_status() except requests.exceptions.HTTPError: raise else: return widget
def __init__(self, msg, q, **kwargs): self.msg = msg self.q = q self.id = self.msg.id self.body = BaseMsg().get_body(self.msg) self.widget_id = self.body.get('WidgetId', None) self.widget_maker_url = kwargs.get('widget_maker_url', 'http://localhost:5000/add')
def __init__(self, status, missing_refs=None): BaseMsg.__init__(self, 'state') if missing_refs is None: missing_refs = [] if status != 'running' and status != 'blocked': raise ValueError, 'unknown value for status parameter' if status == 'blocked': if len(missing_refs) == 0: raise ValueError, '''len(missing_refs) should be non-zero if status is 'blocked' ''' elif status == 'running' and len(missing_refs) != 0: raise ValueError, '''len(missing_refs) should be zero if status is not 'blocked' ''' self.__status = status self.__missing_refs = missing_refs
def __init__(self, result_object): BaseMsg.__init__(self, 'result') self.__result = result_object
def __init__(self, error_str): BaseMsg.__init__(self, "error") self.__error = error_str