Custom error handling

pyproject.toml

@@ -29,6 +29,7 @@ fastapi = "^0.110.0"
 starlette = "^0.36.3"
 uvicorn = "^0.29.0"
 keylimiter = "^0.1.5"
+loguru = "^0.7.2"
 
 
 [tool.poetry.group.dev]

src/communex/cli/module.py

@@ -7,7 +7,7 @@
 
 import communex.balance as c_balance
 from communex.compat.key import classic_load_key
-from communex.errors import ChainTransactionError
+from communex.errors import ChainTransactionError, InvalidIPError, InvalidClassError, InvalidModuleError
 from communex.misc import get_map_modules
 from communex.module.server import ModuleServer
 from communex.util import is_ip_valid
@@ -45,7 +45,7 @@ def register(
         resolved_key = classic_load_key(key)
 
         if not is_ip_valid(ip):
-            raise ValueError("Invalid ip address")
+            raise InvalidIPError("Invalid ip address")
 
         address = f"{ip}:{port}"
         subnet = client.get_name(netuid)
@@ -70,7 +70,7 @@ def update(key: str, name: str, ip: str, port: int, delegation_fee: int = 20, ne
     client = make_client()
 
     if not is_ip_valid(ip):
-        raise ValueError("Invalid ip address")
+        raise InvalidIPError("Invalid ip address")
 
     address = f"{ip}:{port}"
 
@@ -107,24 +107,24 @@ def serve(
             module_path = ".".join(module_parts)
             if not module_path:
                 # This could do some kind of relative import somehow?
-                raise ValueError(f"Invalid class path: `{class_path}`, module name is missing")
+                raise InvalidModuleError(f"Invalid class path: `{class_path}`, module name is missing")
             if not class_name:
-                raise ValueError(f"Invalid class path: `{class_path}`, class name is missing")
+                raise InvalidClassError(f"Invalid class path: `{class_path}`, class name is missing")
         case _:
             # This is impossible
-            raise Exception(f"Invalid class path: `{class_path}`")
+            raise TypeError(f"Invalid class path: `{class_path}`")
 
     try:
         module = importlib.import_module(module_path)
-    except ModuleNotFoundError:
+    except ModuleNotFoundError as e:
         context.error(f"Module `{module_path}` not found")
-        raise typer.Exit(code=1)
+        raise typer.Exit(code=1) from e
 
     try:
         class_obj = getattr(module, class_name)
-    except AttributeError:
+    except AttributeError as e:
         context.error(f"Class `{class_name}` not found in module `{module}`")
-        raise typer.Exit(code=1)
+        raise typer.Exit(code=1) from e
 
     keypair = classic_load_key(key)
     server = ModuleServer(class_obj(), keypair, whitelist=whitelist,
@@ -150,7 +150,7 @@ def info(name: str, balance: bool = False, netuid: int = 0):
         module = next((item for item in modules_to_list if item["name"] == name), None)
 
     if module is None:
-        raise ValueError("Module not found")
+        raise InvalidModuleError("Module not found")
 
     general_module = cast(dict[str, Any], module)
     print_table_from_plain_dict(general_module, ["Params", "Values"], console)

src/communex/cli/network.py

@@ -91,19 +91,21 @@ def propose_globally(
 
     resolved_key = classic_load_key(key)
 
-    proposal: NetworkParams = {"max_allowed_subnets": max_allowed_subnets,
-                               "max_allowed_modules": max_allowed_modules,
-                               "max_registrations_per_block": max_registrations_per_block,
-                               "unit_emission": unit_emission,
-                               "tx_rate_limit": tx_rate_limit,
-                               "vote_threshold": vote_threshold,
-                               "vote_mode": vote_mode,
-                               "max_proposals": max_proposals,
-                               "max_name_length": max_name_length,
-                               "burn_rate": burn_rate,
-                               "min_burn": min_burn,
-                               "min_stake": min_stake,
-                               "min_weight_stake": min_weight_stake}
+    proposal = NetworkParams(
+        max_allowed_subnets=max_allowed_subnets,
+        max_allowed_modules=max_allowed_modules,
+        max_registrations_per_block=max_registrations_per_block,
+        unit_emission=unit_emission,
+        tx_rate_limit=tx_rate_limit,
+        vote_threshold=vote_threshold,
+        vote_mode=vote_mode,
+        max_proposals=max_proposals,
+        max_name_length=max_name_length,
+        burn_rate=burn_rate,
+        min_burn=min_burn,
+        min_stake=min_stake,
+        min_weight_stake=min_weight_stake
+    )
 
     with console.status("Adding a proposal..."):
         client.add_global_proposal(resolved_key, proposal)