✨ [feat][backend] account timezone field

Author: tomy0000000

backend/alembic/versions/db51787ee569_add_nullable_timezone_to_account.py

@@ -0,0 +1,26 @@
+"""Add nullable timezone to account
+
+Revision ID: db51787ee569
+Revises: 3f6b892b23da
+Create Date: 2026-06-10 00:30:51.945994
+
+"""
+
+import sqlalchemy as sa
+
+from alembic import op
+from kayman.schemas._custom_types import SATimezone
+
+# revision identifiers, used by Alembic.
+revision = "db51787ee569"
+down_revision = "3f6b892b23da"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.add_column("account", sa.Column("timezone", SATimezone(), nullable=True))
+
+
+def downgrade():
+    op.drop_column("account", "timezone")

backend/kayman/schemas/account.py

@@ -1,7 +1,10 @@
 from decimal import Decimal
 from typing import TYPE_CHECKING
 
-from sqlmodel import Field, Relationship, SQLModel
+from pydantic_extra_types.timezone_name import TimeZoneName
+from sqlmodel import Column, Field, Relationship, SQLModel
+
+from kayman.schemas._custom_types import SATimezone
 
 if TYPE_CHECKING:
     from kayman.schemas.currency import Currency
@@ -11,12 +14,14 @@
 class AccountBase(SQLModel):
     name: str
     currency_code: str = Field(foreign_key="currency.code")
+    timezone: TimeZoneName | None = None
 
 
 class Account(AccountBase, table=True):
     __tablename__ = "account"
     id: int | None = Field(primary_key=True, default=None)
     balance: Decimal
+    timezone: TimeZoneName | None = Field(sa_column=Column(SATimezone(), nullable=True))
     currency: "Currency" = Relationship(back_populates="accounts")
     transactions: list["Transaction"] = Relationship(back_populates="account")
 
@@ -28,6 +33,7 @@ class AccountCreate(AccountBase):
 class AccountRead(AccountBase):
     id: int
     balance: Decimal
+    timezone: TimeZoneName | None = None
 
 
 class AccountUpdate(SQLModel):

backend/kayman/tests/crud/test_account.py

@@ -25,6 +25,7 @@ def test_create_account(session: Session):
     assert db_account.id is not None
     assert db_account.name == account.name
     assert db_account.currency_code == account.currency_code
+    assert db_account.timezone == account.timezone
     assert db_account.balance == 0
 
 
@@ -35,6 +36,7 @@ def test_read_account(session: Session):
     assert db_account.id == account.id
     assert db_account.name == account.name
     assert db_account.currency_code == account.currency_code
+    assert db_account.timezone == account.timezone
     assert db_account.balance == account.balance
 
 
@@ -50,6 +52,7 @@ def test_read_accounts(session: Session):
     for account, db_account in zip(accounts, db_accounts, strict=True):
         assert db_account.balance == account.balance
         assert db_account.currency_code == account.currency_code
+        assert db_account.timezone == account.timezone
         assert db_account.id == account.id
         assert db_account.name == account.name
 
@@ -63,6 +66,7 @@ def test_read_accounts_by_ids(session: Session):
     for account, db_account in zip(interest_accounts, db_accounts, strict=True):
         assert db_account.balance == account.balance
         assert db_account.currency_code == account.currency_code
+        assert db_account.timezone == account.timezone
         assert db_account.id == account.id
         assert db_account.name == account.name
 
@@ -94,6 +98,7 @@ def test_update_accounts(session: Session):
         assert updated_account.id == account.id
         assert updated_account.name == account_update.name  # Updated
         assert updated_account.currency_code == account.currency_code  # Not updated
+        assert updated_account.timezone == account.timezone  # Not updated
         assert updated_account.balance == account.balance  # Not updated
 
 
@@ -124,6 +129,7 @@ def test_update_account_balances(session: Session):
         assert updated_account.id == account.id
         assert updated_account.name == account.name
         assert updated_account.currency_code == account.currency_code
+        assert updated_account.timezone == account.timezone
         assert updated_account.balance == balance + amount
 
 
@@ -138,13 +144,15 @@ def test_update_account_balances_no_commit(session: Session, session_2: Session)
     assert updated_account.id == account.id
     assert updated_account.name == account.name
     assert updated_account.currency_code == account.currency_code
+    assert updated_account.timezone == account.timezone
     assert updated_account.balance == account_balance + account_amounts[account.id]
 
     # The account balance should not be updated from other sessions (yet)
     session_2_account = session_2.get(Account, account.id)
     assert session_2_account.id == account.id
     assert session_2_account.name == account.name
     assert session_2_account.currency_code == account.currency_code
+    assert session_2_account.timezone == account.timezone
     assert session_2_account.balance == account_balance
 
     # Commit the change from main session
@@ -156,6 +164,7 @@ def test_update_account_balances_no_commit(session: Session, session_2: Session)
     assert session_2_account.id == account.id
     assert session_2_account.name == account.name
     assert session_2_account.currency_code == account.currency_code
+    assert session_2_account.timezone == account.timezone
     assert session_2_account.balance == account_balance + account_amounts[account.id]
 
 

backend/kayman/tests/factories/account.py

@@ -14,3 +14,4 @@ class Meta:
     currency = factory.SubFactory("kayman.tests.factories.currency.CurrencyFactory")
     currency_code = factory.SelfAttribute("currency.code")
     balance = factory.Faker("pydecimal", left_digits=5, right_digits=2)
+    timezone = "UTC"