def add_jobs():
form = JobsForm()
if form.validate_on_submit():
session = db_session.create_session()
jobs = Jobs()
jobs.job = form.title.data
jobs.team_leader = form.team_leader.data
jobs.work_size = form.work_size.data
jobs.collaborators = form.collaborators.data
jobs.is_finished = form.is_finished.data
jobs.hazard_category = form.hazard_category.data
current_user.jobs.append(jobs)
session.merge(current_user)
session.commit()
return redirect('/')
return render_template('job.html', title='Add job', form=form)
def post(self):
args = parser.parse_args()
session = db_session.create_session()
if session.query(Jobs).get(args['id']):
abort(400, message='Id already exists')
abort_if_user_not_found(args['team_leader'])
jobs = Jobs()
jobs.user = session.query(User).get(args['team_leader'])
jobs.id = args['id']
jobs.team_leader = args['team_leader']
jobs.job = args['job']
jobs.work_size = args['work_size']
jobs.collaborators = args['collaborators']
jobs.speciality = args['speciality']
jobs.hazard_category = args['hazard_category']
jobs.is_finished = args['is_finished']
session.add(jobs)
session.commit()
return jsonify({'success': 'OK'})
