Add script/seed.py for convenience

This commit is contained in:
richard-dds
2018-08-07 21:32:02 -04:00
parent dcb45c64e8
commit f80668c638
2 changed files with 37 additions and 8 deletions

25
script/seed.py Normal file
View File

@@ -0,0 +1,25 @@
# Add root project dir to the python path
import os
import sys
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.append(parent_dir)
from atst.app import make_config, make_app
from atst.domain.users import Users
from atst.domain.requests import Requests
from atst.routes.dev import _DEV_USERS as DEV_USERS
def seed_db():
users = [Users.create(**dev_user) for (_, dev_user) in DEV_USERS.items()]
for user in users:
[Requests.create(user, {}) for _ in range(5)]
if __name__ == "__main__":
config = make_config()
app = make_app(config)
with app.app_context():
seed_db()