initial_data.py

import os
from ...scripts.initial_data import InitialData

from ..models import (
    SipkdBase as ModuleBase,
    SipkdDBSession as ModuleDBSession,
    CommonModel,
    )

from .data.routes import RouteData
from .data.pegawai import PegawaiData
from .data.daftunit import DaftUnitData
#from .data.sp2d import Sp2dData
#from .data.sp2d_potongan import Sp2dPotonganData

fixtures = [
    ('routes', RouteData),
    ('pegawai', PegawaiData),
    ('daftunit', DaftUnitData)
    ]
    
def insert():
    initialData = InitialData(ModuleBase, ModuleDBSession)
    base_dir = os.path.split(__file__)[0]
    initialData.insert(base_dir, fixtures)