Skip to content

Commit d39bfd5

Browse files
committed
Fix iteration and py2
1 parent df52f90 commit d39bfd5

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

pyevmasm/__main__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ def main():
2424
'Also an unsigned block number is accepted to select the fork.')
2525

2626
args = parser.parse_args(sys.argv[1:])
27-
2827
arg_fork = args.fork.lower()
2928
if arg_fork not in accepted_forks:
3029
try:

pyevmasm/evmasm.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -658,7 +658,7 @@ def _search_by_name(self, k):
658658
return self._search_by_opcode(self._name_to_opcode[k])
659659

660660
def _search_by_opcode(self, k):
661-
return (k, *self._instruction_list[k])
661+
return (k,) + self._instruction_list[k]
662662

663663
def _search(self, k):
664664
try:
@@ -677,13 +677,18 @@ def get(self, k, default=None):
677677
return default
678678

679679
def __contains__(self, k):
680-
return k in self._instruction_list or k in self._name_to_opcode
680+
return k in self._instruction_list or k in self._name_to_opcode
681+
682+
def __iter__(self):
683+
for k, i in self._instruction_list.items():
684+
yield Instruction(*((k,)+ i))
681685

682686
def keys(self):
683687
return self._instruction_list.keys()
684688

685689
def __repr__(self):
686-
return '{}({})'.format(type(self).__name__, super(InstructionTable, self).__repr__())
690+
return repr(self._instruction_list)
691+
687692

688693

689694
# from http://gavwood.com/paper.pdf

0 commit comments

Comments
 (0)