Sharelock / app.py
mike23415's picture
Create app.py
9cb31ae verified
raw
history blame
1.45 kB
from flask import Flask, request, jsonify
from flask_cors import CORS
from datetime import datetime, timedelta
import uuid
import threading
import time
app = Flask(__name__)
CORS(app)
storage = {}
@app.route('/api/store', methods=['POST'])
def store():
req = request.get_json()
data = req.get('data')
ttl = int(req.get('ttl', 3600)) # default 1 hour
view_once = bool(req.get('view_once', False))
secret_id = str(uuid.uuid4())[:8]
expire_at = datetime.utcnow() + timedelta(seconds=ttl)
storage[secret_id] = {
'data': data,
'expire_at': expire_at,
'view_once': view_once,
}
return jsonify({'id': secret_id})
@app.route('/api/fetch/<secret_id>', methods=['GET'])
def fetch(secret_id):
secret = storage.get(secret_id)
if not secret:
return jsonify({'error': 'Not found'}), 404
if datetime.utcnow() > secret['expire_at']:
del storage[secret_id]
return jsonify({'error': 'Expired'}), 410
data = secret['data']
if secret['view_once']:
del storage[secret_id]
return jsonify({'data': data})
def cleanup():
while True:
now = datetime.utcnow()
expired_keys = [k for k, v in storage.items() if now > v['expire_at']]
for k in expired_keys:
del storage[k]
time.sleep(60)
threading.Thread(target=cleanup, daemon=True).start()
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)