327 lines
15 KiB
Python
327 lines
15 KiB
Python
"""充值管理、充值优惠活动、API路由"""
|
|
from datetime import datetime
|
|
from flask import render_template, request, redirect, url_for, flash, session, jsonify
|
|
from models import (db, Student, Course, StudentAccount, RechargeActivity,
|
|
RechargeRecord)
|
|
from utils import (
|
|
login_required, log_operation, permission_required, export_excel,
|
|
filter_students_query, is_teacher_role,
|
|
)
|
|
import business_rules as rules
|
|
|
|
|
|
def register_routes(app):
|
|
|
|
@app.route('/recharge_activities', endpoint='recharge_activity_list')
|
|
@login_required
|
|
@permission_required('recharge_activity_view')
|
|
def recharge_activity_list():
|
|
keyword = request.args.get('keyword', '')
|
|
query = RechargeActivity.query
|
|
if keyword:
|
|
query = query.filter(RechargeActivity.name.contains(keyword))
|
|
activities = query.order_by(RechargeActivity.id.desc()).all()
|
|
return render_template('recharge_activity_list.html', activities=activities, keyword=keyword)
|
|
|
|
@app.route('/recharge_activities/add', methods=['GET', 'POST'], endpoint='recharge_activity_add')
|
|
@login_required
|
|
@permission_required('recharge_activity_add')
|
|
def recharge_activity_add():
|
|
if request.method == 'POST':
|
|
defaults = rules.recharge_activity_defaults()
|
|
activity = RechargeActivity(
|
|
name=request.form.get('name'),
|
|
trigger_amount=float(request.form.get('trigger_amount', 0)),
|
|
gifted_hours=float(request.form.get('gifted_hours', 0)),
|
|
gift_type=defaults['gift_type'],
|
|
is_cumulative=defaults['is_cumulative'],
|
|
cumulative_window=defaults['cumulative_window'],
|
|
cumulative_clear=defaults['cumulative_clear'],
|
|
allow_multi_trigger=defaults['allow_multi_trigger'],
|
|
is_time_limited=int(request.form.get('is_time_limited', 0)),
|
|
start_date=_parse_date(request.form.get('start_date')),
|
|
end_date=_parse_date(request.form.get('end_date')),
|
|
target_audience=defaults['target_audience'],
|
|
can_stack=int(request.form.get('can_stack', 1)),
|
|
gift_cap=defaults['gift_cap'],
|
|
group_discount=defaults['group_discount'],
|
|
group_min_people=defaults['group_min_people'],
|
|
group_extra_hours=defaults['group_extra_hours'],
|
|
referral_reward=defaults['referral_reward'],
|
|
referral_reward_hours=defaults['referral_reward_hours'],
|
|
reward_timing='immediate',
|
|
status=1,
|
|
)
|
|
db.session.add(activity)
|
|
db.session.commit()
|
|
log_operation('创建充值活动', f'活动:{activity.name}')
|
|
flash('充值活动创建成功', 'success')
|
|
return redirect(url_for('recharge_activity_list'))
|
|
|
|
return render_template('recharge_activity_form.html', activity=None)
|
|
|
|
@app.route('/recharge_activities/edit/<int:id>', methods=['GET', 'POST'], endpoint='recharge_activity_edit')
|
|
@login_required
|
|
@permission_required('recharge_activity_edit')
|
|
def recharge_activity_edit(id):
|
|
activity = db.session.get(RechargeActivity, id)
|
|
if request.method == 'POST':
|
|
activity.name = request.form.get('name')
|
|
activity.trigger_amount = float(request.form.get('trigger_amount', 0))
|
|
activity.gifted_hours = float(request.form.get('gifted_hours', 0))
|
|
activity.is_time_limited = int(request.form.get('is_time_limited', 0))
|
|
activity.start_date = _parse_date(request.form.get('start_date'))
|
|
activity.end_date = _parse_date(request.form.get('end_date'))
|
|
activity.can_stack = int(request.form.get('can_stack', 1))
|
|
activity.status = int(request.form.get('status', 1))
|
|
db.session.commit()
|
|
log_operation('编辑充值活动', f'活动ID:{id}')
|
|
flash('充值活动更新成功', 'success')
|
|
return redirect(url_for('recharge_activity_list'))
|
|
|
|
return render_template('recharge_activity_form.html', activity=activity)
|
|
|
|
@app.route('/recharge_activities/delete/<int:id>', endpoint='recharge_activity_delete')
|
|
@login_required
|
|
@permission_required('recharge_activity_edit')
|
|
def recharge_activity_delete(id):
|
|
activity = db.session.get(RechargeActivity, id)
|
|
if activity:
|
|
activity.status = 0
|
|
db.session.commit()
|
|
log_operation('停用充值活动', f'活动ID:{id}')
|
|
flash('充值活动已停用', 'success')
|
|
return redirect(url_for('recharge_activity_list'))
|
|
|
|
@app.route('/recharges', endpoint='recharge_list')
|
|
@login_required
|
|
@permission_required('recharge_view')
|
|
def recharge_list():
|
|
keyword = request.args.get('keyword', '')
|
|
query = RechargeRecord.query
|
|
if keyword:
|
|
students = filter_students_query(
|
|
Student.query.filter(Student.name.contains(keyword)),
|
|
).all()
|
|
student_ids = [s.id for s in students]
|
|
query = query.filter(RechargeRecord.student_id.in_(student_ids)) if student_ids else query.filter(False)
|
|
if is_teacher_role():
|
|
allowed = {s.id for s in filter_students_query(Student.query).all()}
|
|
records = [
|
|
r for r in query.order_by(RechargeRecord.created_at.desc()).all()
|
|
if r.student_id in allowed
|
|
]
|
|
else:
|
|
records = query.order_by(RechargeRecord.created_at.desc()).all()
|
|
return render_template('recharge_list.html', records=records, keyword=keyword)
|
|
|
|
@app.route('/recharges/add', methods=['GET', 'POST'], endpoint='recharge_add')
|
|
@login_required
|
|
@permission_required('recharge_add')
|
|
def recharge_add():
|
|
if request.method == 'POST':
|
|
student_id = int(request.form.get('student_id'))
|
|
course_id = int(request.form.get('course_id'))
|
|
amount = float(request.form.get('amount', 0))
|
|
if amount <= 0:
|
|
flash('充值金额必须大于 0', 'danger')
|
|
return redirect(url_for('recharge_add'))
|
|
payment_method = request.form.get('payment_method', '微信')
|
|
remark = request.form.get('remark', '')
|
|
|
|
course = db.session.get(Course, course_id)
|
|
normal_hours = rules.calc_normal_hours_from_amount(
|
|
amount, course.price_per_hour if course else 0
|
|
)
|
|
gifted_hours, activity_id = rules.match_recharge_activities(
|
|
student_id, course_id, amount
|
|
)
|
|
|
|
record = RechargeRecord(
|
|
student_id=student_id, course_id=course_id, amount=amount,
|
|
activity_id=activity_id,
|
|
normal_hours=normal_hours, gifted_hours=gifted_hours,
|
|
payment_method=payment_method, operator_id=session.get('user_id'),
|
|
remark=remark,
|
|
)
|
|
db.session.add(record)
|
|
|
|
account = rules.lock_student_account(student_id, course_id)
|
|
is_new = not account
|
|
if is_new:
|
|
account = StudentAccount(student_id=student_id, course_id=course_id)
|
|
db.session.add(account)
|
|
db.session.flush()
|
|
|
|
before = rules.account_snapshot_dict(account)
|
|
rules.apply_recharge_account(
|
|
account, amount, normal_hours, gifted_hours, is_new_account=is_new,
|
|
)
|
|
db.session.commit()
|
|
log_operation(
|
|
'充值',
|
|
f'学员ID:{student_id} 充值{amount}元(正课{normal_hours}+赠课{int(gifted_hours)})',
|
|
before=before, after=rules.account_snapshot_dict(account),
|
|
)
|
|
flash(f'充值成功！正课{normal_hours}课时，赠课{int(gifted_hours)}课时', 'success')
|
|
return redirect(url_for('recharge_list'))
|
|
|
|
students = filter_students_query(Student.query.filter_by(status=1)).all()
|
|
courses = Course.query.filter_by(status=1).all()
|
|
return render_template('recharge_form.html', students=students, courses=courses)
|
|
|
|
@app.route('/referral_reward', methods=['GET', 'POST'], endpoint='referral_reward')
|
|
@login_required
|
|
@permission_required('recharge_add')
|
|
def referral_reward():
|
|
"""§一 1.5 老带新奖励（管理员手动发放）"""
|
|
if request.method == 'POST':
|
|
referrer_id = int(request.form.get('referrer_id'))
|
|
student_id = int(request.form.get('student_id'))
|
|
course_id = int(request.form.get('course_id'))
|
|
reward_hours = rules.normalize_consumption_hours(
|
|
request.form.get('reward_hours', 1)
|
|
)
|
|
remark = request.form.get('remark', '老带新奖励')
|
|
|
|
account = StudentAccount.query.filter_by(
|
|
student_id=referrer_id, course_id=course_id
|
|
).first()
|
|
if not account:
|
|
account = StudentAccount(student_id=referrer_id, course_id=course_id)
|
|
db.session.add(account)
|
|
db.session.flush()
|
|
account.gifted_hours = float(account.gifted_hours or 0) + reward_hours
|
|
db.session.commit()
|
|
log_operation(
|
|
'老带新奖励',
|
|
f'推荐人学员ID:{referrer_id} 获赠{reward_hours}课时 新学员ID:{student_id}',
|
|
)
|
|
flash(f'已向推荐人发放 {reward_hours} 课时赠课', 'success')
|
|
return redirect(url_for('referral_reward'))
|
|
|
|
students = Student.query.filter_by(status=1).all()
|
|
courses = Course.query.filter_by(status=1).all()
|
|
return render_template(
|
|
'referral_reward.html', students=students, courses=courses
|
|
)
|
|
|
|
@app.route('/api/student_account/<int:student_id>/<int:course_id>')
|
|
@login_required
|
|
@permission_required('recharge_view')
|
|
def api_student_account(student_id, course_id):
|
|
account = StudentAccount.query.filter_by(
|
|
student_id=student_id, course_id=course_id
|
|
).first()
|
|
if account:
|
|
nh = int(float(account.normal_hours or 0))
|
|
gh = int(float(account.gifted_hours or 0))
|
|
return jsonify({
|
|
'success': True,
|
|
'data': {
|
|
'normal_hours': nh,
|
|
'gifted_hours': gh,
|
|
'total_hours': nh + gh,
|
|
'consumed_normal': float(account.consumed_normal or 0),
|
|
'consumed_gifted': float(account.consumed_gifted or 0),
|
|
'cumulative_amount': float(account.cumulative_amount or 0),
|
|
'is_stopped': bool(account.is_stopped),
|
|
},
|
|
})
|
|
return jsonify({'success': False, 'message': '未找到账户'})
|
|
|
|
@app.route('/api/student_accounts/<int:student_id>')
|
|
@login_required
|
|
@permission_required('recharge_view')
|
|
def api_student_accounts(student_id):
|
|
accounts = StudentAccount.query.filter_by(student_id=student_id).all()
|
|
result = []
|
|
for a in accounts:
|
|
course = db.session.get(Course, a.course_id)
|
|
nh = int(float(a.normal_hours or 0))
|
|
gh = int(float(a.gifted_hours or 0))
|
|
result.append({
|
|
'course_id': a.course_id,
|
|
'course_name': course.name if course else '',
|
|
'normal_hours': nh,
|
|
'gifted_hours': gh,
|
|
'cumulative_amount': float(a.cumulative_amount or 0),
|
|
'is_stopped': bool(a.is_stopped),
|
|
})
|
|
return jsonify({'success': True, 'data': result})
|
|
|
|
@app.route('/api/calculate_recharge', methods=['POST'])
|
|
@login_required
|
|
@permission_required('recharge_view')
|
|
def api_calculate_recharge():
|
|
data = request.get_json()
|
|
student_id = data.get('student_id')
|
|
course_id = data.get('course_id')
|
|
amount = float(data.get('amount', 0))
|
|
|
|
course = db.session.get(Course, course_id)
|
|
normal_hours = rules.calc_normal_hours_from_amount(
|
|
amount, course.price_per_hour if course else 0
|
|
)
|
|
gifted_hours, _ = rules.match_recharge_activities(student_id, course_id, amount)
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'data': {
|
|
'normal_hours': normal_hours,
|
|
'gifted_hours': int(gifted_hours),
|
|
'total_hours': normal_hours + int(gifted_hours),
|
|
},
|
|
})
|
|
|
|
@app.route('/recharge_activities/export', endpoint='recharge_activity_export')
|
|
@login_required
|
|
@permission_required('recharge_activity_export')
|
|
def recharge_activity_export():
|
|
activities = RechargeActivity.query.order_by(RechargeActivity.id.desc()).all()
|
|
headers = ['序号', '活动名称', '触发金额', '赠送课时', '可叠加', '期限活动',
|
|
'开始日期', '结束日期', '状态', '备注']
|
|
rows = []
|
|
for i, a in enumerate(activities, 1):
|
|
rows.append([
|
|
i, a.name, a.trigger_amount, a.gifted_hours,
|
|
'是' if a.can_stack else '否',
|
|
'是' if a.is_time_limited else '否',
|
|
str(a.start_date) if a.start_date else '',
|
|
str(a.end_date) if a.end_date else '',
|
|
'启用' if a.status == 1 else '禁用',
|
|
'',
|
|
])
|
|
return export_excel('充值优惠活动', headers, rows)
|
|
|
|
@app.route('/recharges/export', endpoint='recharge_export')
|
|
@login_required
|
|
@permission_required('recharge_export')
|
|
def recharge_export():
|
|
from models import User
|
|
records = RechargeRecord.query.order_by(RechargeRecord.id.desc()).all()
|
|
headers = ['序号', '学员', '课程', '充值金额', '正课课时', '赠课课时',
|
|
'支付方式', '操作人', '备注', '创建时间']
|
|
rows = []
|
|
for i, r in enumerate(records, 1):
|
|
student = db.session.get(Student, r.student_id)
|
|
course = db.session.get(Course, r.course_id)
|
|
operator = db.session.get(User, r.operator_id) if r.operator_id else None
|
|
rows.append([
|
|
i, student.name if student else '', course.name if course else '',
|
|
r.amount or 0, r.normal_hours or 0, r.gifted_hours or 0,
|
|
r.payment_method or '', operator.real_name if operator else '',
|
|
r.remark or '',
|
|
r.created_at.strftime('%Y-%m-%d %H:%M') if r.created_at else '',
|
|
])
|
|
return export_excel('充值记录', headers, rows)
|
|
|
|
|
|
def _parse_date(date_str):
|
|
if date_str:
|
|
try:
|
|
return datetime.strptime(date_str, '%Y-%m-%d').date()
|
|
except (ValueError, TypeError):
|
|
return None
|
|
return None
|