class ConditionService(INamed):
'''
This class provide a facade for managing Condition objects
'''
def __init__(self, name='ConditionSrv'):
'''
@param name: The unique name of the service
'''
self._conditions = Case().tag(name) \
.tag(Condition.__name__) \
.tag(CategoricalCondition.__name__) \
.tag(AttributeCondition.__name__) \
.tag(RangeCondition.__name__) \
.tag(QueryCondition.__name__)
INamed.__init__(self, name)
def register_in(self, dispatcher):
dispatcher.add_method(self.new_condition)
dispatcher.add_method(self.expose_condition)
dispatcher.add_method(self.del_condition)
dispatcher.add_method(self.update_conditions)
dispatcher.add_method(self.update_conditions2)
# Condition Properties
dispatcher.add_method(partial(self._proxy_property, 'name'), 'name')
dispatcher.add_method(partial(self._proxy_property, 'data'), 'data')
dispatcher.add_method(partial(self._proxy_property, 'grammar'), 'grammar')
dispatcher.add_method(partial(self._proxy_property, 'enabled'), 'enabled')
# Condition Methods
dispatcher.add_method(partial(self._proxy, 'include_all'), 'include_all')
dispatcher.add_method(partial(self._proxy, 'exclude_all'), 'exclude_all')
dispatcher.add_method(partial(self._proxy, 'enable'), 'enable')
dispatcher.add_method(partial(self._proxy, 'disable'), 'disable')
# CategoricalCondition Properties
dispatcher.add_method(partial(self._proxy_property, 'attr'), 'attr')
dispatcher.add_method(partial(self._proxy, 'get_condition'), 'get_condition')
# CategoricalCondition Methods
dispatcher.add_method(partial(self._proxy, 'included_categories'), 'included_categories')
dispatcher.add_method(partial(self._proxy, 'excluded_categories'), 'excluded_categories')
dispatcher.add_method(partial(self._proxy, 'included_items'), 'included_items')
dispatcher.add_method(partial(self._proxy, 'excluded_items'), 'excluded_items')
dispatcher.add_method(partial(self._proxy, 'add_category'), 'add_category')
dispatcher.add_method(partial(self._proxy, 'remove_category'), 'remove_category')
dispatcher.add_method(partial(self._proxy, 'toggle_category'), 'toggle_category')
# AttributeCondition Methods
dispatcher.add_method(partial(self._proxy, 'included_attributes'), 'included_attributes')
dispatcher.add_method(partial(self._proxy, 'excluded_attributes'), 'excluded_attributes')
dispatcher.add_method(partial(self._proxy, 'add_attribute'), 'add_attribute')
dispatcher.add_method(partial(self._proxy, 'remove_attribute'), 'remove_attribute')
dispatcher.add_method(partial(self._proxy, 'toggle_attribute'), 'toggle_attribute')
# RangeCondition Properties
dispatcher.add_method(partial(self._proxy_property, 'range'), 'range')
dispatcher.add_method(partial(self._proxy_property, 'domain'), 'domain')
# RangeCondition Methods
dispatcher.add_method(partial(self._proxy, 'set_range'), 'set_range')
# QueryCondition Properties
dispatcher.add_method(partial(self._proxy_property, 'query'), 'query')
# QueryCondition Methods
dispatcher.add_method(partial(self._proxy, 'set_query'), 'set_query')
def _proxy(self, method, condition_oid, *args, **kwargs):
condition = self._conditions[condition_oid]
result = condition.__getattribute__(method)(*args, **kwargs)
if isinstance(result, Condition):
self._conditions[result.oid] = result
return result
def _proxy_property(self, method, condition_oid):
condition = self._conditions[condition_oid]
result = condition.__getattribute__(method)
if isinstance(result, Condition):
self._conditions[result.oid] = result
return result
def new_condition(self, kind, data, *args, **kwargs):
dataset = Showcase.instance().get(data)
if kind == 'categorical':
new_condition = CategoricalCondition(dataset, *args, **kwargs)
if kind == 'attribute':
new_condition = AttributeCondition(dataset, *args, **kwargs)
if kind == 'range':
new_condition = RangeCondition(dataset, *args, **kwargs)
if kind == 'query':
new_condition = QueryCondition(dataset, *args, **kwargs)
self._conditions[new_condition.oid] = new_condition
return new_condition
def expose_condition(self, condition):
self._conditions[condition.oid] = condition
return condition
def del_condition(self, oid):
self._conditions.pop(oid)
def update_conditions(self): # Update all conditions (variable _domain)
from pprint import pprint
for ind in self._conditions.keys():
cond = self._conditions[ind]
cond._sieve._domain = set(cond._sieve._data.distinct(cond._sieve._data_index)) # UPDATE
#domm = cond._sieve._data.aggregate([{'$match': {cond._attr: {'$type': 1}}}, # Only numbers
# {'$group':
# {'_id': {},
# 'min': {'$min': "$"+cond._attr},
# 'max': {'$max': "$"+cond._attr}}}]).get_data()
#ran = cond._sieve._data.distinct(cond._attr)
#min_ran = min(ran)
#max_ran = max(ran)
#cond._range = {'min': min_ran, 'max': max_ran}
#cond._sieve._domain = {'min': min_ran, 'max': max_ran}
#pprint(cond._sieve._domain)
return self._conditions
def update_conditions2(self, conditions): # Update all conditions (variable _domain)
from pprint import pprint
for ind in conditions.keys():
cond = conditions[ind]['grammar']
condition = self._conditions[cond['name']]
if cond['type'] == "categorical":
condition._sieve._domain = set(condition._sieve._data.distinct(condition._sieve._data_index)) # UPDATE
cond['included_categories'] = list(set(condition._sieve._data.distinct(condition._sieve._data_index)))
cond['excluded_categories'] = []
elif cond['type'] == "range":
ran = condition._sieve._data.distinct(condition._attr)
min_ran = min(ran)
max_ran = max(ran)
if max_ran == min_ran:
min_ran -= 1
max_ran += 1
relative_max = condition._to_relative(condition._range['max'])
relative_min = condition._to_relative(condition._range['min'])
condition._domain = {'min': min_ran, 'max': max_ran}
condition._range = {'min': min_ran, 'max': max_ran, 'relative_min': relative_min, 'relative_max': relative_max}
cond['domain'] = condition._domain
cond['range'] = condition._range
return conditions
