(fix) Turned market classes and token class into dataclass with eq and hash support

Author: Abel Armoa

pyinjective/core/market.py

@@ -2,11 +2,11 @@
 from decimal import Decimal
 from typing import Optional
 
-from constant import ADDITIONAL_CHAIN_FORMAT_DECIMALS, Denom
+from pyinjective.constant import ADDITIONAL_CHAIN_FORMAT_DECIMALS, Denom
 from pyinjective.core.token import Token
 
 
-@dataclass
+@dataclass(eq=True, frozen=True)
 class SpotMarket:
     id: str
     status: str
@@ -34,7 +34,7 @@ def price_to_chain_format(self, human_readable_value: Decimal) -> Decimal:
 
         return extended_chain_formatted_value
 
-@dataclass
+@dataclass(eq=True, frozen=True)
 class DerivativeMarket:
     id: str
     status: str
@@ -92,7 +92,7 @@ def calculate_margin_in_chain_format(
 
         return extended_chain_formatted_margin
 
-@dataclass
+@dataclass(eq=True, frozen=True)
 class BinaryOptionMarket:
     id: str
     status: str

pyinjective/core/token.py

@@ -1,25 +1,15 @@
+from dataclasses import dataclass
 from decimal import Decimal
 
-
+@dataclass(eq=True, frozen=True)
 class Token:
-
-    def __init__(
-        self,
-        name: str,
-        symbol: str,
-        denom: str,
-        address: str,
-        decimals: int,
-        logo: str,
-        updated: int,
-    ):
-        self.name = name
-        self.symbol = symbol
-        self.denom = denom
-        self.address = address
-        self.decimals = decimals
-        self.logo = logo
-        self.updated = updated
+    name: str
+    symbol: str
+    denom: str
+    address: str
+    decimals: int
+    logo: str
+    updated: int
 
     def chain_formatted_value(self, human_readable_value: Decimal) -> Decimal:
         return human_readable_value * Decimal(f"1e{self.decimals}")