class MarketService:
def __init__(self):
self.validations = MarketValidations()
self.database = Postgre()
self.parser = MarketParser()
def get_market_by_registry(self, registry: str):
self.validations.is_valid_registry(registry=registry)
LOGGER.debug(f'Getting maker by record {registry} service')
market = self.database.get_market_by_registry(registry=registry)
return self.parser.object_to_json(market=market)
def create_market(self, data: dict):
self.validations.new_market(data=data)
self.validations.is_valid_registry(registry=data.get('registro'))
LOGGER.debug('Creating new market service')
new_market = Market(**data)
market = self.database.insert_market(market=new_market)
return self.parser.object_to_json(market=market)
def delete_market(self, registry: str):
self.validations.is_valid_registry(registry=registry)
LOGGER.debug('Deleting market service')
self.database.delete_market_by_registry(registry=registry)
def get_markets(self, params: list):
LOGGER.debug('Getting markets service')
query = self.parser.params_to_query(params=params)
markets = self.database.get_markets(query=query)
return self.parser.markets_to_json(markets=markets)
def update_market(self, data: dict, registry: str):
self.validations.update_market(data=data)
self.validations.is_valid_registry(registry=registry)
LOGGER.debug('Updating new market service')
market = self.database.get_market_by_registry(registry=registry)
market = self.database.update_market(market=market, data=data)
return self.parser.object_to_json(market=market)
def test_params_to_query_return_bairro(self):
query = MarketParser.params_to_query(params=[('bairro', 1)])
self.assertEqual(query, '"bairro" = \'1\'')
def test_params_to_query_return_nome_feira(self):
query = MarketParser.params_to_query(params=[('nome_feira', 1)])
self.assertEqual(query, '"nome_feira" = \'1\'')
def test_params_to_query_return_regiao5(self):
query = MarketParser.params_to_query(params=[('regiao5', 1)])
self.assertEqual(query, '"regiao5" = \'1\'')
def test_params_to_query_return_distrito(self):
query = MarketParser.params_to_query(params=[('distrito', 1)])
self.assertEqual(query, '"distrito" = \'1\'')
def test_params_to_query_return_empty_string(self):
query = MarketParser.params_to_query(params=[('pudim', 1)])
self.assertEqual(query, '')
def test_params_to_query_failure_raise_exception(self):
with self.assertRaises(Exception):
MarketParser.params_to_query(params=[1, 2, 3, 4, 5])