internal.queries.admin_issue_reports

  1# Code generated by sqlc. DO NOT EDIT.
  2# versions:
  3#   sqlc v1.30.0
  4# source: admin_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_ADMIN_ISSUE_REPORT = """-- name: create_admin_issue_report \\:one
 15INSERT INTO admin_issue_reports (user_id, issue_type, description)
 16VALUES (:p1, :p2, :p3)
 17RETURNING report_id, user_id, issue_type, description, created_at, status
 18"""
 19
 20
 21class CreateAdminIssueReportParams(pydantic.BaseModel):
 22    user_id: int
 23    issue_type: models.AdminIssueType
 24    description: str
 25
 26
 27DELETE_ADMIN_ISSUE_REPORT = """-- name: delete_admin_issue_report \\:one
 28DELETE FROM admin_issue_reports
 29WHERE report_id = :p1
 30RETURNING report_id, user_id, issue_type, description, created_at, status
 31"""
 32
 33
 34GET_ADMIN_ISSUE_REPORTS = """-- name: get_admin_issue_reports \\:many
 35SELECT report_id, user_id, issue_type, description, created_at, status FROM admin_issue_reports
 36"""
 37
 38
 39GET_ADMIN_ISSUE_REPORTS_BY_USER = """-- name: get_admin_issue_reports_by_user \\:many
 40SELECT report_id, user_id, issue_type, description, created_at, status FROM admin_issue_reports
 41WHERE user_id = :p1
 42"""
 43
 44
 45UPDATE_ADMIN_ISSUE_REPORT_STATUS = """-- name: update_admin_issue_report_status \\:one
 46UPDATE admin_issue_reports
 47SET status = :p2
 48WHERE report_id = :p1
 49RETURNING report_id, user_id, issue_type, description, created_at, status
 50"""
 51
 52
 53class UpdateAdminIssueReportStatusParams(pydantic.BaseModel):
 54    report_id: int
 55    status: models.IssueStatus
 56
 57
 58class AsyncQuerier:
 59    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
 60        self._conn = conn
 61
 62    async def create_admin_issue_report(self, arg: CreateAdminIssueReportParams) -> Optional[models.AdminIssueReport]:
 63        row = (await self._conn.execute(sqlalchemy.text(CREATE_ADMIN_ISSUE_REPORT), {"p1": arg.user_id, "p2": arg.issue_type, "p3": arg.description})).first()
 64        if row is None:
 65            return None
 66        return models.AdminIssueReport(
 67            report_id=row[0],
 68            user_id=row[1],
 69            issue_type=row[2],
 70            description=row[3],
 71            created_at=row[4],
 72            status=row[5],
 73        )
 74
 75    async def delete_admin_issue_report(self, *, report_id: int) -> Optional[models.AdminIssueReport]:
 76        row = (await self._conn.execute(sqlalchemy.text(DELETE_ADMIN_ISSUE_REPORT), {"p1": report_id})).first()
 77        if row is None:
 78            return None
 79        return models.AdminIssueReport(
 80            report_id=row[0],
 81            user_id=row[1],
 82            issue_type=row[2],
 83            description=row[3],
 84            created_at=row[4],
 85            status=row[5],
 86        )
 87
 88    async def get_admin_issue_reports(self) -> AsyncIterator[models.AdminIssueReport]:
 89        result = await self._conn.stream(sqlalchemy.text(GET_ADMIN_ISSUE_REPORTS))
 90        async for row in result:
 91            yield models.AdminIssueReport(
 92                report_id=row[0],
 93                user_id=row[1],
 94                issue_type=row[2],
 95                description=row[3],
 96                created_at=row[4],
 97                status=row[5],
 98            )
 99
100    async def get_admin_issue_reports_by_user(self, *, user_id: int) -> AsyncIterator[models.AdminIssueReport]:
101        result = await self._conn.stream(sqlalchemy.text(GET_ADMIN_ISSUE_REPORTS_BY_USER), {"p1": user_id})
102        async for row in result:
103            yield models.AdminIssueReport(
104                report_id=row[0],
105                user_id=row[1],
106                issue_type=row[2],
107                description=row[3],
108                created_at=row[4],
109                status=row[5],
110            )
111
112    async def update_admin_issue_report_status(self, arg: UpdateAdminIssueReportStatusParams) -> Optional[models.AdminIssueReport]:
113        row = (await self._conn.execute(sqlalchemy.text(UPDATE_ADMIN_ISSUE_REPORT_STATUS), {"p1": arg.report_id, "p2": arg.status})).first()
114        if row is None:
115            return None
116        return models.AdminIssueReport(
117            report_id=row[0],
118            user_id=row[1],
119            issue_type=row[2],
120            description=row[3],
121            created_at=row[4],
122            status=row[5],
123        )