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
