mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-23 16:33:11 +08:00
52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
import logging
|
|
|
|
from felicity.apps.instrument import schemas
|
|
from felicity.apps.instrument.services import (InstrumentService,
|
|
InstrumentTypeService,
|
|
LaboratoryInstrumentService,
|
|
MethodService)
|
|
from felicity.core.config import get_settings
|
|
|
|
from .data import get_seeds
|
|
|
|
settings = get_settings()
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def seed_instrument_categories() -> None:
|
|
logger.info("Setting up instrument categories .....")
|
|
instrument_type_service = InstrumentTypeService()
|
|
instrument_service = InstrumentService()
|
|
method_service = MethodService()
|
|
laboratory_instrument_service = LaboratoryInstrumentService()
|
|
|
|
data = get_seeds("instrument")
|
|
|
|
for inst_type in data.get("categories"):
|
|
instrument_type = await instrument_type_service.get(name=inst_type)
|
|
if not instrument_type:
|
|
inst_type_in = schemas.InstrumentTypeCreate(
|
|
name=inst_type, description=inst_type
|
|
)
|
|
await instrument_type_service.create(inst_type_in)
|
|
|
|
for _meth in data.get("methods"):
|
|
if not (await method_service.get(name=_meth)):
|
|
method_in = schemas.MethodCreate(name=_meth, description="", keyword="")
|
|
await method_service.create(method_in)
|
|
|
|
for _inst in data.get("instruments"):
|
|
if not (await instrument_service.get(name=_inst)):
|
|
inst_in = schemas.InstrumentCreate(name=_inst, description="", keyword="")
|
|
instrument = await instrument_service.create(inst_in)
|
|
|
|
if _inst == "Manual":
|
|
if not (await laboratory_instrument_service.get(lab_name=_inst)):
|
|
lab_inst_in = schemas.LaboratoryInstrumentCreate(
|
|
instrument_uid=instrument.uid,
|
|
lab_name=_inst,
|
|
serial_number="Man-001",
|
|
)
|
|
await laboratory_instrument_service.create(lab_inst_in)
|