Пример #1
0
def generate_items():
    items = [
        ItemModel(user_id=1,
                  category_id=1,
                  name="lamp",
                  description="This is a lamp, it emit light"),
        ItemModel(user_id=1,
                  category_id=3,
                  name="macbookpro",
                  description="Expensive shit"),
        ItemModel(user_id=1,
                  category_id=3,
                  name="dellxps15",
                  description="Less Expensive shit"),
        ItemModel(user_id=1,
                  category_id=3,
                  name="dellmonitor",
                  description="Good stuff"),
        ItemModel(user_id=2,
                  category_id=2,
                  name="toiletpaper",
                  description="We all need this"),
    ]
    with app.app_context():
        db.session.add_all(items)
        db.session.commit()
        return ItemSchema(many=True).dump(items)
Пример #2
0
def generate_categories():
    categories = [
        CategoryModel(name="household"),
        CategoryModel(name="groceries"),
        CategoryModel(name="tech")
    ]
    with app.app_context():
        db.session.add_all(categories)
        db.session.commit()
        return CategorySchema(many=True).dump(categories)
Пример #3
0
def generate_users():
    users = [{
        "id": 1,
        "username": "******",
        "password": "******"
    }, {
        "id": 2,
        "username": "******",
        "password": "******"
    }]
    with app.app_context():
        for user in users:
            new_user = UserModel(username=user["username"],
                                 password=generate_hashed_password(
                                     user["password"]))
            db.session.add(new_user)
            db.session.commit()
    return UserSchema(many=True).dump(users)
Пример #4
0
def session():
    # Create app context
    app.app_context().push()

    yield app
Пример #5
0
def reset_db():
    with app.app_context():
        db.drop_all()
        db.create_all()
        create_db_samples()
def create_tables():
    with app.app_context():
        db.create_all()