151 lines
5.6 KiB
Python
151 lines
5.6 KiB
Python
from flask import Blueprint, request, jsonify
|
|
from app.models import Department, User, db
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
bp = Blueprint('departments', __name__)
|
|
|
|
@bp.route('', methods=['POST'])
|
|
def create_department():
|
|
data = request.get_json()
|
|
|
|
if not data or 'name' not in data:
|
|
return jsonify({'message': '部门名称是必需的'}), 400
|
|
|
|
name = data['name']
|
|
|
|
# 检查部门名称是否已存在
|
|
if Department.query.filter_by(name=name).first():
|
|
return jsonify({'message': '部门名称已存在'}), 400
|
|
|
|
new_department = Department(name=name)
|
|
db.session.add(new_department)
|
|
db.session.flush() # 这会给 new_department 分配一个 id
|
|
|
|
# 如果提供了其他字段，设置它们
|
|
for field in ['department_code', 'manager', 'phone', 'email', 'location']:
|
|
if field in data:
|
|
new_department.set_detail(field, data[field])
|
|
|
|
try:
|
|
db.session.commit()
|
|
except IntegrityError:
|
|
db.session.rollback()
|
|
return jsonify({'message': '创建部门失败，可能是由于数据完整性问题'}), 500
|
|
|
|
return jsonify({'message': '部门创建成功', 'id': new_department.id}), 201
|
|
|
|
@bp.route('/<int:parent_id>/children', methods=['POST'])
|
|
def create_child_department(parent_id):
|
|
parent_department = Department.query.get(parent_id)
|
|
if not parent_department:
|
|
return jsonify({'message': '父部门不存在'}), 404
|
|
|
|
data = request.get_json()
|
|
|
|
if not data or 'name' not in data:
|
|
return jsonify({'message': '子部门名称是必需的'}), 400
|
|
|
|
name = data['name']
|
|
|
|
# 检查部门名称是否已存在
|
|
if Department.query.filter_by(name=name).first():
|
|
return jsonify({'message': '部门名称已存在'}), 400
|
|
|
|
new_department = Department(name=name, parent_id=parent_id)
|
|
db.session.add(new_department)
|
|
db.session.flush() # 这会给 new_department 分配一个 id
|
|
|
|
# 如果提供了其他字段，设置它们
|
|
for field in ['department_code', 'manager', 'phone', 'email', 'location']:
|
|
if field in data:
|
|
new_department.set_detail(field, data[field])
|
|
|
|
try:
|
|
db.session.commit()
|
|
except IntegrityError:
|
|
db.session.rollback()
|
|
return jsonify({'message': '创建子部门失败，可能是由于数据完整性问题'}), 500
|
|
|
|
return jsonify({'message': '子部门创建成功', 'id': new_department.id}), 201
|
|
|
|
@bp.route('/<int:department_id>/add_user', methods=['POST'])
|
|
def add_user_to_department(department_id):
|
|
data = request.get_json()
|
|
|
|
if not data or 'user_id' not in data:
|
|
return jsonify({'message': '用户ID是必需的'}), 400
|
|
|
|
user_id = data['user_id']
|
|
|
|
department = Department.query.get(department_id)
|
|
if not department:
|
|
return jsonify({'message': '部门不存在'}), 404
|
|
|
|
user = User.query.get(user_id)
|
|
if not user:
|
|
return jsonify({'message': '用户不存在'}), 404
|
|
|
|
if 'is_primary' in data and data['is_primary']:
|
|
# 如果是主要部门，直接设置
|
|
user.primary_department = department
|
|
else:
|
|
# 如果不是主要部门，添加到次要部门
|
|
if department not in user.secondary_departments:
|
|
user.secondary_departments.append(department)
|
|
|
|
try:
|
|
db.session.commit()
|
|
return jsonify({'message': '用户成功添加到部门'}), 200
|
|
except IntegrityError:
|
|
db.session.rollback()
|
|
return jsonify({'message': '添加用户到部门失败，可能是由于数据完整性问题'}), 500
|
|
|
|
@bp.route('', methods=['GET'])
|
|
def get_departments():
|
|
departments = Department.query.all()
|
|
departments_list = [{
|
|
'id': dept.id,
|
|
'name': dept.name,
|
|
'parent_id': dept.parent_id,
|
|
'created_at': dept.created_at.isoformat() if dept.created_at else None,
|
|
'updated_at': dept.updated_at.isoformat() if dept.updated_at else None
|
|
} for dept in departments]
|
|
return jsonify(departments_list), 200
|
|
|
|
@bp.route('/<int:department_id>', methods=['GET'])
|
|
def get_department_details(department_id):
|
|
department = Department.query.get(department_id)
|
|
if not department:
|
|
return jsonify({'message': '部门不存在'}), 404
|
|
|
|
# 获取主要部门的用户
|
|
primary_users = User.query.filter_by(primary_department_id=department.id).all()
|
|
|
|
# 获取次要部门的用户
|
|
secondary_users = department.users.all()
|
|
|
|
# 合并用户列表并去重
|
|
all_users = list(set(primary_users + secondary_users))
|
|
|
|
users = [{'id': u.id, 'username': u.username} for u in all_users]
|
|
children = [{'id': c.id, 'name': c.name} for c in department.children]
|
|
|
|
department_details = {
|
|
'id': department.id,
|
|
'name': department.name,
|
|
'parent_id': department.parent_id,
|
|
'created_at': department.created_at.isoformat() if department.created_at else None,
|
|
'updated_at': department.updated_at.isoformat() if department.updated_at else None,
|
|
'department_code': department.get_detail('department_code'),
|
|
'manager': department.get_detail('manager'),
|
|
'phone': department.get_detail('phone'),
|
|
'email': department.get_detail('email'),
|
|
'location': department.get_detail('location'),
|
|
'users': users,
|
|
'children': children
|
|
}
|
|
|
|
return jsonify(department_details), 200
|
|
|
|
# 你可以在这里添加更多的部门相关路由...
|