Add balance loading to accounts_exist_or_load

mythril/laser/ethereum/state/account.py

@@ -127,18 +127,6 @@ class Account:
         self._balances = balances
         self.balance = lambda: self._balances[self.address]
 
-        """
-        if not self.address.symbolic and dynamic_loader is not None:
-            try:
-                _balance = dynamic_loader.read_balance(
-                    "{0:#0{1}x}".format(self.address.value, 42)
-                )
-                self.set_balance(_balance)
-            except:
-                # Initial balance will be a symbolic variable
-                pass
-        """
-
     def __str__(self) -> str:
         return str(self.as_dict)
 

mythril/laser/ethereum/state/world_state.py

@@ -90,6 +90,22 @@ class WorldState:
             return self.accounts[addr_bitvec.value]
         if dynamic_loader is None:
             raise ValueError("dynamic_loader is None")
+
+        if isinstance(addr, int):
+            try:
+                _balance = dynamic_loader.read_balance("{0:#0{1}x}".format(addr, 42))
+                self.set_balance(_balance)
+            except:
+                # Initial balance will be a symbolic variable
+                pass
+        elif isinstance(addr, str):
+            try:
+                _balance = dynamic_loader.read_balance(addr)
+                self.set_balance(_balance)
+            except:
+                # Initial balance will be a symbolic variable
+                pass
+
         return self.create_account(
             balance=0,
             address=addr_bitvec.value,