mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-23 08:23:00 +08:00
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import logging
|
|
|
|
from felicity.apps.inventory import models, schemas
|
|
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_stock_units():
|
|
logger.info("Setting up stock units .....")
|
|
data = get_seeds("inventory")
|
|
|
|
stock_units = data.get("units", [])
|
|
|
|
for _unit in stock_units:
|
|
stock_unit = await models.StockUnit.get(name=_unit.get("name"))
|
|
if not stock_unit:
|
|
su_in = schemas.StockUnitCreate(
|
|
name=_unit.get("name"),
|
|
description=_unit.get("description"),
|
|
synonym="",
|
|
)
|
|
await models.StockUnit.create(su_in)
|
|
|
|
|
|
async def seed_stock_categories():
|
|
logger.info("Setting up stock categories .....")
|
|
data = get_seeds("inventory")
|
|
|
|
stock_categories = data.get("categories", [])
|
|
|
|
for _cat in stock_categories:
|
|
stock_category = await models.StockCategory.get(name=_cat.get("name"))
|
|
if not stock_category:
|
|
cu_in = schemas.StockCategoryCreate(
|
|
name=_cat.get("name"),
|
|
description=_cat.get("description"),
|
|
)
|
|
await models.StockCategory.create(cu_in)
|
|
|
|
|
|
async def seed_stock_hazards():
|
|
logger.info("Setting up stock hazards .....")
|
|
data = get_seeds("inventory")
|
|
|
|
hazards = data.get("hazards", [])
|
|
|
|
for _hzd in hazards:
|
|
hazard = await models.Hazard.get(name=_hzd.get("name"))
|
|
if not hazard:
|
|
sh_in = schemas.HazardCreate(
|
|
name=_hzd.get("name"),
|
|
description=_hzd.get("description"),
|
|
)
|
|
await models.Hazard.create(sh_in)
|