dibbler/menus/__init__.py

@@ -1,8 +1,37 @@
 # -*- coding: utf-8 -*-
 
+from .addstock import AddStockMenu
+from .buymenu import BuyMenu
+from .editing import (
+    AddUserMenu,
+    EditUserMenu,
+    AddProductMenu,
+    EditProductMenu,
+    AdjustStockMenu,
+    CleanupStockMenu,
+)
+from .faq import FAQMenu
+from .helpermenus import Menu
+from .mainmenu import MainMenu
+from .miscmenus import (
+    ProductSearchMenu,
+    TransferMenu,
+    AdjustCreditMenu,
+    UserListMenu,
+    ShowUserMenu,
+    ProductListMenu,
+)
+from .printermenu import PrintLabelMenu
+from .stats import (
+  ProductPopularityMenu,
+  ProductRevenueMenu,
+  BalanceMenu,
+  LoggedStatisticsMenu
+)
+
 exit_commands = ['exit', 'abort', 'quit', 'bye', 'eat flaming death', 'q']
 help_commands = ['help', '?']
 context_commands = ['what', '??']
 local_help_commands = ['help!', '???']
 faq_commands = ['faq']
-restart_commands = ['restart']
+restart_commands = ['restart']

dibbler/text_based.py

@@ -5,27 +5,9 @@ import random
 import sys
 import traceback
 
-from .helpers import *
-from .menus.addstock import AddStockMenu
-from .menus.buymenu import BuyMenu
-from .menus.editing import *
-from .menus.faq import FAQMenu
-from .menus.helpermenus import Menu
-from .menus.mainmenu import MainMenu
-
-from .menus.miscmenus import (
-    ProductSearchMenu,
-    TransferMenu,
-    AdjustCreditMenu,
-    UserListMenu,
-    ShowUserMenu,
-    ProductListMenu,
-)
-
-from .menus.printermenu import PrintLabelMenu
-from .menus.stats import *
-
 from .conf import config
+from .helpers import *
+from .menus import *
 
 random.seed()