Added custom extra fields to backend
No integration tests yet, TBD
This commit is contained in:
@@ -18,12 +18,14 @@ async def create(
|
||||
db: AsyncSession,
|
||||
name: Optional[str] = None,
|
||||
comment: Optional[str] = None,
|
||||
extra: Optional[dict[str, str]] = None,
|
||||
) -> models.Vendor:
|
||||
"""Add a new vendor to the database."""
|
||||
vendor = models.Vendor(
|
||||
name=name,
|
||||
registered=datetime.utcnow().replace(microsecond=0),
|
||||
comment=comment,
|
||||
extra=[models.VendorField(key=k, value=v) for k, v in (extra or {}).items()],
|
||||
)
|
||||
db.add(vendor)
|
||||
await db.commit()
|
||||
@@ -72,7 +74,7 @@ async def find(
|
||||
stmt = stmt.order_by(field.desc())
|
||||
|
||||
rows = await db.execute(stmt)
|
||||
result = list(rows.scalars().all())
|
||||
result = list(rows.unique().scalars().all())
|
||||
if total_count is None:
|
||||
total_count = len(result)
|
||||
|
||||
@@ -88,7 +90,10 @@ async def update(
|
||||
"""Update the fields of a vendor object."""
|
||||
vendor = await get_by_id(db, vendor_id)
|
||||
for k, v in data.items():
|
||||
setattr(vendor, k, v)
|
||||
if k == "extra":
|
||||
vendor.extra = [models.VendorField(key=k, value=v) for k, v in v.items()]
|
||||
else:
|
||||
setattr(vendor, k, v)
|
||||
await db.commit()
|
||||
await vendor_changed(vendor, EventType.UPDATED)
|
||||
return vendor
|
||||
|
||||
Reference in New Issue
Block a user