internal.queries.bundle

  1# Code generated by sqlc. DO NOT EDIT.
  2# versions:
  3#   sqlc v1.30.0
  4# source: bundle.sql
  5import datetime
  6import decimal
  7import pydantic
  8from typing import AsyncIterator, Optional
  9
 10import sqlalchemy
 11import sqlalchemy.ext.asyncio
 12
 13from internal.queries import models
 14
 15
 16CREATE_BUNDLE = """-- name: create_bundle \\:one
 17INSERT INTO bundles (seller_id, bundle_name, description, total_qty, carbon_dioxide, price, discount_percentage, window_start, window_end)
 18VALUES (:p1, :p2, :p3, :p4, :p5, :p6, :p7, :p8, :p9)
 19RETURNING bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 20"""
 21
 22
 23class CreateBundleParams(pydantic.BaseModel):
 24    seller_id: int
 25    bundle_name: str
 26    description: str
 27    total_qty: int
 28    carbon_dioxide: int
 29    price: decimal.Decimal
 30    discount_percentage: int
 31    window_start: datetime.datetime
 32    window_end: datetime.datetime
 33
 34
 35DELETE_BUNDLE = """-- name: delete_bundle \\:one
 36DELETE FROM bundles
 37WHERE bundle_id = :p1
 38RETURNING bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 39"""
 40
 41
 42GET_BUNDLE = """-- name: get_bundle \\:one
 43SELECT bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 44FROM bundles
 45WHERE bundle_id=:p1
 46LIMIT 1
 47"""
 48
 49
 50GET_BUNDLE_LOCK = """-- name: get_bundle_lock \\:one
 51SELECT bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 52FROM bundles
 53WHERE bundle_id=:p1
 54FOR UPDATE
 55LIMIT 1
 56"""
 57
 58
 59GET_BUNDLES = """-- name: get_bundles \\:many
 60SELECT bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 61FROM bundles
 62"""
 63
 64
 65GET_SELLERS_ACTIVE_BUNDLES = """-- name: get_sellers_active_bundles \\:many
 66SELECT bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 67FROM bundles
 68WHERE seller_id=:p1 AND window_end >= NOW()
 69"""
 70
 71
 72GET_SELLERS_BUNDLE = """-- name: get_sellers_bundle \\:one
 73SELECT bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 74FROM bundles
 75WHERE seller_id=:p1 AND bundle_id=:p2
 76LIMIT 1
 77"""
 78
 79
 80class GetSellersBundleParams(pydantic.BaseModel):
 81    seller_id: int
 82    bundle_id: int
 83
 84
 85GET_SELLERS_BUNDLES = """-- name: get_sellers_bundles \\:many
 86SELECT bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 87FROM bundles
 88WHERE seller_id=:p1
 89"""
 90
 91
 92UPDATE_BUNDLE = """-- name: update_bundle \\:one
 93UPDATE bundles
 94SET bundle_name=:p3, description=:p4, total_qty=:p5, price=:p6, discount_percentage=:p7, window_start=:p8, window_end=:p9, carbon_dioxide=:p10
 95WHERE bundle_id=:p1 AND seller_id=:p2
 96RETURNING bundle_id, seller_id, bundle_name, description, carbon_dioxide, total_qty, price, discount_percentage, window_start, window_end, created_at
 97"""
 98
 99
100class UpdateBundleParams(pydantic.BaseModel):
101    bundle_id: int
102    seller_id: int
103    bundle_name: str
104    description: str
105    total_qty: int
106    price: decimal.Decimal
107    discount_percentage: int
108    window_start: datetime.datetime
109    window_end: datetime.datetime
110    carbon_dioxide: int
111
112
113class AsyncQuerier:
114    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
115        self._conn = conn
116
117    async def create_bundle(self, arg: CreateBundleParams) -> Optional[models.Bundle]:
118        row = (await self._conn.execute(sqlalchemy.text(CREATE_BUNDLE), {
119            "p1": arg.seller_id,
120            "p2": arg.bundle_name,
121            "p3": arg.description,
122            "p4": arg.total_qty,
123            "p5": arg.carbon_dioxide,
124            "p6": arg.price,
125            "p7": arg.discount_percentage,
126            "p8": arg.window_start,
127            "p9": arg.window_end,
128        })).first()
129        if row is None:
130            return None
131        return models.Bundle(
132            bundle_id=row[0],
133            seller_id=row[1],
134            bundle_name=row[2],
135            description=row[3],
136            carbon_dioxide=row[4],
137            total_qty=row[5],
138            price=row[6],
139            discount_percentage=row[7],
140            window_start=row[8],
141            window_end=row[9],
142            created_at=row[10],
143        )
144
145    async def delete_bundle(self, *, bundle_id: int) -> Optional[models.Bundle]:
146        row = (await self._conn.execute(sqlalchemy.text(DELETE_BUNDLE), {"p1": bundle_id})).first()
147        if row is None:
148            return None
149        return models.Bundle(
150            bundle_id=row[0],
151            seller_id=row[1],
152            bundle_name=row[2],
153            description=row[3],
154            carbon_dioxide=row[4],
155            total_qty=row[5],
156            price=row[6],
157            discount_percentage=row[7],
158            window_start=row[8],
159            window_end=row[9],
160            created_at=row[10],
161        )
162
163    async def get_bundle(self, *, bundle_id: int) -> Optional[models.Bundle]:
164        row = (await self._conn.execute(sqlalchemy.text(GET_BUNDLE), {"p1": bundle_id})).first()
165        if row is None:
166            return None
167        return models.Bundle(
168            bundle_id=row[0],
169            seller_id=row[1],
170            bundle_name=row[2],
171            description=row[3],
172            carbon_dioxide=row[4],
173            total_qty=row[5],
174            price=row[6],
175            discount_percentage=row[7],
176            window_start=row[8],
177            window_end=row[9],
178            created_at=row[10],
179        )
180
181    async def get_bundle_lock(self, *, bundle_id: int) -> Optional[models.Bundle]:
182        row = (await self._conn.execute(sqlalchemy.text(GET_BUNDLE_LOCK), {"p1": bundle_id})).first()
183        if row is None:
184            return None
185        return models.Bundle(
186            bundle_id=row[0],
187            seller_id=row[1],
188            bundle_name=row[2],
189            description=row[3],
190            carbon_dioxide=row[4],
191            total_qty=row[5],
192            price=row[6],
193            discount_percentage=row[7],
194            window_start=row[8],
195            window_end=row[9],
196            created_at=row[10],
197        )
198
199    async def get_bundles(self) -> AsyncIterator[models.Bundle]:
200        result = await self._conn.stream(sqlalchemy.text(GET_BUNDLES))
201        async for row in result:
202            yield models.Bundle(
203                bundle_id=row[0],
204                seller_id=row[1],
205                bundle_name=row[2],
206                description=row[3],
207                carbon_dioxide=row[4],
208                total_qty=row[5],
209                price=row[6],
210                discount_percentage=row[7],
211                window_start=row[8],
212                window_end=row[9],
213                created_at=row[10],
214            )
215
216    async def get_sellers_active_bundles(self, *, seller_id: int) -> AsyncIterator[models.Bundle]:
217        result = await self._conn.stream(sqlalchemy.text(GET_SELLERS_ACTIVE_BUNDLES), {"p1": seller_id})
218        async for row in result:
219            yield models.Bundle(
220                bundle_id=row[0],
221                seller_id=row[1],
222                bundle_name=row[2],
223                description=row[3],
224                carbon_dioxide=row[4],
225                total_qty=row[5],
226                price=row[6],
227                discount_percentage=row[7],
228                window_start=row[8],
229                window_end=row[9],
230                created_at=row[10],
231            )
232
233    async def get_sellers_bundle(self, arg: GetSellersBundleParams) -> Optional[models.Bundle]:
234        row = (await self._conn.execute(sqlalchemy.text(GET_SELLERS_BUNDLE), {"p1": arg.seller_id, "p2": arg.bundle_id})).first()
235        if row is None:
236            return None
237        return models.Bundle(
238            bundle_id=row[0],
239            seller_id=row[1],
240            bundle_name=row[2],
241            description=row[3],
242            carbon_dioxide=row[4],
243            total_qty=row[5],
244            price=row[6],
245            discount_percentage=row[7],
246            window_start=row[8],
247            window_end=row[9],
248            created_at=row[10],
249        )
250
251    async def get_sellers_bundles(self, *, seller_id: int) -> AsyncIterator[models.Bundle]:
252        result = await self._conn.stream(sqlalchemy.text(GET_SELLERS_BUNDLES), {"p1": seller_id})
253        async for row in result:
254            yield models.Bundle(
255                bundle_id=row[0],
256                seller_id=row[1],
257                bundle_name=row[2],
258                description=row[3],
259                carbon_dioxide=row[4],
260                total_qty=row[5],
261                price=row[6],
262                discount_percentage=row[7],
263                window_start=row[8],
264                window_end=row[9],
265                created_at=row[10],
266            )
267
268    async def update_bundle(self, arg: UpdateBundleParams) -> Optional[models.Bundle]:
269        row = (await self._conn.execute(sqlalchemy.text(UPDATE_BUNDLE), {
270            "p1": arg.bundle_id,
271            "p2": arg.seller_id,
272            "p3": arg.bundle_name,
273            "p4": arg.description,
274            "p5": arg.total_qty,
275            "p6": arg.price,
276            "p7": arg.discount_percentage,
277            "p8": arg.window_start,
278            "p9": arg.window_end,
279            "p10": arg.carbon_dioxide,
280        })).first()
281        if row is None:
282            return None
283        return models.Bundle(
284            bundle_id=row[0],
285            seller_id=row[1],
286            bundle_name=row[2],
287            description=row[3],
288            carbon_dioxide=row[4],
289            total_qty=row[5],
290            price=row[6],
291            discount_percentage=row[7],
292            window_start=row[8],
293            window_end=row[9],
294            created_at=row[10],
295        )