def _price_info(self, session, curs, data, rule_field_name): # Getting required data for price calculation to_id = data['tariffication_object_id'] to_f = TarifficationObjectFilter(session, {'id': to_id}, {}, None) to = to_f.filter_one_obj(curs) # TODO: handle user tariffs all_ts_f = TariffFilter(session, {}, {}, None) all_ts = all_ts_f.filter_objs(curs) all_ts_idx = build_index(all_ts) t_id = data['tariff_id'] if t_id not in all_ts_idx: raise TariffNotFound(tariff_id=t_id) if 'user_id' in data: ut_f = UserTariffFilter(session, { 'user_id': data['user_id'], 'tariff_id': t_id }, {}, None) ut_f.filter_one_obj(curs) all_curs_idx = self._all_curs_idx(curs) t = all_ts_idx[t_id] ts_chain_data = self._tariffs_chain_data(all_ts_idx, t, all_curs_idx) ts_chain_ids = [t_data['id'] for t_data in ts_chain_data] # Fetching active rules r_f = RuleFilter( session, { 'tariff_ids': ts_chain_ids, 'tariffication_object_id': to_id, 'status': Rule.STATUS_ACTIVE }, {}, None) rs = r_f.filter_objs(curs) rs_to_t_idx = build_complex_index( rs, ('tariffication_object_id', 'tariff_id')) # Generation price info calculation_ctx = data.get('calculation_context', {}) r, t = self._find_rule_tariff(to, ts_chain_ids, all_ts_idx, rs_to_t_idx, rule_field_name, calculation_ctx) if r is None: raise PriceNotFound price_info = self._calculate_rule_info(r, rule_field_name, t, calculation_ctx) price_info['tariffication_object_id'] = to.id price_info['tariffication_object_name'] = to.name if 'calculation_context' in data: price_info['calculation_context'] = calculation_ctx return price_info
def _price_info(self, session, curs, data, rule_field_name): # Getting required data for price calculation to_id = data['tariffication_object_id'] to_f = TarifficationObjectFilter(session, {'id': to_id}, {}, None) to = to_f.filter_one_obj(curs) # TODO: handle user tariffs all_ts_f = TariffFilter(session, {}, {}, None) all_ts = all_ts_f.filter_objs(curs) all_ts_idx = build_index(all_ts) t_id = data['tariff_id'] if t_id not in all_ts_idx: raise TariffNotFound(tariff_id=t_id) if 'user_id' in data: ut_f = UserTariffFilter(session, {'user_id': data['user_id'], 'tariff_id': t_id}, {}, None) ut_f.filter_one_obj(curs) all_curs_idx = self._all_curs_idx(curs) t = all_ts_idx[t_id] ts_chain_data = self._tariffs_chain_data(all_ts_idx, t, all_curs_idx) ts_chain_ids = [t_data['id'] for t_data in ts_chain_data] # Fetching active rules r_f = RuleFilter(session, {'tariff_ids': ts_chain_ids, 'tariffication_object_id': to_id, 'status': Rule.STATUS_ACTIVE}, {}, None) rs = r_f.filter_objs(curs) rs_to_t_idx = build_complex_index(rs, ('tariffication_object_id', 'tariff_id')) # Generation price info calculation_ctx = data.get('calculation_context', {}) r, t = self._find_rule_tariff(to, ts_chain_ids, all_ts_idx, rs_to_t_idx, rule_field_name, calculation_ctx) if r is None: raise PriceNotFound price_info = self._calculate_rule_info(r, rule_field_name, t, calculation_ctx) price_info['tariffication_object_id'] = to.id price_info['tariffication_object_name'] = to.name if 'calculation_context' in data: price_info['calculation_context'] = calculation_ctx return price_info
def delete_tariffication_object(self, data, session, curs=None): f = TarifficationObjectFilter(session, {'id': data.get('id')}, {}, None) mapping.delete(curs, f.filter_one_obj(curs)) return response_ok()