✨ Make call trace configurable

Ackee-Blockchain · Oct 5, 2023 · c233d24 · c233d24
1 parent 50708a3
commit c233d24
Show file tree

Hide file tree

Showing 3 changed files with 154 additions and 58 deletions.
diff --git a/woke/config/data_model.py b/woke/config/data_model.py
@@ -211,6 +211,21 @@ class DeploymentConfig(WokeConfigModel):
     silent: bool = False
 
 
+class GeneralConfig(WokeConfigModel):
+    call_trace_options: FrozenSet[str] = frozenset(
+        [
+            "contract_name",
+            "function_name",
+            "arguments",
+            "status",
+            "call_type",
+            "value",
+            "return_value",
+            "error",
+        ]
+    )
+
+
 class TopLevelConfig(WokeConfigModel):
     subconfigs: List[Path] = []
     api_keys: Dict[str, str] = {}
@@ -220,6 +235,7 @@ class TopLevelConfig(WokeConfigModel):
     lsp: LspConfig = Field(default_factory=LspConfig)
     testing: TestingConfig = Field(default_factory=TestingConfig)
     deployment: DeploymentConfig = Field(default_factory=DeploymentConfig)
+    general: GeneralConfig = Field(default_factory=GeneralConfig)
 
     @validator("subconfigs", pre=True, each_item=True)
     def set_subconfig(cls, v):

diff --git a/woke/config/woke_config.py b/woke/config/woke_config.py
@@ -16,6 +16,7 @@
     CompilerConfig,
     DeploymentConfig,
     DetectorsConfig,
+    GeneralConfig,
     GeneratorConfig,
     LspConfig,
     TestingConfig,
@@ -371,3 +372,10 @@ def deployment(self) -> DeploymentConfig:
         Return deployment-specific config options.
         """
         return self.__config.deployment
+
+    @property
+    def general(self) -> GeneralConfig:
+        """
+        Return general config options.
+        """
+        return self.__config.general