def favorite_creator(num_favorites):
for i in range(num_favorites):
Favorite.create(
uuid=fake.uuid4(),
item=Item.select().order_by(fn.Random()).get(),
user=User.select().order_by(fn.Random()).get(),
)
def add_favorite(user, item, id=None):
"""Link the favorite item to user."""
return Favorite.create(
uuid=id or uuid.uuid4(),
item=item,
user=user,
)
def create_favorite():
try:
# Must send media id with payload when favorite button clicked.
# The following assumes payload includes media_id and timestamp.
payload = request.form.to_dict()
# id of currently logged-in user is foreign key for favorite
current_user_id = model_to_dict(current_user)['id']
payload['user_id'] = current_user_id
favorite = Favorite.create(**payload)
favorite_dict = model_to_dict(favorite)
except:
return jsonify(data={}, status={"code": 401,
"message": "Error: Resource not created"})
return jsonify(data=favorite_dict, status={"code": 200,
"message": "Resource added"})
