Add dump_to_file() functionality for exploit.

cleanunicorn · Jul 23, 2019 · 94679a0 · 94679a0
1 parent 62ab9cb
commit 94679a0
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 5 deletions.
diff --git a/test/input-tx.json b/test/input-tx.json
@@ -1,17 +1,17 @@
 [
     [
         {
-            "input": "0x4e71e0c8",
+            "data": "0x4e71e0c8",
             "value": 100000000000000000
         },
         {
-            "input": "0x2e64cec1",
+            "data": "0x2e64cec1",
             "value": "0x0"
         }    
     ],
     [
         {
-            "input": "0xdeadbeef"
+            "data": "0xdeadbeef"
         }
     ]
 ]
diff --git a/theo/exploit/exploit.py b/theo/exploit/exploit.py
@@ -208,3 +208,12 @@ def wait_for(self, contract, tx, flush=False):
                         )
                     )
                     return pending_tx
+
+    def dump_to_file(self, file=None):
+        from theo import dump
+
+        exploit_object = []
+        for tx in self.txs:
+            exploit_object.append(tx.__dict__)
+
+        dump(ob=[exploit_object], filename=file)
diff --git a/theo/file/__init__.py b/theo/file/__init__.py
@@ -26,7 +26,7 @@ def exploits_from_file(
     for exploit in exploit_list:
         txs = []
         for tx in exploit:
-            txs.append(Tx(data=tx.get("input", "0x"), value=tx.get("value", 0)))
+            txs.append(Tx(data=tx.get("data", "0x"), value=tx.get("value", 0)))
 
         exploits.append(
             Exploit(

diff --git a/theo/version.py b/theo/version.py
@@ -1 +1 @@
-__version__ = "v0.5.6"
+__version__ = "v0.6.0"