Reformat code and add launch.json.

cleanunicorn · Jul 20, 2019 · 308fbe4 · 308fbe4
1 parent f6c910c
commit 308fbe4
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 11 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -19,9 +19,9 @@
             "console": "integratedTerminal",
             "args": [
                 "--account-pk=0xb0057716d5917badaf911b193b12b910811c1497b5bada8d7711f758981c3773",
-                "--contract=0x7c728214be9a0049e6a86f2137ec61030d0aa964",
+                "--contract=0xb87bF323AB5174806911CED29E9A2B9b6e839120",
                 "--rpc-http=http://localhost:8545",
-                "--source=mythril",
+                "--rpc-ws=ws://localhost:8546",
                 // "--rpc-ws=ws://localhost:8545",
             ],
         },

diff --git a/theo/interfaces/cli.py b/theo/interfaces/cli.py
@@ -42,12 +42,9 @@ def main():
 
     # Load exploits from file
     parser.add_argument(
-        "--load-file",
-        type=str,
-        help="Load exploit from file",
-        default="",
+        "--load-file", type=str, help="Load exploit from file", default=""
     )
-    
+
     args = parser.parse_args()
 
     # Get account from the private key
@@ -86,31 +83,33 @@ def start_repl(args):
         ]
 
     if len(exploits) == 0:
-        print(
-            "No exploits found. You're going to need to load some exploits."
-        )
+        print("No exploits found. You're going to need to load some exploits.")
     else:
         print("Found exploits(s)", exploits)
 
     # Load history
     history_path = "./.theo_history"
+
     def save_history(historyPath=history_path):
         import readline
 
         readline.write_history_file(history_path)
 
     import os
     import readline
+
     if os.path.isfile(history_path):
         readline.read_history_file(history_path)
     # Trigger history save on exit
     import atexit
+
     atexit.register(save_history)
     # Load variables
     vars = globals()
     vars.update(locals())
     # Start REPL
     import rlcompleter
+
     readline.set_completer(rlcompleter.Completer(vars).complete)
     readline.parse_and_bind("tab: complete")
     del os, atexit, readline, rlcompleter, save_history

diff --git a/theo/scanner/__init__.py b/theo/scanner/__init__.py
@@ -38,7 +38,7 @@ def exploits_from_mythril(
             eth=EthJsonRpc(host=host, port=port, tls=rpc_tls),
             solc_version=None,
             solc_args=None,
-            enable_online_lookup=enable_online_lookup
+            enable_online_lookup=enable_online_lookup,
         )
         disassembler.load_from_address(contract)
         # Analyzer