def assign_category(club_id):
club = Club.query.filter_by(id=club_id).first()
if not club:
return json.dumps({'success': False, 'error': 'Club not found'}), 404
post_body = json.loads(request.data)
category = Category.query.filter_by(name=post_body.get('name')).first()
if not category:
category = Category(name=post_body.get('name', ''))
club.categories.append(category)
db.session.add(category)
db.session.commit()
return json.dumps({'success': True, 'data': category.serialize()}), 200
def create_category(budget_id):
budget = Budget.query.filter_by(id=budget_id).first()
if budget is not None:
category_body = json.loads(request.data)
category_name = category_body.get('name')
category = Category.query.filter_by(budget_id=budget.id, name=category_name).first()
if category is None:
category = Category(
name = category_name,
total = 0,
budget_id = budget.id
)
budget.categories.append(category)
db.session.add(category)
db.session.commit()
return json.dumps({'success': True, 'data': category.serialize()}), 201
return json.dumps({'success': False, 'error': 'Category already exists!'}), 201
return json.dumps({'success': False, 'error': 'Budget not found!'}), 404
def assign_category(user_id, trip_id, event_id):
user = User.query.filter_by(id=user_id).first()
if not user:
return json.dumps({'success': False, 'error': 'User not found!'}), 404
trip = Trip.query.filter_by(id=trip_id).first()
if not trip:
return json.dumps({'success': False, 'error': 'Trip not found!'}), 404
if ((trip in user.trips) == False):
return json.dumps({'success': False, 'error': 'Trip not found!'}), 404
event = Event.query.filter_by(id=event_id).first()
if not event:
return json.dumps({'success': False, 'error': 'Event not found!'}), 404
if ((event in trip.events) == False):
return json.dumps({'success': False, 'error': 'Event not found!'}), 404
post_body = json.loads(request.data)
category = Category.query.filter_by(name=post_body.get('name', '')).first()
if not category:
category = Category(name=post_body.get('name', ''))
event.categories.append(category)
db.session.add(category)
db.session.commit()
return json.dumps({'success': True, 'data': category.serialize()}), 200
def create_category(body):
category = Category(description=body.get("description"))
db.session.add(category)
db.session.commit()
return category.serialize()
def create_category(name):
new_category = Category(name=name)
db.session.add(new_category)
db.session.commit()
return new_category.serialize()
