401 lines
15 KiB
Python
401 lines
15 KiB
Python
from datetime import datetime
|
||
import pytz
|
||
from app.extensions import db
|
||
from app.models.borrow import BorrowApproval
|
||
from app.models.system import SysUser
|
||
|
||
|
||
class BorrowApprovalService:
|
||
"""借库审批服务"""
|
||
|
||
@staticmethod
|
||
def generate_request_no():
|
||
"""
|
||
生成审批单号: APR-BOR-yyyyMMdd-HHmm-当日流水(4位)
|
||
"""
|
||
beijing_tz = pytz.timezone('Asia/Shanghai')
|
||
now = datetime.now(beijing_tz)
|
||
|
||
date_str = now.strftime('%Y%m%d')
|
||
time_str = now.strftime('%H%M')
|
||
|
||
prefix = f"APR-BOR-{date_str}-"
|
||
|
||
latest = db.session.query(BorrowApproval.request_no).filter(
|
||
BorrowApproval.request_no.like(f"{prefix}%")
|
||
).order_by(BorrowApproval.id.desc()).first()
|
||
|
||
if latest:
|
||
last_seq = int(latest[0].split('-')[-1])
|
||
sequence = last_seq + 1
|
||
else:
|
||
sequence = 1
|
||
|
||
return f"APR-BOR-{date_str}-{time_str}-{sequence:04d}"
|
||
|
||
@staticmethod
|
||
def submit_approval(applicant_id, items, allowed_approvers, remark=None, approver_id=None,
|
||
borrower_name=None):
|
||
"""
|
||
提交借库申请（仅存储意向，不扣库存）
|
||
|
||
Args:
|
||
applicant_id: 申请人ID
|
||
items: 借库物品明细列表，每个物品应包含:
|
||
- name: 物料名称 (必填)
|
||
- spec_model: 规格型号 (必填)
|
||
- quantity: 计划借库数量 (必填)
|
||
- warehouse_location: 库位 (可选)
|
||
- remark: 物品备注 (可选)
|
||
allowed_approvers: 允许审批的人员/角色列表
|
||
approver_id: 指定审批人ID（可选）
|
||
remark: 申请说明
|
||
borrower_name: 借库人姓名（必填）
|
||
|
||
Returns:
|
||
BorrowApproval 实例
|
||
|
||
Raises:
|
||
ValueError: 当 items 为空或缺少必填字段时抛出
|
||
"""
|
||
if not items:
|
||
raise ValueError("借库物品列表不能为空")
|
||
|
||
required_fields = ['name', 'spec_model', 'quantity']
|
||
for idx, item in enumerate(items):
|
||
missing_fields = [f for f in required_fields if f not in item or str(item.get(f) or '').strip() == '']
|
||
if missing_fields:
|
||
raise ValueError(
|
||
f"第 {idx + 1} 条物品缺少必填字段: {', '.join(missing_fields)}。"
|
||
f"必须包含: name, spec_model, quantity"
|
||
)
|
||
try:
|
||
qty = float(item.get('quantity', 0))
|
||
if qty <= 0:
|
||
raise ValueError(f"第 {idx + 1} 条物品的借库数量必须大于0")
|
||
except (TypeError, ValueError) as e:
|
||
raise ValueError(f"第 {idx + 1} 条物品的 quantity 格式无效: {str(e)}")
|
||
|
||
if not allowed_approvers:
|
||
raise ValueError("必须指定至少一位审批人")
|
||
|
||
if approver_id:
|
||
allowed_approvers = [{"type": "user", "value": int(approver_id)}]
|
||
|
||
request_no = BorrowApprovalService.generate_request_no()
|
||
|
||
approval = BorrowApproval(
|
||
request_no=request_no,
|
||
applicant_id=applicant_id,
|
||
remark=remark,
|
||
borrower_name=borrower_name,
|
||
status=0, # 待审批
|
||
)
|
||
|
||
approval.set_items(items)
|
||
approval.set_allowed_approvers(allowed_approvers)
|
||
|
||
db.session.add(approval)
|
||
db.session.commit()
|
||
|
||
# ★ 创建成功后，发送邮件通知审批人（静默处理，不阻断主流程）
|
||
BorrowApprovalService._notify_new_request(approval, applicant_id, approver_id=approver_id)
|
||
|
||
return approval
|
||
|
||
@staticmethod
|
||
def _get_emails_by_identifiers(applicant_id=None, role_codes=None):
|
||
"""
|
||
根据用户ID或角色列表查询邮箱地址
|
||
|
||
Args:
|
||
applicant_id: 用户ID (按 SysUser.id 查找)
|
||
role_codes: 角色代码列表，如 ['SUPERVISOR', 'SUPER_ADMIN']
|
||
|
||
Returns:
|
||
去重后的邮箱地址列表
|
||
"""
|
||
emails = []
|
||
|
||
if applicant_id:
|
||
user = SysUser.query.get(int(applicant_id))
|
||
if user and user.email:
|
||
emails.append(user.email)
|
||
|
||
if role_codes:
|
||
for code in role_codes:
|
||
users = SysUser.query.filter_by(role=code).all()
|
||
for u in users:
|
||
if u.email:
|
||
emails.append(u.email)
|
||
|
||
return list(set(emails))
|
||
|
||
@staticmethod
|
||
def _notify_new_request(approval, applicant_id, approver_id=None):
|
||
"""发送新借库申请通知邮件给审批人和申请人（静默处理，不阻断主流程）"""
|
||
try:
|
||
from flask import current_app
|
||
from app.utils.email_service import send_borrow_new_request_notify
|
||
from app.models.system import SysUser
|
||
|
||
applicant_name = ''
|
||
applicant_emails = []
|
||
|
||
# 1. 收集申请人信息
|
||
if applicant_id:
|
||
user = SysUser.query.get(int(applicant_id))
|
||
if user and user.email:
|
||
applicant_emails.append(user.email)
|
||
applicant_name = str(user.username).split('/')[0] if '/' in (user.username or '') else (user.username or str(applicant_id))
|
||
|
||
# 2. 收集审批人信息
|
||
approver_emails = []
|
||
if approver_id:
|
||
user = SysUser.query.get(int(approver_id))
|
||
if user and user.email:
|
||
approver_emails.append(user.email)
|
||
else:
|
||
# 兜底：按角色查询
|
||
approvers = approval.get_allowed_approvers()
|
||
role_codes = []
|
||
for a in approvers:
|
||
if a.get('type') == 'role':
|
||
role_codes.append(a.get('value', ''))
|
||
approver_emails = BorrowApprovalService._get_emails_by_identifiers(role_codes=role_codes)
|
||
|
||
# 去重
|
||
all_emails = list(set(applicant_emails + approver_emails))
|
||
if not all_emails:
|
||
current_app.logger.info(f"[Email] 借库审批单 {approval.request_no} 无收件人邮箱，跳过通知")
|
||
return
|
||
|
||
# 3. 获取物料明细
|
||
items = approval.get_items()
|
||
|
||
# 4. 分别发送邮件
|
||
if applicant_emails:
|
||
try:
|
||
send_borrow_new_request_notify(
|
||
to_emails=applicant_emails,
|
||
request_no=approval.request_no,
|
||
applicant_name=applicant_name,
|
||
remark=f"您的借库申请已提交，等待审批。{approval.remark or ''}",
|
||
items=items,
|
||
is_applicant_notify=True
|
||
)
|
||
except Exception as e:
|
||
current_app.logger.error(f"[Email] 通知申请人失败: {e}")
|
||
|
||
if approver_emails:
|
||
try:
|
||
send_borrow_new_request_notify(
|
||
to_emails=approver_emails,
|
||
request_no=approval.request_no,
|
||
applicant_name=applicant_name,
|
||
remark=approval.remark or '',
|
||
items=items,
|
||
is_applicant_notify=False
|
||
)
|
||
except Exception as e:
|
||
current_app.logger.error(f"[Email] 通知审批人失败: {e}")
|
||
|
||
except Exception as e:
|
||
try:
|
||
from flask import current_app
|
||
current_app.logger.error(f"[Email] 发送新借库申请通知邮件失败: {e}")
|
||
except RuntimeError:
|
||
import traceback
|
||
traceback.print_exc()
|
||
|
||
@staticmethod
|
||
def _notify_approval_result(approval, approver_id, action):
|
||
"""发送借库审批结果通知邮件（静默处理，不阻断主流程）"""
|
||
import logging
|
||
logger = logging.getLogger(__name__)
|
||
|
||
try:
|
||
from app.utils.email_service import send_borrow_approval_result_notify, send_borrow_dispatch_notify
|
||
from app.models.system import SysUser as SU
|
||
|
||
# 1. 提取申请人信息
|
||
applicant_name = ''
|
||
applicant_emails = []
|
||
if approval.applicant_id:
|
||
user = SU.query.get(approval.applicant_id)
|
||
if user:
|
||
applicant_name = str(user.username).split('/')[0] if '/' in (user.username or '') else (user.username or '')
|
||
if user.email:
|
||
applicant_emails.append(user.email)
|
||
|
||
# 2. 提取物料明细
|
||
items = approval.get_items() if approval else []
|
||
|
||
# 3. 分支逻辑
|
||
if action == 'approve':
|
||
# 3.1 通知申请人（审批已通过，明确告知结果）
|
||
if applicant_emails:
|
||
try:
|
||
send_borrow_approval_result_notify(
|
||
to_emails=applicant_emails,
|
||
request_no=approval.request_no,
|
||
is_passed=True,
|
||
reject_reason='',
|
||
applicant_name=applicant_name
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"[Email] 通知申请人（通过）失败: {e}")
|
||
else:
|
||
logger.warning("[Email] 申请人无邮箱，无法发送审批通过通知")
|
||
|
||
# 3.2 通知库管（请备货）
|
||
warehouse_role_codes = ['WAREHOUSE_MGR', 'OUTBOUND']
|
||
warehouse_emails = BorrowApprovalService._get_emails_by_identifiers(role_codes=warehouse_role_codes)
|
||
|
||
if warehouse_emails:
|
||
try:
|
||
send_borrow_dispatch_notify(
|
||
to_emails=warehouse_emails,
|
||
request_no=approval.request_no,
|
||
applicant_name=applicant_name,
|
||
items=items
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"[Email] 通知库管失败: {e}")
|
||
else:
|
||
logger.warning("[Email] 无库管角色邮箱，无法发送备货通知")
|
||
|
||
elif action == 'reject':
|
||
# 3.3 通知申请人（已驳回）
|
||
if applicant_emails:
|
||
try:
|
||
send_borrow_approval_result_notify(
|
||
to_emails=applicant_emails,
|
||
request_no=approval.request_no,
|
||
is_passed=False,
|
||
reject_reason=approval.reject_reason or '未说明原因',
|
||
applicant_name=applicant_name
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"[Email] 通知申请人驳回失败: {e}")
|
||
else:
|
||
logger.warning("[Email] 申请人无邮箱，无法发送驳回通知")
|
||
|
||
except Exception as e:
|
||
import traceback
|
||
traceback.print_exc()
|
||
logger.error(f"[Email] 外层发送异常: {e}")
|
||
|
||
@staticmethod
|
||
def can_approve(approval, user_id, user_role):
|
||
"""
|
||
检查用户是否有权限审批
|
||
"""
|
||
approvers = approval.get_allowed_approvers()
|
||
|
||
if user_role and user_role.upper() == 'SUPER_ADMIN':
|
||
return True
|
||
|
||
for approver in approvers:
|
||
approver_type = approver.get('type', '')
|
||
approver_value = approver.get('value', '')
|
||
|
||
if approver_type == 'user' and str(approver_value) == str(user_id):
|
||
return True
|
||
|
||
if approver_type == 'role' and approver_value == user_role:
|
||
return True
|
||
|
||
return False
|
||
|
||
@staticmethod
|
||
def approve(request_id, user_id, user_role, action='approve', reject_reason=None):
|
||
"""
|
||
执行审批操作
|
||
|
||
Returns:
|
||
(success: bool, message: str, approval: BorrowApproval or None)
|
||
"""
|
||
beijing_tz = pytz.timezone('Asia/Shanghai')
|
||
current_time = datetime.now(beijing_tz).replace(tzinfo=None)
|
||
|
||
approval = BorrowApproval.query.get(request_id)
|
||
if not approval:
|
||
return False, "审批单不存在", None
|
||
|
||
if approval.status != 0:
|
||
return False, f"审批单状态已更新，无法重复审批 (当前状态: {approval.status})", None
|
||
|
||
if not BorrowApprovalService.can_approve(approval, user_id, user_role):
|
||
return False, "您没有审批此单的权限", None
|
||
|
||
try:
|
||
if action == 'approve':
|
||
approval.status = 1 # 已通过
|
||
approval.actual_approver_id = user_id
|
||
approval.approved_at = current_time
|
||
elif action == 'reject':
|
||
approval.status = 2 # 已驳回
|
||
approval.reject_reason = reject_reason
|
||
else:
|
||
return False, "无效的审批操作", None
|
||
|
||
db.session.commit()
|
||
|
||
# ★ 审批后，发送邮件通知（静默处理，不阻断主流程）
|
||
BorrowApprovalService._notify_approval_result(approval, user_id, action)
|
||
|
||
return True, "审批成功", approval
|
||
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
return False, f"审批失败: {str(e)}", None
|
||
|
||
@staticmethod
|
||
def get_request_list(page=1, per_page=10, applicant_id=None, status=None):
|
||
"""
|
||
获取审批单列表
|
||
"""
|
||
from sqlalchemy import desc
|
||
|
||
query = BorrowApproval.query
|
||
|
||
if applicant_id:
|
||
query = query.filter(BorrowApproval.applicant_id == applicant_id)
|
||
|
||
if status is not None:
|
||
query = query.filter(BorrowApproval.status == status)
|
||
|
||
query = query.order_by(desc(BorrowApproval.created_at))
|
||
|
||
pagination = query.paginate(page=page, per_page=per_page, error_out=False)
|
||
|
||
return {
|
||
'items': [item.to_dict() for item in pagination.items],
|
||
'total': pagination.total,
|
||
'pages': pagination.pages,
|
||
'current_page': page
|
||
}
|
||
|
||
@staticmethod
|
||
def get_request_by_id(request_id):
|
||
"""根据ID获取审批单"""
|
||
return BorrowApproval.query.get(request_id)
|
||
|
||
@staticmethod
|
||
def mark_completed(request_id):
|
||
"""标记审批单为已完成（借库执行完成后调用）"""
|
||
approval = BorrowApproval.query.get(request_id)
|
||
if not approval:
|
||
return False, "审批单不存在", None
|
||
|
||
if approval.status != 1:
|
||
return False, f"只有已通过的审批单才能标记为完成 (当前状态: {approval.status})", None
|
||
|
||
try:
|
||
approval.status = 3 # 已完成
|
||
db.session.commit()
|
||
return True, "审批单已完成", approval
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
return False, f"操作失败: {str(e)}", None |