def run(self): mon_lunch_central = 'Arroz/feijão/arroz integral\n' + \ 'Hambúrguer barbecue\n' + \ 'Creme de milho\n' + \ 'Salada de escarola\n' + \ 'Opcional: PVT com legumes\n' + \ 'Bananinha/refresco\n' + \ 'Valor calórico de 1 refeição 1055 kcal' sat_lunch_phys = 'FECHADO' wed_dinner_chem = 'Arroz/Feijão/Arroz integral\n' + \ 'Peixe ao forno\n' + \ 'Chuchu na salsa\n' + \ 'Salada de alface\n' + \ 'Opcional: Quibe de PVT\n' + \ 'Mamão\n' + \ 'Refresco\n' + \ '876 kcal' fri_lunch_pco = 'Arroz/feijão/arroz integral\n' + \ 'Tiras de carne à cigana\n' + \ 'Mandioca corada\n' + \ 'Salada de almeirão\n' + \ 'Opcional: Bolinho de PVT ao molho shoyo\n' + \ 'Maçã/refresco\n' + \ 'Valor calórico de 1 refeição 829 kcal' fri_dinner_pco = 'Fechado' rest_central = Restaurante('central') rest_fisica = Restaurante('fisica') rest_quimica = Restaurante('quimica') rest_pco = Restaurante('pco') print('Comparando - Bandejão central - Segunda-feira - Almoço:') self.compare(rest_central.print_menu('segunda', True, False), mon_lunch_central) print(mon_lunch_central) print('Comparando - Bandejão da Física - Sábado - Almoço:') self.compare(rest_fisica.print_menu('sabado', True, False), sat_lunch_phys) print(sat_lunch_phys) print('Comparando - Bandejão da Química - Quarta-feira - Janta:') self.compare(rest_quimica.print_menu('quarta', False, True), wed_dinner_chem) print(wed_dinner_chem) print('Comparando - Bandejão PCO - Sexta-feira - Almoço:') self.compare(rest_pco.print_menu('sexta', True, False), fri_lunch_pco) print('Comparando - Bandejão PCO - Sexta-feira - Janta:') self.compare(rest_pco.print_menu('sexta', False, True), fri_dinner_pco) print('Testando para um restaurante desconhecido:') try: fake = Restaurante('fake', False, False) print('O teste falhou! =(') except: print('O teste passou com sucesso!')
from restaurante import Restaurante def day_of_week(day_number): day_list = ['segunda', 'terca', 'quarta', 'quinta', 'sexta', 'sabado', 'domingo'] return day_list[day_number] if __name__ == '__main__': parser = argparse.ArgumentParser() day_number = datetime.datetime.today().weekday() parser = argparse.ArgumentParser(fromfile_prefix_chars='@') parser.add_argument('-b', '--bandex', help='Local') parser.add_argument('-d', '--dia', help='Dia da semana', default=day_of_week(day_number)) parser.add_argument('-a', '--almoco', help='Almoço', action='store_true') parser.add_argument('-j', '--janta', help='Janta', action='store_true') args = parser.parse_args() if not args.bandex: args = parser.parse_args(['@.bandexrc']) restaurante = Restaurante(args.bandex) if not args.almoco and not args.janta: restaurante.print_menu(args.dia, True, True) else: restaurante.print_menu(args.dia, args.almoco, args.janta)