def post(self):
parser = reqparse.RequestParser()
parser.add_argument('name', type=str)
parser.add_argument('head_id', type=int)
parser.add_argument('head_salary', type=int)
args = parser.parse_args()
dept = Department.query.get(args['name'])
if dept is not None:
abort(409)
dept = Department(name=args['name'])
emp = None
if args['head_id'] is not None:
emp = Employee.query.get(args['head_id'])
if emp is None:
abort(400)
head = Head(department_name=dept.name,
head_id=args['head_id'])
db.session.add(dept)
db.session.add(head)
if emp is not None:
emp.salary = args['head_salary']
emp.department = dept.name
db.session.commit()
return args['name'], 201
def post(self):
title = request.json.get('title')
print(title)
department = Department(title=title)
db.session.add(department)
db.session.commit()
return jsonify({'department': title})
def add_department():
"""
Add a department to the database
"""
# Throws a 403 Forbidden error if a non-admin user attempts to access these views.
check_admin()
add_department = True
form = DepartmentForm()
if form.validate_on_submit():
department = Department(name=form.name.data)
try:
# add a department to the database
db.session.add(department)
db.session.commit()
logger.info(f'Department {department.name} has been added')
flash('The new department have been successfully added !')
except exc.IntegrityError:
logger.exception('Add department exc.IntegrityError')
flash('Error: department name already exists.')
return redirect(url_for('admin.list_departments'))
return render_template('admin/departments/department.html',
action='Add',
add_department=add_department,
form=form,
title='Add Department')
def add_department(title):
"""
Adds department to the database db
:param title: title of department to add
:return: id of added department
"""
department = Department(title=title)
db.session.add(department)
db.session.commit()
return department.id_dept
def api_add_department():
"""Add a new department."""
# Get name from json.
name = request.json["name"]
# Create department with name from json.
new_department = Department(name=name)
db.session.add(new_department)
db.session.commit()
return department_schema.jsonify(new_department)
def add_department():
"""Add a new department using a form."""
form = DepartmentForm()
if form.validate_on_submit():
# Set department name to a value from the form.
department = Department(name=form.name.data)
db.session.add(department)
db.session.commit()
flash("Department has been added!", "success")
return redirect(url_for("dep.show_departments"))
return render_template("departament_add.html",
title="Add new department",
form=form,
legend="New Department")
def test_department():
d = Department(title='xxxx')
assert d.title == 'xxxx'
def run(self):
depts = [Department(name='IT'),
Department(name='Sales'),
Department(name='Research'),
Department(name='PR')]
for dept in depts:
db.session.add(dept)
db.session.commit()
employees = [
Employee(first_name='Siusan',
last_name='Turner',
department='PR',
salary=3340,
birth_date=date(1995, 4, 7)
),
Employee(first_name='Andrew',
last_name='Miles',
department='IT',
salary=1700,
birth_date=date(1989, 1, 17)
),
Employee(first_name='Oleg',
last_name='Sidorov',
department='IT',
salary=1600,
birth_date=date(1990, 7, 13)
),
Employee(first_name='Brian',
last_name='Fox',
department='Research',
salary=3500,
birth_date=date(1990, 1, 6)
),
Employee(first_name='Steve',
last_name='Johnson',
birth_date=date(1999, 9, 9)
),
Employee(first_name='Martin',
last_name='Black',
birth_date=date(1988, 8, 18)
),
Employee(first_name='Alex',
last_name='Jones',
department='PR',
salary=3400,
birth_date=date(1975, 3, 17)
),
Employee(first_name='Michael',
last_name='Payne',
department='IT',
salary=3200,
birth_date=date(1980, 7, 25)
),
Employee(first_name='Jonathan',
last_name='Miles',
department='Research',
salary=1100,
birth_date=date(1989, 9, 17)
),
Employee(first_name='Stefanie',
last_name='Miller',
department='Research',
salary=1300,
birth_date=date(1990, 5, 12)
),
Employee(first_name='Taras',
last_name='Prokopenko',
department='Sales',
salary=1000,
birth_date=date(1990, 4, 2)
)
]
for employee in employees:
db.session.add(employee)
db.session.commit()
heads = [
Head(department_name='IT', head_id=8),
Head(department_name='PR', head_id=1),
Head(department_name='Research', head_id=4),
Head(department_name='Sales')
]
for head in heads:
db.session.add(head)
db.session.commit()