Revert graph generation code to process a single contract

mythril/interfaces/cli.py

@@ -2,7 +2,7 @@
 # -*- coding: UTF-8 -*-
 """mythril.py: Bug hunting on the Ethereum blockchain
 
-   http://www.github.com/b-mueller/mythril
+   http://www.github.com/ConsenSys/mythril
 """
 
 import logging
@@ -30,7 +30,7 @@ def main():
     parser.add_argument("solidity_file", nargs='*')
 
     commands = parser.add_argument_group('commands')
-    commands.add_argument('-g', '--graph', help='generate a control flow graph', metavar='OUTPUT_FILE')
+    commands.add_argument('-g', '--graph', help='generate a control flow graph')
     commands.add_argument('-x', '--fire-lasers', action='store_true',
                           help='detect vulnerabilities, use with -c, -a or solidity file(s)')
     commands.add_argument('-t', '--truffle', action='store_true',
@@ -157,9 +157,9 @@ def main():
             address, _ = mythril.load_from_address(args.address)
         elif args.solidity_file:
             # Compile Solidity source file(s)
-            if args.graph and len(args.solidity_file) > 1:
+            #if args.graph and len(args.solidity_file) > 1:
-                exit_with_error(args.outform,
+            #    exit_with_error(args.outform,
-                                "Cannot generate call graphs from multiple input files. Please do it one at a time.")
+            #                    "Cannot generate call graphs from multiple input files. Please do it one at a time.")
             address, _ = mythril.load_from_solidity(args.solidity_file)  # list of files
         else:
             exit_with_error(args.outform,
@@ -185,17 +185,15 @@ def main():
                 exit_with_error(args.outform, "input files do not contain any valid contracts")
 
             if args.graph:
-                # dot this for all contracts or just the first?
+                html = mythril.graph_html(mythril.contracts[0], address=address,
-                for nr, contract in enumerate(mythril.contracts):
+                                          enable_physics=args.enable_physics, phrackify=args.phrack,
-                    html = mythril.graph_html(contract, address=address,
+                                          max_depth=args.max_depth)
-                                              enable_physics=args.enable_physics, phrackify=args.phrack,
+
-                                              max_depth=args.max_depth)
+                try:
-
+                    with open(args.graph, "w") as f:
-                    try:
+                        f.write(html)
-                        with open("graph_%s_%d_%s" % (args.graph, nr, contract.name), "w") as f:
+                except Exception as e:
-                            f.write(html)
+                    exit_with_error(args.outform, "Error saving graph: " + str(e))
-                    except Exception as e:
-                        exit_with_error(args.outform, "Error saving graph: " + str(e))
 
             else:
                 report = mythril.fire_lasers(address=address,