Skip to content

Instantly share code, notes, and snippets.

Created May 9, 2023 01:11
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
What would you like to do?
Flask file server, post or get files, requires api key for post. Can be used for nostr photo/video sharing.
from flask import Flask, request, send_from_directory
from werkzeug.utils import secure_filename
import os
import time
import threading
app = Flask(__name__)
# Set the upload folder and allowed extensions
app.config['UPLOAD_FOLDER'] = 'uploads'
app.config['ALLOWED_EXTENSIONS'] = {'png', 'jpg', 'jpeg', 'gif'}
app.config['API_KEY'] = ''
# Create the upload folder if it doesn't exist
if not os.path.exists(app.config['UPLOAD_FOLDER']):
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS']
#Expire old files if you want
def delete_old_files():
while True:
# Get the current time
now = time.time()
# Loop through all files in the upload folder
for filename in os.listdir(app.config['UPLOAD_FOLDER']):
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
# Check if the file is older than 30 minutes
if os.path.getmtime(filepath) < now - 1800:
# Delete the file
print(f"{filename} expired")
# Sleep for 1 minute before checking again
# Start the background thread
thread = threading.Thread(target=delete_old_files, daemon=True)
@app.route('/upload', methods=['POST'])
def upload_file():
# Check if the API key is valid
api_key = request.headers.get('X-Api-Key')
if not api_key or api_key != app.config['API_KEY']:
return {'error': 'Invalid API key'}
# Check if the post request has the file part
if 'file' not in request.files:
return {'error': 'No file part'}
file = request.files['file']
# If the user does not select a file, the browser submits an
# empty file without a filename.
if file.filename == '':
return {'error': 'No selected file'}
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
print(f"{filename} uploaded!")['UPLOAD_FOLDER'], filename))
return {'success': 'File uploaded successfully', 'url': f'URLHERE/files/{filename}'}
return {'error': 'Invalid file type'}
def uploaded_file(filename):
# Check if the file exists
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
if os.path.exists(filepath):
print(f"{filename} Gotten!")
return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
return {'error': 'File not found'}
if __name__ == '__main__':
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment