add default values directly to modules

Author: yungwine

modules/__init__.py

@@ -0,0 +1,20 @@
+import typing
+
+from .module import MtcModule
+from .pool import PoolModule
+from .nominator_pool import NominatorPoolModule
+from .single_pool import SingleNominatorModule
+from .validator import ValidatorModule
+from .controller import ControllerModule
+
+
+MODES = {
+    'validator': ValidatorModule,
+    'nominator-pool': NominatorPoolModule,
+    'single-nominator': SingleNominatorModule,
+    'liquid-staking': ControllerModule,
+}
+
+
+def get_mode(mode_name: str) -> typing.Optional[MtcModule]:
+    return MODES.get(mode_name)

modules/controller.py

@@ -11,6 +11,7 @@
 class ControllerModule(MtcModule):
 
     description = 'Liquid staking controllers.'
+    default_value = False
 
     def do_create_controllers(self):
         new_controllers = self.ton.GetControllers()

modules/module.py

@@ -1,12 +1,13 @@
 from abc import ABC, abstractmethod
-from mytoncore.mytoncore import MyTonCore
 
 
 class MtcModule(ABC):
 
-    description = ''
+    description = ''  # module text description
+    default_value = True  # is module enabled by default
 
     def __init__(self, ton, local, *args, **kwargs):
+        from mytoncore.mytoncore import MyTonCore
         self.ton: MyTonCore = ton
         self.local = local
 

modules/nominator_pool.py

@@ -8,6 +8,7 @@
 class NominatorPoolModule(PoolModule):
 
     description = 'Standard nominator pools.'
+    default_value = False
 
     def do_create_pool(self, pool_name, validator_reward_share_percent, max_nominators_count, min_validator_stake,
                        min_nominator_stake):

modules/pool.py

@@ -7,6 +7,7 @@
 class PoolModule(MtcModule):
 
     description = 'Basic pools functions.'
+    default_value = False
 
     def print_pools_list(self, args):
         table = list()

modules/single_pool.py

@@ -9,6 +9,7 @@
 class SingleNominatorModule(PoolModule):
 
     description = 'Orbs\'s single nominator pools.'
+    default_value = False
 
     def do_create_single_pool(self, pool_name, owner_address):
         self.ton.local.add_log("start create_single_pool function", "debug")

modules/validator.py

@@ -9,6 +9,8 @@ class ValidatorModule(MtcModule):
     description = ('Validator functions. Activates participating in elections and staking. '
                    'If pools and l/s modes are disabled stakes from validator wallet.')
 
+    default_value = True
+
     def vote_offer(self, args):
         if len(args) == 0:
             color_print("{red}Bad args. Usage:{endc} vo <offer-hash>")

mytoncore/modes.py

@@ -11,7 +11,7 @@
 import requests
 from fastcrc import crc16
 
-from mytoncore.modes import MODES
+from modules import MODES
 from mytoncore.utils import xhex2hex, ng2g
 from mytoncore.liteclient import LiteClient
 from mytoncore.validator_console import ValidatorConsole
@@ -3172,9 +3172,9 @@ def get_modes(self):
 		if 'modes' not in self.local.db:
 			self.local.db['modes'] = current_modes
 			self.migrate_to_modes()
-		for mode in MODES:
-			if mode not in current_modes:
-				current_modes[mode] = MODES[mode]  # assign default mode value
+		for name, mode in MODES.items():
+			if name not in current_modes:
+				current_modes[name] = mode.default_value  # assign default mode value
 		return current_modes
 
 	def enable_mode(self, name):

mytoncore/mytoncore.py

@@ -424,13 +424,18 @@ def sl(ton, args):
 	Slashing(ton.local, ton)
 #end define
 
+
 def mode_status(ton, args):
+	from modules import get_mode
 	modes = ton.get_modes()
-	text = "########## {bold}Modes{endc} ########## \n"
-	for mode in modes:
-		status = '{green}enabled{endc}' if modes[mode] else '{red}disabled{endc}'
-		text += f"{mode}: {status}\n"
-	color_print(text)
+	table = [["Name", "Status", "Description"]]
+	for mode_name in modes:
+		mode = get_mode(mode_name)
+		status = color_text('{green}enabled{endc}' if modes[mode_name] else '{red}disabled{endc}')
+		table.append([mode_name, status, mode.description])
+	print_table(table)
+#end define
+
 
 def PrintStatus(local, ton, args):
 	opt = None