def category(session: Session, user: User) -> Category:
category = Category.create(session,
name="Guitar Lesson",
color="121212",
user_id=user.id)
yield category
session.delete(category)
session.commit()
async def set_category(request: Request,
name: str = Form(None),
color: str = Form(None),
db_sess: Session = Depends(get_db)):
message = ""
user_id = 1 # until issue#29 will get current user_id from session
color = color.replace('#', '')
if not validate_color_format(color):
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST,
detail=f"Color {color} if not from "
f"expected format.")
try:
Category.create(db_sess, name=name, color=color, user_id=user_id)
except IntegrityError:
db_sess.rollback()
message = "Category already exists"
return templates.TemplateResponse(
"categories.html", dictionary_req(request, message, name, color))
message = f"Congratulation! You have created a new category: {name}"
return templates.TemplateResponse(
"categories.html", dictionary_req(request, message, name, color))
async def set_category(category: CategoryModel,
db_sess: Session = Depends(get_db)) -> Dict[str, Any]:
try:
cat = Category.create(db_sess,
name=category.name,
color=category.color,
user_id=category.user_id)
except IntegrityError:
db_sess.rollback()
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST,
detail=f"category is already exists for "
f"user {category.user_id}.")
else:
return {"category": cat.to_dict()}
def category_create(comp_id):
"""Create categories"""
check_permissions()
form = CategoryForm()
if form.validate_on_submit():
category = Category(name=form.name.data,
body=form.body.data,
comp_id=comp_id)
db.session.add(category)
db.session.commit()
flash('Category created successfully')
return redirect(
url_for('competition.submissions_overview',
comp_id=comp_id,
cat_id=category.id))
return render_template('competition/categoryCreate.html',
title='Create Category',
form=form)