feat: generate all transaction equipment for all assetnum data

main
MrWaradana 11 months ago
parent 1ddb931ce1
commit 0b12f52142

@ -9,7 +9,15 @@ from .schema import (
EquipmentCreate,
EquipmentUpdate,
)
from .service import get_master_by_assetnum, get_by_id, get_all, create, update, delete
from .service import (
get_master_by_assetnum,
get_by_id,
get_all,
create,
update,
delete,
generate_all_transaction,
)
from src.database.service import CommonParameters, search_filter_sort_paginate
from src.database.core import DbSession
@ -46,6 +54,18 @@ async def get_equipment_tree():
pass
@router.get(
"/generate-equipment-transaction", response_model=StandardResponse[List[str]]
)
async def get_generated_equipment_transaction(db_session: DbSession, token: Token):
equipment_list = await generate_all_transaction(db_session=db_session, token=token)
return StandardResponse(
data=equipment_list,
message="Equipment transactions generated successfully",
)
@router.get("/{assetnum}", response_model=StandardResponse[EquipmentRead])
async def get_equipment(db_session: DbSession, assetnum: str):
(

@ -103,6 +103,16 @@ async def get_all(
return result
async def generate_all_transaction(*, db_session: DbSession, token):
"""Generate transaction for all equipments in the database based on equipments assetnum."""
query = Select(Equipment)
query_result = await db_session.execute(query)
equipments = query_result.scalars().all()
for equipment in equipments:
await main(equipment.assetnum, token, RELIABILITY_APP_URL)
return [equipment.assetnum for equipment in equipments]
async def generate_transaction(
*, db_session: DbSession, data_in: EquipmentCreate, token
):

Loading…
Cancel
Save