- feat(translate.py): add support for ContractCreation action type

- feat(action_type.py): add ContractCreation action type
- feat(base.py): add base Action class and EventHandleFunc type
- feat(tx_type.py): update actions field in TaggedTx to accept List[Action]

decodex/translate/translate.py

@@ -39,6 +39,7 @@
 from decodex.type import AccountBalanceChanged
 from decodex.type import Action
 from decodex.type import AssetBalanceChanged
+from decodex.type import ContractCreation
 from decodex.type import EventHandleFunc
 from decodex.type import TaggedAddr
 from decodex.type import TaggedTx
@@ -382,6 +383,8 @@ def _process_tx(self, tx: Tx, max_workers: int) -> TaggedTx:
  actions = others
  elif len(transfers) > 0:
  actions = transfers
+ elif tx.get("contract_created"):
+ actions = [ContractCreation(deployer=tx_from, contract=tx_contract_created)]
  elif len(tx["input"]) > 0 and tx["input"] != "0x":
  msg = parse_utf8(tx["input"])
  if msg:

decodex/type/__init__.py

@@ -1,11 +1,10 @@
-from .action_type import Action
 from .action_type import AddLiquidityAction
 from .action_type import BorrowAction
 from .action_type import CollectAction
+from .action_type import ContractCreation
 from .action_type import DepositAction
 from .action_type import DisableCollateralAction
 from .action_type import EnableCollateralAction
-from .action_type import EventHandleFunc
 from .action_type import FlashloanAction
 from .action_type import OwnerChangedAction
 from .action_type import PoolCreatedAction
@@ -16,6 +15,8 @@
 from .action_type import TransferAction
 from .action_type import UTF8Message
 from .action_type import WithdrawAction
+from .base import Action
+from .base import EventHandleFunc
 from .rpc_type import RawTraceCallResponse
 from .rpc_type import RawTraceCallResult
 from .tx_type import AccountBalanceChanged
@@ -36,6 +37,7 @@
  "EventHandleFunc",
  "EventPayload",
  "Action",
+ "ContractCreation",
  "TransferAction",
  "UTF8Message",
  "SwapAction",

decodex/type/action_type.py

@@ -1,22 +1,26 @@
 from abc import abstractmethod
+from dataclasses import asdict
 from dataclasses import dataclass
 from typing import Any
 from typing import Callable
 from typing import Dict
 from typing import Optional
 
+from .base import Action
 from .tx_type import ERC20Compatible
 from .tx_type import TaggedAddr
 from decodex.utils import fmt_addr
 
 
-class Action:
- @abstractmethod
- def __repr__(self):
-  raise NotImplementedError
+@dataclass(frozen=True)
+class ContractCreation(Action):
+ deployer: TaggedAddr
+ contract: TaggedAddr
 
+ action: str = "contract_creation"
 
-EventHandleFunc = Callable[[Dict[str, Any]], Optional[Action]]
+ def __str__(self):
+ return f"{fmt_addr(self.deployer)} deployed {fmt_addr(self.contract)}"
 
 
 @dataclass(frozen=True)
@@ -31,7 +35,7 @@ class UTF8Message(Action):
 
  action: str = "utf8_message"
 
- def __repr__(self):
+ def __str__(self):
  return f"{fmt_addr(self.sender)} sent message to {fmt_addr(self.receiver)}: {self.message}"
 
 
@@ -48,7 +52,7 @@ class TransferAction(Action):
 
  action: str = "transfer"
 
- def __repr__(self):
+ def __str__(self):
  return f"{fmt_addr(self.sender)} transferred {self.amount} {fmt_addr(self.token)} to {fmt_addr(self.receiver)}"
 
 
@@ -66,7 +70,7 @@ class SwapAction(Action):
 
  action: str = "swap"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Swap {pay_amount} {pay_token} for {recv_amount} {recv_token} on {pool}"
  return tmpl.format(
  pool=fmt_addr(self.pool),
@@ -91,7 +95,7 @@ class AddLiquidityAction(Action):
 
  action: str = "add_liquidity"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Add {amount_0} {token_0} and {amount_1} {token_1} liquidity to {pool}"
  return tmpl.format(
  amount_1=fmt_addr(self.amount_1),
@@ -116,7 +120,7 @@ class RemoveLiquidityAction(Action):
 
  action: str = "remove_liquidity"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Remove {amount_0} {token_0} and {amount_1} {token_1} liquidity from {pool}"
  return tmpl.format(
  token_0=fmt_addr(self.token_0),
@@ -139,7 +143,7 @@ class PoolCreatedAction(Action):
 
  action: str = "pool_created"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Create {token_0} / {token_1} pool"
  if not self.fee:
  return tmpl.format(
@@ -168,7 +172,7 @@ class CollectAction(Action):
 
  action: str = "collect"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Collect {amount_0} {token_0} and {amount_1} {token_1} from {pool}"
  return tmpl.format(
  token_0=fmt_addr(self.token_0),
@@ -191,7 +195,7 @@ class OwnerChangedAction(Action):
 
  action: str = "owner_changed"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Change owner of {pool} from {old_owner} to {new_owner}"
  return tmpl.format(
  pool=fmt_addr(self.pool),
@@ -212,7 +216,7 @@ class BorrowAction(Action):
 
  action: str = "borrow"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Borrow {amount} {token} from {pool}"
  return tmpl.format(
  amount=self.amount,
@@ -233,7 +237,7 @@ class RepayAction(Action):
 
  action: str = "repay"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Repay {amount} {token} to {pool}"
  return tmpl.format(
  amount=self.amount,
@@ -254,7 +258,7 @@ class DepositAction(Action):
 
  action: str = "deposit"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Deposit {amount} {token} to {pool}"
  return tmpl.format(
  amount=self.amount,
@@ -276,7 +280,7 @@ class WithdrawAction(Action):
 
  action: str = "withdraw"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  if not self.receiver:
  tmpl = "Withdraw {amount} {token} from {pool}"
  return tmpl.format(
@@ -305,7 +309,7 @@ class FlashloanAction(Action):
 
  action: str = "flashloan"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Flashloan {amount} {token} from {pool}"
  return tmpl.format(
  amount=self.amount,
@@ -325,7 +329,7 @@ class EnableCollateralAction(Action):
 
  action: str = "enable_collateral"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Enable {token} as collateral in {pool}"
  return tmpl.format(
  token=fmt_addr(self.token),
@@ -344,7 +348,7 @@ class DisableCollateralAction(Action):
 
  action: str = "disable_collateral"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Disable {token} as collateral in {pool}"
  return tmpl.format(
  token=fmt_addr(self.token),
@@ -364,7 +368,7 @@ class SupplyAction(Action):
 
  action: str = "supply"
 
- def __repr__(self) -> str:
+ def __str__(self) -> str:
  tmpl = "Supply {amount} {token} to {pool}"
  return tmpl.format(
  token=fmt_addr(self.token),

decodex/type/base.py

@@ -0,0 +1,24 @@
+from abc import abstractmethod
+from dataclasses import dataclass
+from typing import Any
+from typing import Callable
+from typing import Dict
+from typing import Optional
+
+
+@dataclass(frozen=True)
+class Action:
+ @property
+ @abstractmethod
+ def action(self) -> str:
+ raise NotImplementedError
+
+ @abstractmethod
+ def __str__(self):
+ raise NotImplementedError
+
+ def dict(self):
+ return vars(self)
+
+
+EventHandleFunc = Callable[[Dict[str, Any]], Optional[Action]]

decodex/type/tx_type.py

@@ -6,6 +6,8 @@
 from typing import Optional
 from typing import TypedDict
 
+from .base import Action
+
 
 Log = TypedDict(
  "Log",
@@ -89,7 +91,7 @@ class ERC20Compatible(TaggedAddr):
  "status": int, # status of the transaction
  "reason": str, # reason of the transaction if failed
  "method": str, # calling method of the transaction
- "actions": List[str], # actions of the transaction
+ "actions": List[Action], # actions of the transaction
  "balance_change": List[AccountBalanceChanged], # balance change of the transaction
  },
  total=False,