diff --git a/mythril/analysis/report.py b/mythril/analysis/report.py index 1ce5219b..44fede86 100644 --- a/mythril/analysis/report.py +++ b/mythril/analysis/report.py @@ -3,7 +3,6 @@ import logging import json import operator from jinja2 import PackageLoader, Environment - from typing import Dict, List import hashlib diff --git a/mythril/support/source_support.py b/mythril/support/source_support.py index 68d7a6aa..51d94ee8 100644 --- a/mythril/support/source_support.py +++ b/mythril/support/source_support.py @@ -48,5 +48,10 @@ class Source: else: assert False # Fail hard - def get_source_index(self, bytecode_hash): + def get_source_index(self, bytecode_hash: str) -> int: + """ + Find the contract index in the list + :param bytecode_hash: The contract hash + :return: The index of the contract in the _source_hash list + """ return self._source_hash.index(bytecode_hash) diff --git a/mythril/support/support_utils.py b/mythril/support/support_utils.py index 6912ddd3..1edbefff 100644 --- a/mythril/support/support_utils.py +++ b/mythril/support/support_utils.py @@ -26,7 +26,7 @@ class Singleton(type): return cls._instances[cls] -def get_code_hash(code): +def get_code_hash(code: str) -> str: """ :param code: bytecode :return: Returns hash of the given bytecode