103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
from flask import Flask, render_template, request, redirect, url_for, send_from_directory, session
|
|
from flask_socketio import SocketIO, emit
|
|
import os
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
from PIL import Image
|
|
from watchdog.observers import Observer
|
|
from watchdog.events import FileSystemEventHandler
|
|
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = 'your_secret_key' # Replace with a strong secret key
|
|
socketio = SocketIO(app)
|
|
|
|
app.config['UPLOAD_FOLDER'] = 'images' # Folder to store images
|
|
app.config['CACHE_FOLDER'] = 'cache' # Folder to store cached resized images
|
|
app.config['ALLOWED_EXTENSIONS'] = {'png', 'jpg', 'jpeg', 'gif'}
|
|
|
|
# Fixed username and password
|
|
USERNAME = 'user'
|
|
PASSWORD = generate_password_hash('password') # Hashed password
|
|
|
|
def allowed_file(filename):
|
|
return '.' in filename and \
|
|
filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS']
|
|
|
|
def resize_image(image_path, cache_path, size=(800, 450)):
|
|
with Image.open(image_path) as img:
|
|
img.thumbnail(size)
|
|
img.save(cache_path)
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return redirect(url_for('login'))
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'POST':
|
|
username = request.form['username']
|
|
password = request.form['password']
|
|
if username == USERNAME and check_password_hash(PASSWORD, password):
|
|
session['logged_in'] = True
|
|
return redirect(url_for('gallery'))
|
|
else:
|
|
error = 'Invalid credentials'
|
|
return render_template('login.html', error=error)
|
|
return render_template('login.html')
|
|
|
|
@app.route('/logout')
|
|
def logout():
|
|
session.pop('logged_in', None)
|
|
return redirect(url_for('login'))
|
|
|
|
@app.route('/gallery')
|
|
def gallery():
|
|
if not session.get('logged_in'):
|
|
return redirect(url_for('login'))
|
|
|
|
images = os.listdir(app.config['UPLOAD_FOLDER'])
|
|
cached_images = []
|
|
|
|
if not os.path.exists(app.config['CACHE_FOLDER']):
|
|
os.makedirs(app.config['CACHE_FOLDER'])
|
|
|
|
for image in images:
|
|
image_path = os.path.join(app.config['UPLOAD_FOLDER'], image)
|
|
cache_path = os.path.join(app.config['CACHE_FOLDER'], image)
|
|
|
|
if not os.path.exists(cache_path):
|
|
resize_image(image_path, cache_path)
|
|
|
|
cached_images.append(image)
|
|
|
|
return render_template('gallery.html', images=cached_images)
|
|
|
|
@app.route('/images/<filename>')
|
|
def uploaded_file(filename):
|
|
return send_from_directory(app.config['CACHE_FOLDER'], filename)
|
|
|
|
def emit_gallery_update():
|
|
socketio.emit('update_gallery')
|
|
|
|
class Watcher(FileSystemEventHandler):
|
|
def on_modified(self, event):
|
|
if not event.is_directory:
|
|
emit_gallery_update()
|
|
|
|
def on_created(self, event):
|
|
if not event.is_directory:
|
|
emit_gallery_update()
|
|
|
|
def on_deleted(self, event):
|
|
if not event.is_directory:
|
|
emit_gallery_update()
|
|
|
|
if __name__ == '__main__':
|
|
observer = Observer()
|
|
observer.schedule(Watcher(), path=app.config['UPLOAD_FOLDER'], recursive=False)
|
|
observer.start()
|
|
try:
|
|
context = ('cert.pem', 'key.pem') # Replace with your self-signed certificate and key files
|
|
socketio.run(app, debug=True, ssl_context=context)
|
|
except KeyboardInterrupt:
|
|
observer.stop()
|
|
observer.join() |