feat: implement billing system with Pally integration
- Added billing models and repository for handling bills. - Introduced BillingService to manage payment initiation and bill creation. - Updated user service to remove ticket-related functionality. - Refactored settings to include new billing-related fields. - Removed ticket-related handlers and schemas. - Added new billing handlers for processing payments. - Updated database schema with new bills table and status enum. - Enhanced utility functions to calculate prices considering discounts. - Introduced prehandling for non-private messages. - Updated main application to initialize new billing services and repositories.
This commit is contained in:
25
repositories/bills.py
Normal file
25
repositories/bills.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from db.models.bills import Bill, BillStatus
|
||||
|
||||
|
||||
class BillsRepository:
|
||||
async def get_bill_by_id(self, session: AsyncSession, bill_id: int) -> Bill | None:
|
||||
stmt = select(Bill).where(Bill.id == bill_id)
|
||||
res = await session.execute(stmt)
|
||||
|
||||
return res.scalar_one_or_none()
|
||||
|
||||
async def create(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
*,
|
||||
creator_id: int,
|
||||
amount: int,
|
||||
status: BillStatus | None = BillStatus.NEW,
|
||||
) -> Bill:
|
||||
bill = Bill(creator_id=creator_id, amount=amount, status=status)
|
||||
session.add(bill)
|
||||
await session.commit()
|
||||
return bill
|
||||
@@ -1,55 +0,0 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from db.models.tickets import Ticket, TicketStatus
|
||||
from schemas.tickets import NewTicketDTO
|
||||
|
||||
|
||||
class TicketRepository:
|
||||
async def get_ticket_by_user_id(
|
||||
self, session: AsyncSession, user_id: int, status: TicketStatus | None = None
|
||||
) -> list[Ticket]:
|
||||
stmt = select(Ticket).where(Ticket.creator_id == user_id)
|
||||
if status:
|
||||
stmt = stmt.where(Ticket.status == status)
|
||||
res = await session.execute(stmt)
|
||||
|
||||
return list(res.scalars().all())
|
||||
|
||||
async def get_available_ticket_by_user_id(
|
||||
self, session: AsyncSession, user_id: int
|
||||
) -> Ticket | None:
|
||||
stmt = (
|
||||
select(Ticket)
|
||||
.where(Ticket.creator_id == user_id)
|
||||
.where(Ticket.status != TicketStatus.CLOSED)
|
||||
)
|
||||
res = await session.execute(stmt)
|
||||
|
||||
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) -> 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:
|
||||
ticket = Ticket(creator_id=ticket_data.creator_id)
|
||||
session.add(ticket)
|
||||
await session.commit()
|
||||
|
||||
return ticket
|
||||
|
||||
async def update_ticket_status(
|
||||
self, session: AsyncSession, ticket: Ticket, status: TicketStatus
|
||||
):
|
||||
ticket.status = status
|
||||
await session.commit()
|
||||
return ticket
|
||||
|
||||
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