ajustes no display do menu

ramalho · ramalho · commit e959b2805484 · 2015-08-14T18:31:29.000-03:00
diff --git a/composite/menu.py b/composite/menu.py
@@ -28,9 +28,11 @@ def description(self):
     def is_vegetarian(self):
         return all(sub_menu.is_vegetarian for sub_menu in self.menu_components)
 
-
     def display(self):
-        print('\n' + self.name, ', ', self.description, sep='')
+        print('\n' + self.name, end='')
+        if self.is_vegetarian:
+            print(' (v)', end='')
+        print(', ', self.description, sep='')
         print('-' * 60);
         for menu_component in self.menu_components:
             menu_component.display()
@@ -63,6 +65,6 @@ def price(self):
     def display(self):
         print('  ' + self.name, end='')
         if self.is_vegetarian:
-            print('(v)', end='')
+            print(' (v)', end='')
         print(',', self.price)
         print('     --', self.description);
