gmao/app_new/documents/routes.py

"""
Documents Routes - GMAO Collège
Gestion des documents attachés aux interventions et équipements
"""
from flask import Blueprint, render_template, redirect, url_for, request, flash, current_app
from flask_login import login_required, current_user
from werkzeug.utils import secure_filename
from datetime import datetime, timezone
import os
from app_new.extensions import db
from ..core.models.maintenance import InterventionDocument

documents_bp = Blueprint('documents', __name__, url_prefix='/documents')

ALLOWED_EXTENSIONS = {'pdf', 'png', 'jpg', 'jpeg', 'gif', 'doc', 'docx', 'xls', 'xlsx'}

def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS


@documents_bp.route('/upload/intervention/<int:intervention_id>', methods=['POST'])
@login_required
def upload_intervention(intervention_id):
    """Upload un document pour une intervention."""
    if 'file' not in request.files:
        flash('Aucun fichier sélectionné.', 'error')
        return redirect(url_for('interventions.detail', id=intervention_id))
    
    file = request.files['file']
    
    if file.filename == '':
        flash('Aucun fichier sélectionné.', 'error')
        return redirect(url_for('interventions.detail', id=intervention_id))
    
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        timestamp = datetime.now(timezone.utc).strftime('%Y%m%d%H%M%S')
        unique_filename = f"{timestamp}_{filename}"
        
        upload_dir = os.path.join(current_app.config.get('UPLOAD_FOLDER', 'uploads'), 'interventions')
        os.makedirs(upload_dir, exist_ok=True)
        
        filepath = os.path.join(upload_dir, unique_filename)
        file.save(filepath)
        
        document = InterventionDocument(
            intervention_id=intervention_id,
            filename=unique_filename,
            filepath=filepath,
            description=request.form.get('description', ''),
            uploaded_by_id=current_user.id
        )
        db.session.add(document)
        db.session.commit()
        
        flash('Document uploadé avec succès.', 'success')
    else:
        flash('Type de fichier non autorisé.', 'error')
    
    return redirect(url_for('interventions.detail', id=intervention_id))


@documents_bp.route('/upload/equipment/<int:equipment_id>', methods=['POST'])
@login_required
def upload_equipment(equipment_id):
    """Upload un document pour un équipement."""
    from ..core.models.equipment import EquipmentDocument
    
    if 'file' not in request.files:
        flash('Aucun fichier sélectionné.', 'error')
        return redirect(url_for('equipments.detail', id=equipment_id))
    
    file = request.files['file']
    
    if file.filename == '':
        flash('Aucun fichier sélectionné.', 'error')
        return redirect(url_for('equipments.detail', id=equipment_id))
    
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        timestamp = datetime.now(timezone.utc).strftime('%Y%m%d%H%M%S')
        unique_filename = f"{timestamp}_{filename}"
        
        upload_dir = os.path.join(current_app.config.get('UPLOAD_FOLDER', 'uploads'), 'equipments')
        os.makedirs(upload_dir, exist_ok=True)
        
        filepath = os.path.join(upload_dir, unique_filename)
        file.save(filepath)
        
        document = EquipmentDocument(
            equipment_id=equipment_id,
            filename=unique_filename,
            filepath=filepath,
            file_type=filename.rsplit('.', 1)[1].lower() if '.' in filename else 'unknown',
            file_size=os.path.getsize(filepath),
            uploaded_by_id=current_user.id
        )
        db.session.add(document)
        db.session.commit()
        
        flash('Document uploadé avec succès.', 'success')
    else:
        flash('Type de fichier non autorisé.', 'error')
    
    return redirect(url_for('equipments.detail', id=equipment_id))


@documents_bp.route('/intervention/delete/<int:doc_id>', methods=['POST'])
@login_required
def delete_intervention_document(doc_id):
    """Supprimer un document d'intervention."""
    document = InterventionDocument.query.get_or_404(doc_id)
    intervention_id = document.intervention_id
    
    # Supprimer le fichier physique
    if document.filepath and os.path.exists(document.filepath):
        os.remove(document.filepath)
    
    db.session.delete(document)
    db.session.commit()
    
    flash('Document supprimé.', 'success')
    return redirect(url_for('interventions.detail', id=intervention_id))


@documents_bp.route('/intervention/<int:doc_id>/edit', methods=['POST'])
@login_required
def edit_intervention(doc_id):
    """Modifier uniquement la description d'un document d'intervention."""
    document = InterventionDocument.query.get_or_404(doc_id)
    document.description = (request.form.get('description') or '').strip() or None
    db.session.commit()
    flash('Description du document mise à jour.', 'success')
    return redirect(url_for('interventions.detail', id=document.intervention_id))


@documents_bp.route('/intervention/download/<int:doc_id>')
@login_required
def download_intervention_document(doc_id):
    """Télécharger un document d'intervention."""
    from flask import send_file
    document = InterventionDocument.query.get_or_404(doc_id)
    
    if document.filepath and os.path.exists(document.filepath):
        return send_file(document.filepath, as_attachment=True, download_name=document.filename)
    
    flash('Fichier non trouvé.', 'error')
    return redirect(url_for('interventions.detail', id=document.intervention_id))


@documents_bp.route('/equipment/delete/<int:doc_id>', methods=['POST'])
@login_required
def delete_equipment_document(doc_id):
    """Supprimer un document d'équipement."""
    from ..core.models.equipment import EquipmentDocument
    document = EquipmentDocument.query.get_or_404(doc_id)
    equipment_id = document.equipment_id
    
    # Supprimer le fichier physique
    if document.filepath and os.path.exists(document.filepath):
        os.remove(document.filepath)
    
    db.session.delete(document)
    db.session.commit()
    
    flash('Document supprimé.', 'success')
    return redirect(url_for('equipments.detail', id=equipment_id))


@documents_bp.route('/equipment/download/<int:doc_id>')
@login_required
def download_equipment_document(doc_id):
    """Télécharger un document d'équipement."""
    from flask import send_file
    from ..core.models.equipment import EquipmentDocument
    document = EquipmentDocument.query.get_or_404(doc_id)
    
    if document.filepath and os.path.exists(document.filepath):
        return send_file(document.filepath, as_attachment=True, download_name=document.filename)
    
    flash('Fichier non trouvé.', 'error')
    return redirect(url_for('equipments.detail', id=document.equipment_id))