Coverage for src/kwai/modules/identity/tokens/user_log_db_repository.py: 100%

1"""Module that defines a User Log repository for a database.""" 

2 

3from kwai.core.db.database import Database 

4from kwai.modules.identity.tokens.token_tables import UserLogRow 

5from kwai.modules.identity.tokens.user_log import UserLogEntity, UserLogIdentifier 

6from kwai.modules.identity.tokens.user_log_repository import UserLogRepository 

7 

8 

9class UserLogDbRepository(UserLogRepository): 

10 """Class that represents a User Log repository for a database.""" 

11 

12 def __init__(self, database: Database): 

13 self._db = database 

14 

15 async def create(self, user_log: UserLogEntity) -> UserLogEntity: 

16 new_id = await self._db.insert( 

17 UserLogRow.__table_name__, UserLogRow.persist(user_log) 

18 ) 

19 return user_log.set_id(UserLogIdentifier(new_id))