internal.queries.seller_issue_reports

  1# Code generated by sqlc. DO NOT EDIT.
  2# versions:
  3#   sqlc v1.30.0
  4# source: seller_issue_reports.sql
  5import pydantic
  6from typing import AsyncIterator, Optional
  7
  8import sqlalchemy
  9import sqlalchemy.ext.asyncio
 10
 11from internal.queries import models
 12
 13
 14CREATE_SELLER_ISSUE_REPORT = """-- name: create_seller_issue_report \\:one
 15INSERT INTO seller_issue_reports (reservation_id, issue_type, description)
 16VALUES (:p1, :p2, :p3)
 17RETURNING report_id, reservation_id, issue_type, description, created_at, status
 18"""
 19
 20
 21class CreateSellerIssueReportParams(pydantic.BaseModel):
 22    reservation_id: int
 23    issue_type: models.SellerIssueType
 24    description: str
 25
 26
 27DELETE_SELLER_ISSUE_REPORT = """-- name: delete_seller_issue_report \\:one
 28DELETE FROM seller_issue_reports
 29WHERE report_id = :p1
 30RETURNING report_id, reservation_id, issue_type, description, created_at, status
 31"""
 32
 33
 34GET_SELLER_ISSUE_REPORTS = """-- name: get_seller_issue_reports \\:many
 35SELECT report_id, reservation_id, issue_type, description, created_at, status FROM seller_issue_reports
 36"""
 37
 38
 39GET_SELLER_ISSUE_REPORTS_BY_USER = """-- name: get_seller_issue_reports_by_user \\:many
 40SELECT r.report_id, r.reservation_id, r.issue_type, r.description, r.created_at, r.status FROM seller_issue_reports r
 41JOIN reservations res ON r.reservation_id = res.reservation_id
 42WHERE res.consumer_id = :p1
 43"""
 44
 45
 46UPDATE_SELLER_ISSUE_REPORT_STATUS = """-- name: update_seller_issue_report_status \\:one
 47UPDATE seller_issue_reports
 48SET status = :p2
 49WHERE report_id = :p1
 50RETURNING report_id, reservation_id, issue_type, description, created_at, status
 51"""
 52
 53
 54class UpdateSellerIssueReportStatusParams(pydantic.BaseModel):
 55    report_id: int
 56    status: models.IssueStatus
 57
 58
 59class AsyncQuerier:
 60    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
 61        self._conn = conn
 62
 63    async def create_seller_issue_report(self, arg: CreateSellerIssueReportParams) -> Optional[models.SellerIssueReport]:
 64        row = (await self._conn.execute(sqlalchemy.text(CREATE_SELLER_ISSUE_REPORT), {"p1": arg.reservation_id, "p2": arg.issue_type, "p3": arg.description})).first()
 65        if row is None:
 66            return None
 67        return models.SellerIssueReport(
 68            report_id=row[0],
 69            reservation_id=row[1],
 70            issue_type=row[2],
 71            description=row[3],
 72            created_at=row[4],
 73            status=row[5],
 74        )
 75
 76    async def delete_seller_issue_report(self, *, report_id: int) -> Optional[models.SellerIssueReport]:
 77        row = (await self._conn.execute(sqlalchemy.text(DELETE_SELLER_ISSUE_REPORT), {"p1": report_id})).first()
 78        if row is None:
 79            return None
 80        return models.SellerIssueReport(
 81            report_id=row[0],
 82            reservation_id=row[1],
 83            issue_type=row[2],
 84            description=row[3],
 85            created_at=row[4],
 86            status=row[5],
 87        )
 88
 89    async def get_seller_issue_reports(self) -> AsyncIterator[models.SellerIssueReport]:
 90        result = await self._conn.stream(sqlalchemy.text(GET_SELLER_ISSUE_REPORTS))
 91        async for row in result:
 92            yield models.SellerIssueReport(
 93                report_id=row[0],
 94                reservation_id=row[1],
 95                issue_type=row[2],
 96                description=row[3],
 97                created_at=row[4],
 98                status=row[5],
 99            )
100
101    async def get_seller_issue_reports_by_user(self, *, consumer_id: int) -> AsyncIterator[models.SellerIssueReport]:
102        result = await self._conn.stream(sqlalchemy.text(GET_SELLER_ISSUE_REPORTS_BY_USER), {"p1": consumer_id})
103        async for row in result:
104            yield models.SellerIssueReport(
105                report_id=row[0],
106                reservation_id=row[1],
107                issue_type=row[2],
108                description=row[3],
109                created_at=row[4],
110                status=row[5],
111            )
112
113    async def update_seller_issue_report_status(self, arg: UpdateSellerIssueReportStatusParams) -> Optional[models.SellerIssueReport]:
114        row = (await self._conn.execute(sqlalchemy.text(UPDATE_SELLER_ISSUE_REPORT_STATUS), {"p1": arg.report_id, "p2": arg.status})).first()
115        if row is None:
116            return None
117        return models.SellerIssueReport(
118            report_id=row[0],
119            reservation_id=row[1],
120            issue_type=row[2],
121            description=row[3],
122            created_at=row[4],
123            status=row[5],
124        )