from account.models import Account from account.schemas import AccountCreate def inject_fixtures(session): for f in fixtures: if f['parent_path']: parent = Account.get_by_path(session, f['parent_path']) f['parent_account_id'] = parent.id else: f['parent_account_id'] = None del f['parent_path'] schema = AccountCreate(**f) Account.create(schema) fixtures = [{ "name": "Current Assets", "parent_path": None, "type": "Asset" },{ "name": "Cash in Wallet", "parent_path": "/Accounts/Asset/", "type": "Asset", },{ "name": "Checking Account", "parent_path": "/Accounts/Asset/", "type": "Asset", },{ "name": "Savings Account", "parent_path": "/Accounts/Asset/", "type": "Asset", }, { "name": "Debt Accounts", "parent_path": None, "type": "Liability", }, { "name": "Credit Card", "parent_path": "/Accounts/Liability/", "type": "Liability", }, { "name": "Salary", "parent_path": None, "type": "Income", }, { "name": "Other Income", "parent_path": None, "type": "Income", }, { "name": "Auto", "parent_path": None, "type": "Expense", }, { "name": "Home", "parent_path": None, "type": "Expense", }, { "name": "Rent", "parent_path": "/Categories/Expense/Home", "type": "Expense", }, { "name": "Electricity", "parent_path": "/Categories/Expense/Home", "type": "Expense", }, { "name": "Entertainment", "parent_path": None, "type": "Expense", }, { "name": "Groceries", "parent_path": None, "type": "Expense", }, ] """ """