added uploading image from url
This commit is contained in:
parent
c997e68791
commit
0fa57ac7a3
3
.gitignore
vendored
3
.gitignore
vendored
|
|
@ -1,5 +1,4 @@
|
||||||
__pycache__/
|
__pycache__/
|
||||||
*/__pycache__/
|
|
||||||
secret_key
|
secret_key
|
||||||
*.db
|
*.db
|
||||||
|
*.swp
|
||||||
|
|
|
||||||
55
fileHost.py
55
fileHost.py
|
|
@ -3,11 +3,13 @@
|
||||||
Simple file host using Flask.
|
Simple file host using Flask.
|
||||||
"""
|
"""
|
||||||
import os
|
import os
|
||||||
|
import re
|
||||||
import time
|
import time
|
||||||
import atexit
|
import atexit
|
||||||
import string
|
import string
|
||||||
import secrets
|
import secrets
|
||||||
import sqlite3
|
import sqlite3
|
||||||
|
import tempfile
|
||||||
import functools
|
import functools
|
||||||
import threading
|
import threading
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
@ -16,12 +18,14 @@ from passlib.hash import argon2
|
||||||
from flask import Flask, session, request, abort, redirect, url_for, g, \
|
from flask import Flask, session, request, abort, redirect, url_for, g, \
|
||||||
render_template
|
render_template
|
||||||
from werkzeug.utils import secure_filename
|
from werkzeug.utils import secure_filename
|
||||||
|
from werkzeug.datastructures import FileStorage
|
||||||
from flask_paranoid import Paranoid
|
from flask_paranoid import Paranoid
|
||||||
from apscheduler.schedulers.background import BackgroundScheduler
|
from apscheduler.schedulers.background import BackgroundScheduler
|
||||||
|
import requests
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
app.config['MAX_CONTENT_LENGTH'] = 128 * 1024 * 1024
|
app.config['MAX_CONTENT_LENGTH'] = 128 * 1024 * 1024
|
||||||
app.config["UPLOAD_DIR"] = "/usr/local/www/html/up"
|
app.config["UPLOAD_DIR"] = "/var/www/html/up"
|
||||||
app.config["UPLOAD_URL"] = "https://steelbea.me/up/"
|
app.config["UPLOAD_URL"] = "https://steelbea.me/up/"
|
||||||
app.config["DB_NAME"] = "fileHost.db"
|
app.config["DB_NAME"] = "fileHost.db"
|
||||||
app.config["DB_LOCK"] = threading.Lock()
|
app.config["DB_LOCK"] = threading.Lock()
|
||||||
|
|
@ -153,7 +157,7 @@ def delete_file(filename):
|
||||||
def login_required(url=None):
|
def login_required(url=None):
|
||||||
"""
|
"""
|
||||||
A decorator function to protect certain endpoints by requiring the user
|
A decorator function to protect certain endpoints by requiring the user
|
||||||
to either pass a valid session cookie, or pass thier username and
|
to either pass a valid session cookie, or pass their username and
|
||||||
password along with the request to login.
|
password along with the request to login.
|
||||||
"""
|
"""
|
||||||
def actual_decorator(func):
|
def actual_decorator(func):
|
||||||
|
|
@ -334,10 +338,17 @@ def upload():
|
||||||
Saves the uploaded files and returns URLs pointing to them.
|
Saves the uploaded files and returns URLs pointing to them.
|
||||||
"""
|
"""
|
||||||
username = session.get("username")
|
username = session.get("username")
|
||||||
if not username:
|
|
||||||
username = request.form.get("user")
|
if request.form.get("url"):
|
||||||
|
try:
|
||||||
|
f = [download_file(request.form.get("url"))]
|
||||||
|
except ValueError as e:
|
||||||
|
return e
|
||||||
|
else:
|
||||||
|
f = []
|
||||||
|
|
||||||
urls = []
|
urls = []
|
||||||
for file in request.files.getlist("files"):
|
for file in request.files.getlist("files") + f:
|
||||||
fname = secure_filename(file.filename)
|
fname = secure_filename(file.filename)
|
||||||
pre = get_rand_chars(8)
|
pre = get_rand_chars(8)
|
||||||
fdir = app.config.get("UPLOAD_DIR")
|
fdir = app.config.get("UPLOAD_DIR")
|
||||||
|
|
@ -397,6 +408,40 @@ def get_rand_chars(n):
|
||||||
return "".join(chars)
|
return "".join(chars)
|
||||||
|
|
||||||
|
|
||||||
|
def download_file(url):
|
||||||
|
"""
|
||||||
|
Downloads the file at the given url while observing file size and
|
||||||
|
timeout limitations.
|
||||||
|
"""
|
||||||
|
timeout = 10
|
||||||
|
requests_kwargs = {
|
||||||
|
'stream': True,
|
||||||
|
'headers': {'User-Agent': "Steelbea.me LTD needs YOUR files."},
|
||||||
|
'timeout': timeout,
|
||||||
|
'verify': True
|
||||||
|
}
|
||||||
|
t = tempfile.TemporaryFile()
|
||||||
|
with requests.get(url, **requests_kwargs) as r:
|
||||||
|
size = 0
|
||||||
|
start_time = time.time()
|
||||||
|
for chunk in r.iter_content(102400):
|
||||||
|
if time.time() - start_time > timeout:
|
||||||
|
raise ValueError('timeout reached')
|
||||||
|
|
||||||
|
size += len(chunk)
|
||||||
|
if size > app.config['MAX_CONTENT_LENGTH']:
|
||||||
|
raise ValueError('response too large')
|
||||||
|
t.write(chunk)
|
||||||
|
if r.headers.get('Content-Disposition'):
|
||||||
|
fname = re.search(r'filename="(.+)"',
|
||||||
|
r.headers['content-disposition'])
|
||||||
|
else:
|
||||||
|
fname = os.path.basename(url)
|
||||||
|
t.seek(0)
|
||||||
|
f = FileStorage(stream=t, filename=fname, name='url')
|
||||||
|
return f
|
||||||
|
|
||||||
|
|
||||||
init()
|
init()
|
||||||
atexit.register(scheduler.shutdown)
|
atexit.register(scheduler.shutdown)
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
|
||||||
|
|
@ -4,23 +4,24 @@
|
||||||
<title>Let's uploading boys!</title>
|
<title>Let's uploading boys!</title>
|
||||||
</head>
|
</head>
|
||||||
<body>
|
<body>
|
||||||
<form method="post" enctype="multipart/form-data" action="{{ url_for('index') }}">
|
Hello, {{ session.username }}<br>
|
||||||
Hello, {{ session.username }}<br />
|
<a href="{{ url_for('change_password') }}">Change password</a><br>
|
||||||
<a href="{{ url_for('change_password') }}">Change password</a><br />
|
<a href="{{ url_for('logout') }}">Logout</a><br>
|
||||||
<a href="{{ url_for('logout') }}">Logout</a><br />
|
<a href="{{ url_for('manage_uploads') }}">Manage Uploads</a><br>
|
||||||
<a href="{{ url_for('manage_uploads') }}">Manage Uploads</a><br />
|
<a href="{{ url_for('gallery', username=session.username) }}">Public Gallery</a><br>
|
||||||
<a href="{{ url_for('gallery', username=session.username) }}">Public Gallery</a><br />
|
<form method="post" enctype="multipart/form-data" action="{{ url_for('index') }}">
|
||||||
<p>Select file to upload:
|
<p>Select file to upload:
|
||||||
<p><input type="file" name="files" required multiple/><br />
|
<p><input type="file" name="files" multiple><br>
|
||||||
<input type="checkbox" name="randname"/> Generate random filename.<br />
|
<input type="url" name="url" placeholder="URL" size="30"><br>
|
||||||
<input type="checkbox" name="delflag"/> Delete this file in <input type="number" name="delnum" min="1" max="59" value="1"/>
|
<input type="checkbox" name="randname"> Generate random filename.<br>
|
||||||
<select name="deltype">
|
<input type="checkbox" name="delflag"> Delete this file in <input type="number" name="delnum" min="1" max="59" value="1">
|
||||||
<option value="minute">Minute</option>
|
<select name="deltype">
|
||||||
<option value="hour">Hour</option>
|
<option value="minute">Minute</option>
|
||||||
<option value="day" selected="selected">Day</option>
|
<option value="hour">Hour</option>
|
||||||
<option value="week">Week</option>
|
<option value="day" selected="selected">Day</option>
|
||||||
</select>
|
<option value="week">Week</option>
|
||||||
<p><input type="submit" value="Upload File" name="submit"/>
|
</select>
|
||||||
</form>
|
<p><input type="submit" value="Upload File" name="submit">
|
||||||
|
</form>
|
||||||
</body>
|
</body>
|
||||||
</html>
|
</html>
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user