fix: change credit amount type from Decimal to int

Author: amalcaraz

deployment/migrations/versions/0035_a1b2c3d4e5f6_add_credit_balances_table.py

@@ -23,7 +23,7 @@ def upgrade() -> None:
         'credit_balances',
         sa.Column('id', sa.BigInteger(), autoincrement=True),
         sa.Column('address', sa.String(), nullable=False),
-        sa.Column('amount', sa.DECIMAL(), nullable=False),
+        sa.Column('amount', sa.BigInteger(), nullable=False),
         sa.Column('ratio', sa.DECIMAL(), nullable=True),
         sa.Column('tx_hash', sa.String(), nullable=True),
         sa.Column('token', sa.String(), nullable=True),

src/aleph/db/accessors/balances.py

@@ -189,7 +189,7 @@ def get_updated_balance_accounts(session: DbSession, last_update: dt.datetime):
     return (session.execute(select_stmt)).scalars().all()
 
 
-def get_credit_balance(session: DbSession, address: str) -> Decimal:
+def get_credit_balance(session: DbSession, address: str) -> int:
     now = utc_now()
 
     # Sum all non-expired credit balances for the address
@@ -203,7 +203,7 @@ def get_credit_balance(session: DbSession, address: str) -> Decimal:
         )
     ).scalar()
 
-    return result if result is not None else Decimal(0)
+    return result if result is not None else 0
 
 
 def get_credit_balances(
@@ -288,7 +288,7 @@ def update_credit_balances(
     csv_rows = []
     for index, credit_entry in enumerate(credits_list):
         address = credit_entry["address"]
-        amount = Decimal(credit_entry["amount"])  # Cast from string to Decimal
+        amount = int(credit_entry["amount"])  # Cast to integer
         ratio = Decimal(credit_entry["ratio"])
         tx_hash = credit_entry["tx_hash"]
         provider = credit_entry["provider"]
@@ -354,7 +354,7 @@ def update_credit_balances_airdrop(
     csv_rows = []
     for index, credit_entry in enumerate(credits_list):
         address = credit_entry["address"]
-        amount = Decimal(credit_entry["amount"])
+        amount = int(credit_entry["amount"])  # Cast to integer
         origin = credit_entry.get("origin", "")
         expiration_timestamp = credit_entry.get("expiration", 0)
 

src/aleph/db/models/balances.py

@@ -48,7 +48,7 @@ class AlephCreditBalanceDb(Base):
     id: int = Column(BigInteger, autoincrement=True)
 
     address: str = Column(String, nullable=False, index=True)
-    amount: Decimal = Column(DECIMAL, nullable=False)
+    amount: int = Column(BigInteger, nullable=False)
     ratio: Optional[Decimal] = Column(DECIMAL, nullable=True)
     tx_hash: Optional[str] = Column(String, nullable=True)
     token: Optional[str] = Column(String, nullable=True)

src/aleph/schemas/api/accounts.py

@@ -30,7 +30,7 @@ class GetAccountBalanceResponse(BaseModel):
 
 class GetAccountCreditBalanceResponse(BaseModel):
     address: str
-    credits: FloatDecimal
+    credits: int
 
 
 class GetAccountFilesQueryParams(BaseModel):
@@ -96,7 +96,7 @@ class AddressCreditBalanceResponse(BaseModel):
     model_config = ConfigDict(from_attributes=True)
 
     address: str
-    credits: FloatDecimal
+    credits: int
 
 
 class GetAccountFilesResponseItem(BaseModel):