def add_job():
form = AddJobForm()
if form.validate_on_submit():
job = Jobs()
job.job = form.job.data
job.work_size = form.work_size.data
job.collaborators = form.collaborators.data
job.is_finished = form.is_finished.data
job.team_leader = form.team_leader.data
job.user = current_user
session = db_session.create_session()
job = session.merge(job)
session.add(job)
session.commit()
return redirect('')
return render_template('add_job.html', title='Добавление работы', form=form)
def add_job():
form = JobsForm()
if request.method == 'POST' and form.validate_on_submit():
s = db_session.create_session()
job = Jobs()
job.team_leader = form.team_leader.data
job.user = s.query(User).filter(User.id == job.team_leader).first()
job.job = form.job.data
job.work_size = form.work_size.data
job.collaborators = form.collaborators.data
job.start_date = form.start_date.data
job.end_date = form.end_date.data
job.is_finished = form.is_finished.data
job.creator = current_user.id
s.add(job)
s.commit()
return redirect('/page')
return render_template('add_job.html', title='Добавление работы',
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'})
