Coverage for dibbler / models / UserCache.py: 100%

1from __future__ import annotations 

2from typing import TYPE_CHECKING 

3 

4from sqlalchemy import Integer, ForeignKey 

5from sqlalchemy.orm import Mapped, mapped_column, relationship 

6 

7from dibbler.models import Base 

8 

9if TYPE_CHECKING: 

10 from dibbler.models import LastCacheTransaction, User 

11 

12 

13# More like user balance cash money flow, amirite? 

14class UserCache(Base): 

15 id: Mapped[int] = mapped_column(Integer, primary_key=True) 

16 """internal database id""" 

17 

18 user_id: Mapped[int] = mapped_column(ForeignKey('user.id')) 

19 user: Mapped[User] = relationship( 

20 lazy="joined", 

21 foreign_keys=[user_id], 

22 ) 

23 

24 balance: Mapped[int] = mapped_column(Integer) 

25 

26 last_cache_transaction_id: Mapped[int | None] = mapped_column(ForeignKey("last_cache_transaction.id"), nullable=True) 

27 last_cache_transaction: Mapped[LastCacheTransaction | None] = relationship( 

28 lazy="joined", 

29 foreign_keys=[last_cache_transaction_id], 

30 )