def get_specials(self, day): cursor = self.mongo.db.Bars.find({}) specials = [] for bar in cursor: for spec in bar['specials'][day]: to_add = Special(spec['special_id'], spec['special_name'], spec['description'], spec['bar_id'], object=spec['object']) specials.append(to_add.serialize()) return specials
def create_special(self, bar_id, special, day): bar = self.mongo.db.Bars.find_one({'bar_id': bar_id}) specials = bar['specials'][day] total = 0 for l in bar['specials']: for s in l: total = max(total, s['special_id']) special['special_id'] = total + 1 to_add = Special(special['special_id'], special['special_name'], special['description'], special['bar_id'], object=special['object']) specials.append(to_add.serialize()) update = self.mongo.db.Bars.update_one( {'bar_id': bar_id}, {'$set': { 'specials': bar['specials'] }})