REMOVE LDAP

This commit is contained in:
beabigegg
2025-09-25 08:44:44 +08:00
commit 333a640a3b
53 changed files with 4231 additions and 0 deletions

0
routes/__init__.py Normal file
View File

120
routes/admin.py Normal file
View File

@@ -0,0 +1,120 @@
from flask import Blueprint, render_template, request, redirect, url_for, flash
from flask_login import login_required, current_user
from models import User, db
from utils import admin_required
admin_bp = Blueprint('admin', __name__, url_prefix='/admin')
ALLOWED_ROLES = {'viewer', 'editor', 'admin'}
@admin_bp.route('/users')
@login_required
@admin_required
def user_list():
users = User.query.order_by(User.username).all()
return render_template('user_management.html', users=users)
@admin_bp.route('/users/create', methods=['POST'])
@login_required
@admin_required
def create_user():
username = request.form.get('username', '').strip()
name = request.form.get('name', '').strip()
password = request.form.get('password', '')
role = request.form.get('role', 'viewer')
errors = []
if not username:
errors.append('請輸入帳號')
if not name:
errors.append('請輸入姓名')
if not password:
errors.append('請輸入密碼')
if password and len(password) < 6:
errors.append('密碼長度至少需 6 碼')
if role not in ALLOWED_ROLES:
errors.append('角色設定不正確')
if username and User.query.filter_by(username=username).first():
errors.append('帳號已存在,請改用其他帳號')
if errors:
for message in errors:
flash(message, 'danger')
return redirect(url_for('admin.user_list'))
new_user = User(username=username, name=name, role=role)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
flash(f"已建立帳號 {username}", 'success')
return redirect(url_for('admin.user_list'))
@admin_bp.route('/users/update/<int:user_id>', methods=['POST'])
@login_required
@admin_required
def update_user(user_id):
user = User.query.get_or_404(user_id)
name = request.form.get('name', '').strip()
role = request.form.get('role', 'viewer')
password = request.form.get('password', '').strip()
if not name:
flash('請輸入姓名', 'danger')
return redirect(url_for('admin.user_list'))
if role not in ALLOWED_ROLES:
flash('角色設定不正確', 'danger')
return redirect(url_for('admin.user_list'))
if password and len(password) < 6:
flash('密碼長度至少需 6 碼', 'danger')
return redirect(url_for('admin.user_list'))
if user.id == current_user.id and user.role == 'admin' and role != 'admin':
flash('無法變更自己的管理員權限', 'danger')
return redirect(url_for('admin.user_list'))
if user.role == 'admin' and role != 'admin':
admin_count = User.query.filter_by(role='admin').count()
if admin_count <= 1:
flash('系統至少需要一位管理員,無法變更該帳號的管理員權限', 'danger')
return redirect(url_for('admin.user_list'))
user.name = name
user.role = role
if password:
user.set_password(password)
db.session.commit()
flash(f"已更新帳號 {user.username}", 'success')
return redirect(url_for('admin.user_list'))
@admin_bp.route('/users/delete/<int:user_id>', methods=['POST'])
@login_required
@admin_required
def delete_user(user_id):
if user_id == current_user.id:
flash('無法刪除自己的帳號', 'danger')
return redirect(url_for('admin.user_list'))
user = User.query.get_or_404(user_id)
if user.role == 'admin':
admin_count = User.query.filter_by(role='admin').count()
if admin_count <= 1:
flash('系統至少需要一位管理員,無法刪除該帳號', 'danger')
return redirect(url_for('admin.user_list'))
db.session.delete(user)
db.session.commit()
flash(f"已刪除帳號 {user.username}", 'success')
return redirect(url_for('admin.user_list'))

9
routes/api.py Normal file
View File

@@ -0,0 +1,9 @@
from flask import Blueprint, jsonify
api_bp = Blueprint('api', __name__, url_prefix='/api')
@api_bp.route('/health', methods=['GET'])
def health_check():
"""簡易 API 健康檢查端點。"""
return jsonify({'status': 'ok'})

93
routes/auth.py Normal file
View File

@@ -0,0 +1,93 @@
from flask import Blueprint, render_template, request, redirect, url_for, flash, current_app
from flask_login import login_user, logout_user, login_required, current_user
from models import User, db
from utils.timezone import taiwan_now
auth_bp = Blueprint('auth', __name__)
@auth_bp.route('/login', methods=['GET', 'POST'])
def login():
if current_user.is_authenticated:
return redirect(url_for('temp_spec.spec_list'))
context = {}
if request.method == 'POST':
username = request.form['username'].strip()
password = request.form['password']
context['username'] = username
if not username or not password:
flash('請輸入帳號與密碼', 'warning')
return render_template('login.html', **context)
user = User.query.filter_by(username=username).first()
if user and user.check_password(password):
user.last_login = taiwan_now()
db.session.commit()
login_user(user)
current_app.logger.info(f"User logged in via local authentication: {username}")
return redirect(url_for('temp_spec.spec_list'))
current_app.logger.warning(f"Failed local login attempt for: {username}")
flash('帳號或密碼錯誤,請重新輸入', 'danger')
return render_template('login.html', **context)
@auth_bp.route('/register', methods=['GET', 'POST'])
def register():
if current_user.is_authenticated:
return redirect(url_for('temp_spec.spec_list'))
context = {}
if request.method == 'POST':
username = request.form['username'].strip()
name = request.form['name'].strip()
password = request.form['password']
confirm_password = request.form['confirm_password']
context['username'] = username
context['name'] = name
errors = []
if not username:
errors.append('請輸入帳號')
if not name:
errors.append('請輸入姓名')
if not password:
errors.append('請輸入密碼')
if password and len(password) < 6:
errors.append('密碼長度至少需 6 碼')
if password != confirm_password:
errors.append('確認密碼不相符')
if username and User.query.filter_by(username=username).first():
errors.append('帳號已存在,請改用其他帳號')
if errors:
for message in errors:
flash(message, 'danger')
return render_template('register.html', **context)
new_user = User(username=username, name=name, role='viewer')
new_user.set_password(password)
new_user.last_login = taiwan_now()
db.session.add(new_user)
db.session.commit()
login_user(new_user)
current_app.logger.info(f"New user registered: {username}")
flash('帳號建立完成,已自動登入', 'success')
return redirect(url_for('temp_spec.spec_list'))
return render_template('register.html', **context)
@auth_bp.route('/logout')
@login_required
def logout():
logout_user()
return redirect(url_for('auth.login'))

522
routes/temp_spec.py Normal file
View File

@@ -0,0 +1,522 @@
# -*- coding: utf-8 -*-
from flask import Blueprint, render_template, request, redirect, url_for, flash, send_file, current_app, jsonify, abort
from flask_login import login_required, current_user
from datetime import datetime, timedelta
from utils.timezone import taiwan_now, format_taiwan_time
from models import TempSpec, db, Upload, SpecHistory
from utils import editor_or_admin_required, add_history_log, admin_required, send_email, process_recipients
import os
import shutil
import jwt
import requests
from werkzeug.utils import secure_filename
from docxtpl import DocxTemplate
temp_spec_bp = Blueprint('temp_spec', __name__)
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# Enforce authentication for this blueprint (except OnlyOffice callback)
@temp_spec_bp.before_request
def before_request():
"""Ensure every request under this blueprint comes from an authenticated user."""
# Allow OnlyOffice callback without authentication
if request.endpoint == 'temp_spec.onlyoffice_callback':
return None
if not current_user.is_authenticated:
return redirect(url_for('auth.login'))
def _generate_next_spec_code():
"""Generate the next temporary spec code based on ROC year/month and a sequence."""
now = taiwan_now()
roc_year = now.year - 1911
prefix = f"PE{roc_year}{now.strftime('%m')}"
latest_spec = TempSpec.query.filter(
TempSpec.spec_code.startswith(prefix)
).order_by(TempSpec.spec_code.desc()).first()
if latest_spec:
last_seq = int(latest_spec.spec_code[-2:])
new_seq = last_seq + 1
else:
new_seq = 1
return f"{prefix}{new_seq:02d}"
def get_file_uri(filename: str) -> str:
"""Return a public URL for a generated file stored under /static/generated."""
return url_for('static', filename=f"generated/{filename}", _external=True)
@temp_spec_bp.route('/create', methods=['GET', 'POST'])
@editor_or_admin_required
def create_temp_spec():
if request.method == 'POST':
spec_code = _generate_next_spec_code()
form_data = request.form
now = taiwan_now()
spec = TempSpec(
spec_code=spec_code,
title=form_data['theme'],
applicant=form_data['applicant'],
status='pending_approval',
created_at=now,
start_date=now.date(),
end_date=(now + timedelta(days=30)).date()
)
db.session.add(spec)
db.session.flush()
context = {
'serial_number': spec_code,
'theme': form_data.get('theme', ''),
'package': form_data.get('package', ''),
'lot_number': form_data.get('lot_number', ''),
'equipment_type': form_data.get('equipment_type', ''),
'applicant': form_data.get('applicant', ''),
'applicant_phone': form_data.get('applicant_phone', ''),
'start_date': now.strftime('%Y-%m-%d'),
'end_date': (now + timedelta(days=30)).strftime('%Y-%m-%d'),
}
selected_stations = request.form.getlist('station')
station_keys = ['probing', 'dicing', 'diebond', 'wirebond', 'solder', 'molding',
'degate', 'deflash', 'plating', 'trimform', 'marking', 'tmtt', 'other']
for key in station_keys:
context[f's_{key}'] = 'Y' if key in selected_stations else 'N'
selected_tccs_level = form_data.get('tccs_level')
level_keys = ['l1', 'l2', 'l3', 'l4']
for key in level_keys:
context[f't_{key}'] = 'Y' if key == selected_tccs_level else 'N'
selected_tccs_4m = form_data.get('tccs_4m')
m_keys = ['man', 'machine', 'material', 'method', 'env']
for key in m_keys:
context[f't_{key}'] = 'Y' if key == selected_tccs_4m else 'N'
generated_folder = os.path.join(current_app.static_folder, 'generated')
os.makedirs(generated_folder, exist_ok=True)
template_path = os.path.join(BASE_DIR, 'template_with_placeholders.docx')
new_file_path = os.path.join(generated_folder, f"{spec_code}.docx")
if not os.path.exists(template_path):
flash('Word template (template_with_placeholders.docx) was not found.', 'danger')
db.session.rollback()
return redirect(url_for('temp_spec.spec_list'))
doc = DocxTemplate(template_path)
doc.render(context)
doc.save(new_file_path)
add_history_log(spec.id, 'create', f"Created draft spec: {spec.spec_code}")
db.session.commit()
return redirect(url_for('temp_spec.edit_spec', spec_id=spec.id))
return render_template('create_temp_spec_form.html')
@temp_spec_bp.route('/edit/<int:spec_id>')
@editor_or_admin_required
def edit_spec(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
doc_filename = f"{spec.spec_code}.docx"
doc_physical_path = os.path.join(current_app.static_folder, 'generated', doc_filename)
if not os.path.exists(doc_physical_path):
flash(f"Generated document not found: {doc_filename}", 'danger')
return redirect(url_for('temp_spec.spec_list'))
doc_url = get_file_uri(doc_filename)
callback_url = url_for('temp_spec.onlyoffice_callback', spec_id=spec_id, _external=True)
if '127.0.0.1' in doc_url or 'localhost' in doc_url:
doc_url = doc_url.replace('127.0.0.1:12013', 'panjit-tempspec-nginx:80').replace('localhost:12013', 'panjit-tempspec-nginx:80')
doc_url = doc_url.replace('127.0.0.1', 'panjit-tempspec-nginx').replace('localhost', 'panjit-tempspec-nginx')
callback_url = callback_url.replace('127.0.0.1:12013', 'panjit-tempspec-nginx:80').replace('localhost:12013', 'panjit-tempspec-nginx:80')
callback_url = callback_url.replace('127.0.0.1', 'panjit-tempspec-nginx').replace('localhost', 'panjit-tempspec-nginx')
oo_secret = current_app.config['ONLYOFFICE_JWT_SECRET']
file_key = f"{spec.id}_{int(os.path.getmtime(doc_physical_path))}"
payload = {
"document": {
"fileType": "docx",
"key": file_key,
"title": doc_filename,
"url": doc_url
},
"documentType": "word",
"editorConfig": {
"callbackUrl": callback_url,
"user": { "id": str(current_user.id), "name": current_user.name or current_user.username },
"customization": {
"autosave": True,
"forcesave": True,
"chat": False,
"comments": True,
"help": False
},
"mode": "edit"
}
}
token = jwt.encode(payload, oo_secret, algorithm='HS256')
config = payload.copy()
config['token'] = token
return render_template(
'onlyoffice_editor.html',
spec=spec,
config=config,
onlyoffice_url=current_app.config['ONLYOFFICE_URL']
)
@temp_spec_bp.route('/onlyoffice-callback/<int:spec_id>', methods=['POST'])
def onlyoffice_callback(spec_id):
data = request.json
status = data.get('status')
current_app.logger.info(f"OnlyOffice callback for spec {spec_id}: status={status}, data={data}")
if status in [2, 6]:
try:
if 'url' not in data:
current_app.logger.error(f"OnlyOffice callback missing URL for spec {spec_id}")
return jsonify({"error": 1, "message": "Missing document URL"})
token = data.get('token')
if token:
try:
oo_secret = current_app.config['ONLYOFFICE_JWT_SECRET']
jwt.decode(token, oo_secret, algorithms=['HS256'])
except jwt.InvalidTokenError:
current_app.logger.error(f"Invalid JWT token in OnlyOffice callback for spec {spec_id}")
return jsonify({"error": 1, "message": "Invalid token"})
download_url = data['url']
download_url = download_url.replace('localhost:12015', 'panjit-tempspec-onlyoffice:80')
download_url = download_url.replace('127.0.0.1:12015', 'panjit-tempspec-onlyoffice:80')
current_app.logger.info(f"Downloading updated document from: {data['url']} -> {download_url}")
response = requests.get(download_url, timeout=30)
response.raise_for_status()
spec = TempSpec.query.get_or_404(spec_id)
doc_filename = f"{spec.spec_code}.docx"
file_path = os.path.join(current_app.static_folder, 'generated', doc_filename)
os.makedirs(os.path.dirname(file_path), exist_ok=True)
with open(file_path, 'wb') as f:
f.write(response.content)
current_app.logger.info(f"Successfully saved updated document for spec {spec_id} to {file_path}")
spec.updated_at = taiwan_now()
db.session.commit()
except requests.RequestException as e:
current_app.logger.error(f"Failed to download document from OnlyOffice for spec {spec_id}: {e}")
return jsonify({"error": 1, "message": f"Download failed: {str(e)}"})
except Exception as e:
current_app.logger.error(f"OnlyOffice callback error for spec {spec_id}: {e}")
return jsonify({"error": 1, "message": str(e)})
elif status == 1:
current_app.logger.info(f"Document {spec_id} is being edited")
elif status == 4:
current_app.logger.info(f"Document {spec_id} closed without changes")
elif status == 7:
current_app.logger.error(f"OnlyOffice error for document {spec_id}")
return jsonify({"error": 1, "message": "OnlyOffice reported an error"})
return jsonify({"error": 0})
@temp_spec_bp.route('/list')
@login_required
def spec_list():
page = request.args.get('page', 1, type=int)
query = request.args.get('query', '')
status_filter = request.args.get('status', '')
specs_query = TempSpec.query
if query:
search_term = f"%{query}%"
specs_query = specs_query.filter(
db.or_(
TempSpec.spec_code.ilike(search_term),
TempSpec.title.ilike(search_term)
)
)
if status_filter:
specs_query = specs_query.filter(TempSpec.status == status_filter)
pagination = specs_query.order_by(TempSpec.created_at.desc()).paginate(
page=page, per_page=15, error_out=False
)
specs = pagination.items
from datetime import date
today = date.today()
return render_template(
'spec_list.html',
specs=specs,
pagination=pagination,
query=query,
status=status_filter,
today=today
)
@temp_spec_bp.route('/activate/<int:spec_id>', methods=['GET', 'POST'])
@admin_required
def activate_spec(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
if request.method == 'POST':
uploaded_file = request.files.get('signed_file')
if not uploaded_file or uploaded_file.filename == '':
flash('Please upload a file first.', 'danger')
return redirect(url_for('temp_spec.activate_spec', spec_id=spec.id))
filename = secure_filename(f"{spec.spec_code}_signed_{taiwan_now().strftime('%Y%m%d%H%M%S')}.pdf")
upload_folder = os.path.join(BASE_DIR, current_app.config['UPLOAD_FOLDER'])
os.makedirs(upload_folder, exist_ok=True)
file_path = os.path.join(upload_folder, filename)
uploaded_file.save(file_path)
new_upload = Upload(
temp_spec_id=spec.id,
filename=filename,
upload_time=taiwan_now()
)
db.session.add(new_upload)
spec.status = 'active'
recipients_str = request.form.get('recipients')
if recipients_str:
spec.notification_emails = recipients_str.strip()
add_history_log(spec.id, 'activate', f"Uploaded signed file '{filename}'")
db.session.commit()
flash(f"Spec '{spec.spec_code}' is now active.", 'success')
# --- Start of Dynamic Email Notification ---
if recipients_str:
recipients = process_recipients(recipients_str)
if recipients:
subject = f"[TempSpec Notice] Spec '{spec.spec_code}' is active"
# Using f-strings and triple quotes for a readable HTML body
body = f"""
<html>
<body>
<p>Hello,</p>
<p>Temp spec <b>{spec.spec_code} - {spec.title}</b> has been approved and is now active.</p>
<p>Start date: {spec.start_date.strftime('%Y-%m-%d')}<br>
End date: {spec.end_date.strftime('%Y-%m-%d')}</p>
<p>Applicant: {spec.applicant}</p>
<p>Please sign in to the system for additional details.</p>
<p>This notification was sent automatically. Please do not reply.</p>
</body>
</html>
"""
send_email(recipients, subject, body)
current_app.logger.info(f"Sent activation notification for spec {spec.spec_code} to {len(recipients)} recipients.")
# --- End of Dynamic Email Notification ---
return redirect(url_for('temp_spec.spec_list'))
return render_template('activate_spec.html', spec=spec)
@temp_spec_bp.route('/terminate/<int:spec_id>', methods=['GET', 'POST'])
@editor_or_admin_required
def terminate_spec(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
if request.method == 'POST':
reason = request.form.get('reason')
if not reason:
flash('Please provide the reason for early termination.', 'danger')
return redirect(url_for('temp_spec.terminate_spec', spec_id=spec.id))
spec.status = 'terminated'
spec.termination_reason = reason
spec.end_date = taiwan_now().date()
add_history_log(spec.id, 'terminate', f"Reason: {reason}")
# --- Start of Dynamic Email Notification ---
recipients_str = request.form.get('recipients')
if not recipients_str and spec.notification_emails:
recipients_str = spec.notification_emails
if recipients_str:
recipients = process_recipients(recipients_str)
if recipients:
subject = f"[TempSpec Notice] Spec '{spec.spec_code}' was terminated early"
body = f"""
<html>
<body>
<p>Hello,</p>
<p>Temp spec <b>{spec.spec_code} - {spec.title}</b> has been terminated.</p>
<p>Termination date: <b>{spec.end_date.strftime('%Y-%m-%d')}</b></p>
<p>Applicant: {spec.applicant}</p>
<p>Reason: {reason}</p>
<p>Please sign in to the system for additional details.</p>
<p>This notification was sent automatically. Please do not reply.</p>
</body>
</html>
"""
send_email(recipients, subject, body)
current_app.logger.info(f"Sent termination notification for spec {spec.spec_code} to {len(recipients)} recipients.")
# --- End of Dynamic Email Notification ---
db.session.commit()
flash(f"Spec '{spec.spec_code}' has been terminated early.", 'warning')
return redirect(url_for('temp_spec.spec_list'))
return render_template('terminate_spec.html', spec=spec, saved_emails=spec.notification_emails)
@temp_spec_bp.route('/download_initial_word/<int:spec_id>')
@login_required
def download_initial_word(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
if current_user.role not in ['editor', 'admin']:
flash('You do not have permission to download the Word document.', 'danger')
abort(403)
generated_folder = os.path.join(current_app.static_folder, 'generated')
word_path = os.path.join(generated_folder, f"{spec.spec_code}.docx")
if not os.path.exists(word_path):
flash('The original Word document cannot be found. It might have been moved or deleted.', 'danger')
return redirect(url_for('temp_spec.spec_list'))
return send_file(word_path, as_attachment=True)
@temp_spec_bp.route('/download_signed/<int:spec_id>')
@login_required
def download_signed_pdf(spec_id):
latest_upload = Upload.query.filter_by(temp_spec_id=spec_id).order_by(Upload.upload_time.desc()).first()
if not latest_upload:
flash('No signed files have been uploaded yet.', 'danger')
return redirect(url_for('temp_spec.spec_list'))
upload_folder = os.path.join(BASE_DIR, current_app.config['UPLOAD_FOLDER'])
return send_file(os.path.join(upload_folder, latest_upload.filename), as_attachment=True)
@temp_spec_bp.route('/extend/<int:spec_id>', methods=['GET', 'POST'])
@editor_or_admin_required
def extend_spec(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
if spec.extension_count >= 2:
flash('This temporary specification has reached the extension limit (2 times) and the total 90-day duration. Extension is no longer allowed.', 'danger')
return redirect(url_for('temp_spec.spec_list'))
if request.method == 'POST':
new_end_date_str = request.form.get('new_end_date')
uploaded_file = request.files.get('new_file')
if not uploaded_file or uploaded_file.filename == '':
flash('Please upload the supporting PDF before extending.', 'danger')
return redirect(url_for('temp_spec.extend_spec', spec_id=spec.id))
if not new_end_date_str:
flash('Please choose a new end date.', 'danger')
return redirect(url_for('temp_spec.extend_spec', spec_id=spec.id))
spec.end_date = datetime.strptime(new_end_date_str, '%Y-%m-%d').date()
spec.extension_count += 1
spec.status = 'active'
filename = secure_filename(f"{spec.spec_code}_extension_{spec.extension_count}_{taiwan_now().strftime('%Y%m%d')}.pdf")
upload_folder = os.path.join(BASE_DIR, current_app.config['UPLOAD_FOLDER'])
os.makedirs(upload_folder, exist_ok=True)
file_path = os.path.join(upload_folder, filename)
uploaded_file.save(file_path)
new_upload = Upload(
temp_spec_id=spec.id,
filename=filename,
upload_time=taiwan_now()
)
db.session.add(new_upload)
details = f"Extended end date to {spec.end_date.strftime('%Y-%m-%d')}"
details += f", uploaded file '{new_upload.filename}'"
add_history_log(spec.id, 'extend', details)
# --- Start of Dynamic Email Notification ---
recipients_str = request.form.get('recipients')
if not recipients_str and spec.notification_emails:
recipients_str = spec.notification_emails
if recipients_str:
spec.notification_emails = recipients_str.strip()
if recipients_str:
recipients = process_recipients(recipients_str)
if recipients:
subject = f"[TempSpec Notice] Spec '{spec.spec_code}' was extended"
body = f"""
<html>
<body>
<p>Hello,</p>
<p>Temp spec <b>{spec.spec_code} - {spec.title}</b> has been extended.</p>
<p>New end date: <b>{spec.end_date.strftime('%Y-%m-%d')}</b></p>
<p>Applicant: {spec.applicant}</p>
<p>Please sign in to the system for additional details.</p>
<p>This notification was sent automatically. Please do not reply.</p>
</body>
</html>
"""
send_email(recipients, subject, body)
current_app.logger.info(f"Sent extension notification for spec {spec.spec_code} to {len(recipients)} recipients.")
# --- End of Dynamic Email Notification ---
db.session.commit()
flash(f"Spec '{spec.spec_code}' was extended successfully.", 'success')
return redirect(url_for('temp_spec.spec_list'))
default_new_end_date = spec.end_date + timedelta(days=30)
return render_template('extend_spec.html', spec=spec, default_new_end_date=default_new_end_date, saved_emails=spec.notification_emails)
@temp_spec_bp.route('/history/<int:spec_id>')
@login_required
def spec_history(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
history = SpecHistory.query.filter_by(spec_id=spec_id).order_by(SpecHistory.timestamp.desc()).all()
return render_template('spec_history.html', spec=spec, history=history)
@temp_spec_bp.route('/delete/<int:spec_id>', methods=['POST'])
@admin_required
def delete_spec(spec_id):
spec = TempSpec.query.get_or_404(spec_id)
spec_code = spec.spec_code
files_to_delete = []
generated_folder = os.path.join(current_app.static_folder, 'generated')
files_to_delete.append(os.path.join(generated_folder, f"{spec.spec_code}.docx"))
upload_folder = os.path.join(BASE_DIR, current_app.config['UPLOAD_FOLDER'])
for upload_record in spec.uploads:
files_to_delete.append(os.path.join(upload_folder, upload_record.filename))
for f_path in files_to_delete:
try:
if os.path.exists(f_path):
os.remove(f_path)
except Exception as e:
current_app.logger.error(f"Failed to delete file: {f_path}, reason: {e}")
db.session.delete(spec)
db.session.commit()
flash(f"Spec '{spec_code}' and related files were deleted successfully.", 'success')
return redirect(url_for('temp_spec.spec_list'))

29
routes/upload.py Normal file
View File

@@ -0,0 +1,29 @@
from flask import Blueprint, request, jsonify, current_app
from werkzeug.utils import secure_filename
import os
import time
upload_bp = Blueprint('upload', __name__)
@upload_bp.route('/image', methods=['POST'])
def upload_image():
file = request.files.get('file')
if not file:
return jsonify({'error': 'No file part'}), 400
# 建立一個獨特的檔名
extension = os.path.splitext(file.filename)[1]
filename = f"{int(time.time())}_{secure_filename(file.filename)}"
# 確保上傳資料夾存在
# 為了讓圖片能被網頁存取,我們將它存在 static 資料夾下
image_folder = os.path.join(current_app.static_folder, 'uploads', 'images')
os.makedirs(image_folder, exist_ok=True)
file_path = os.path.join(image_folder, filename)
file.save(file_path)
# 回傳 TinyMCE 需要的 JSON 格式
# 路徑必須是相對於網域根目錄的 URL
location = f"/static/uploads/images/{filename}"
return jsonify({'location': location})