✨ Add base class to simplify CRUD (#23)

This commit is contained in:
Manu
2020-01-19 22:40:50 +01:00
committed by Sebastián Ramírez
parent 1c975c7f2d
commit ab46165387
33 changed files with 322 additions and 283 deletions
@@ -1,6 +1,6 @@
from app import crud
from app.db.session import db_session
from app.models.item import ItemCreate
from app.schemas.item import ItemCreate
from app.tests.utils.user import create_random_user
from app.tests.utils.utils import random_lower_string
@@ -12,6 +12,6 @@ def create_random_item(owner_id: int = None):
title = random_lower_string()
description = random_lower_string()
item_in = ItemCreate(title=title, description=description, id=id)
return crud.item.create(
db_session=db_session, item_in=item_in, owner_id=owner_id
return crud.item.create_with_owner(
db_session=db_session, obj_in=item_in, owner_id=owner_id
)
@@ -3,7 +3,7 @@ import requests
from app import crud
from app.core import config
from app.db.session import db_session
from app.models.user import UserCreate, UserUpdate
from app.schemas.user import UserCreate, UserUpdate
from app.tests.utils.utils import get_server_api, random_lower_string
@@ -21,7 +21,7 @@ def create_random_user():
email = random_lower_string()
password = random_lower_string()
user_in = UserCreate(username=email, email=email, password=password)
user = crud.user.create(db_session=db_session, user_in=user_in)
user = crud.user.create(db_session=db_session, obj_in=user_in)
return user
@@ -35,9 +35,9 @@ def authentication_token_from_email(email):
user = crud.user.get_by_email(db_session, email=email)
if not user:
user_in = UserCreate(username=email, email=email, password=password)
user = crud.user.create(db_session=db_session, user_in=user_in)
user = crud.user.create(db_session=db_session, obj_in=user_in)
else:
user_in = UserUpdate(password=password)
user = crud.user.update(db_session, user=user, user_in=user_in)
user = crud.user.update(db_session, obj_in=user, db_obj=user_in)
return user_authentication_headers(get_server_api(), email, password)