fix!: removed .commit() from repository level
This commit is contained in:
@@ -1,14 +1,14 @@
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from db.models.link_codes import LinkCode
|
||||
from db.session import UnitOfWork
|
||||
from schemas.enums import LinkCodeStatus
|
||||
|
||||
|
||||
async def create_link_code(
|
||||
session: AsyncSession,
|
||||
uow: UnitOfWork,
|
||||
*,
|
||||
code: str,
|
||||
user_id: int,
|
||||
@@ -22,20 +22,17 @@ async def create_link_code(
|
||||
expires_at=expires_at,
|
||||
)
|
||||
|
||||
session.add(link_code)
|
||||
await session.commit()
|
||||
uow.session.add(link_code)
|
||||
return link_code
|
||||
|
||||
|
||||
async def get_link_code_by_code(session: AsyncSession, code: str) -> LinkCode | None:
|
||||
async def get_link_code_by_code(uow: UnitOfWork, code: str) -> LinkCode | None:
|
||||
stmt = select(LinkCode).where(LinkCode.code == code)
|
||||
r = await session.execute(stmt)
|
||||
r = await uow.session.execute(stmt)
|
||||
|
||||
return r.scalar_one_or_none()
|
||||
|
||||
|
||||
async def use_link_code(session: AsyncSession, code: LinkCode) -> LinkCode:
|
||||
async def use_link_code(uow: UnitOfWork, code: LinkCode) -> LinkCode:
|
||||
code.status = LinkCodeStatus.USED
|
||||
await session.commit()
|
||||
|
||||
return code
|
||||
|
||||
Reference in New Issue
Block a user