show uploaded document

This commit is contained in:
2025-03-01 12:12:09 +08:00
parent 706bb97460
commit 579b67d511
+10 -3
View File
@@ -1,10 +1,10 @@
from flask import render_template, request, redirect, url_for from flask import render_template, request, redirect, url_for, send_from_directory
from flask_login import login_user, login_required, logout_user, current_user from flask_login import login_user, login_required, logout_user, current_user
from app import app, db, bcrypt, login_manager from app import app, db, bcrypt, login_manager
from models import User, Listing, Booking, Image, Document, Amenities from models import User, Listing, Booking, Image, Document, Amenities
from datetime import datetime from datetime import datetime
from utils import generate_password, send_email, save_document from utils import generate_password, send_email, save_document
import re import re, os
@login_manager.user_loader @login_manager.user_loader
def load_user(user_id): def load_user(user_id):
@@ -259,4 +259,11 @@ def view_listing(listing_id):
listing = Listing.query.get_or_404(listing_id) listing = Listing.query.get_or_404(listing_id)
property_type_display = Listing.PROPERTY_TYPES.get(listing.property_type, 'Unknown') property_type_display = Listing.PROPERTY_TYPES.get(listing.property_type, 'Unknown')
amenities = list(listing.amenities) amenities = list(listing.amenities)
return render_template('view.html', listing=listing, property_type_display=property_type_display, amenities=amenities) return render_template('view.html', listing=listing, property_type_display=property_type_display, amenities=amenities)
@login_required
@app.route('/uploads/documents/<filename>')
def uploaded_file(filename):
if current_user.is_authenticated and current_user.is_host:
return send_from_directory(os.path.join(app.root_path, 'uploads/documents'), filename)
return "Access Denied", 403