class StateMutability(StringEnum):
PURE = StringEnum.auto()
VIEW = StringEnum.auto()
NONPAYABLE = StringEnum.auto()
PAYABLE = StringEnum.auto()
@classmethod
def from_abi(cls, abi_dict: Dict) -> "StateMutability":
"""
Extract stateMutability from an entry in a contract's ABI
"""
if "stateMutability" in abi_dict:
return cls(abi_dict["stateMutability"])
elif abi_dict.get("payable"):
return StateMutability.PAYABLE
elif "constant" in abi_dict and abi_dict["constant"]:
return StateMutability.VIEW
else: # Assume nonpayable if neither field is there, or constant/payable not set
return StateMutability.NONPAYABLE
class FunctionVisibility(StringEnum):
EXTERNAL = StringEnum.auto()
INTERNAL = StringEnum.auto()