fix: replace mess with setdefault

eth-brownie · Jun 1, 2020 · ea7b90e · ea7b90e
1 parent 85c7767
commit ea7b90e
Showing 1 changed file with 7 additions and 13 deletions.
diff --git a/brownie/project/main.py b/brownie/project/main.py
@@ -393,19 +393,13 @@ def _add_to_deployment_map(self, contract: ProjectContract) -> None:
 
         chainid = CONFIG.active_network["chainid"] if CONFIG.network_type == "live" else "dev"
         deployment_map = self._load_deployment_map()
-
-        if chainid in deployment_map:
-            if contract._name in deployment_map[chainid] and isinstance(
-                deployment_map[chainid][contract._name], list
-            ):
-                if contract.address in deployment_map[chainid][contract._name]:
-                    deployment_map[chainid][contract._name].remove(contract.address)
-                deployment_map[chainid][contract._name].insert(0, contract.address)
-            else:
-                deployment_map[chainid][contract._name] = [contract.address]
-        else:
-            deployment_map[chainid] = {contract._name: [contract.address]}
-
+        try:
+            deployment_map[chainid][contract._name].remove(contract.address)
+        except (ValueError, KeyError):
+            pass
+        deployment_map.setdefault(chainid, {}).setdefault(contract._name, []).insert(
+            0, contract.address
+        )
         self._save_deployment_map(deployment_map)
 
     def _update_and_register(self, dict_: Any) -> None: