Update project configuration and codebase
This commit is contained in:
@@ -1,4 +1,3 @@
|
||||
from typing import Optional
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
@@ -8,7 +7,7 @@ from schemas.tickets import NewTicketDTO
|
||||
|
||||
class TicketRepository:
|
||||
async def get_ticket_by_user_id(
|
||||
self, session: AsyncSession, user_id: int, status: Optional[TicketStatus] = None
|
||||
self, session: AsyncSession, user_id: int, status: TicketStatus | None = None
|
||||
) -> list[Ticket]:
|
||||
stmt = select(Ticket).where(Ticket.creator_id == user_id)
|
||||
if status:
|
||||
@@ -19,7 +18,7 @@ class TicketRepository:
|
||||
|
||||
async def get_available_ticket_by_user_id(
|
||||
self, session: AsyncSession, user_id: int
|
||||
) -> Optional[Ticket]:
|
||||
) -> Ticket | None:
|
||||
stmt = (
|
||||
select(Ticket)
|
||||
.where(Ticket.creator_id == user_id)
|
||||
@@ -30,17 +29,13 @@ class TicketRepository:
|
||||
tickets = list(res.scalars().all())
|
||||
return tickets[0] if tickets else None
|
||||
|
||||
async def get_ticket_by_thread_id(
|
||||
self, session: AsyncSession, thread_id: int
|
||||
) -> Optional[Ticket]:
|
||||
async def get_ticket_by_thread_id(self, session: AsyncSession, thread_id: int) -> Ticket | None:
|
||||
stmt = select(Ticket).where(Ticket.thread_id == thread_id)
|
||||
res = await session.execute(stmt)
|
||||
|
||||
return res.scalar_one_or_none()
|
||||
|
||||
async def create_ticket(
|
||||
self, session: AsyncSession, ticket_data: NewTicketDTO
|
||||
) -> Ticket:
|
||||
async def create_ticket(self, session: AsyncSession, ticket_data: NewTicketDTO) -> Ticket:
|
||||
ticket = Ticket(creator_id=ticket_data.creator_id)
|
||||
session.add(ticket)
|
||||
await session.commit()
|
||||
@@ -54,9 +49,7 @@ class TicketRepository:
|
||||
await session.commit()
|
||||
return ticket
|
||||
|
||||
async def update_ticket_thread_id(
|
||||
self, session: AsyncSession, ticket: Ticket, thread_id: int
|
||||
):
|
||||
async def update_ticket_thread_id(self, session: AsyncSession, ticket: Ticket, thread_id: int):
|
||||
ticket.thread_id = thread_id
|
||||
await session.commit()
|
||||
return ticket
|
||||
|
||||
Reference in New Issue
Block a user