Exemplo n.º 1
0
 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
Exemplo n.º 2
0
 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']
         }})