alvierahman90/uploadr
1
0
Fork 0
mirror of https://github.com/alvierahman90/uploadr.git synced 2024-10-22 14:11:53 +00:00
Code Issues Projects Releases Wiki Activity

Formatting.

Browse Source
This commit is contained in:
Akbar Rahman 2018-05-27 16:02:37 +01:00
parent 24d6babc73
commit 16af2b2424
Signed by: alvierahman90
GPG Key ID: 20609519444A1269
1 changed files with 32 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
uploadr.py
View File

@ -17,15 +17,18 @@ app.config['SECRET_KEY'] = os.urandom(24)
encode = urllib.parse.quote encode = urllib.parse.quote
decode = urllib.parse.unquote decode = urllib.parse.unquote
def check_create(dir): def check_create(dir):
if not os.path.isdir(dir): if not os.path.isdir(dir):
os.makedirs(dir) os.makedirs(dir)
check_create(config.uploads) check_create(config.uploads)
check_create(config.qrcodes) check_create(config.qrcodes)
@app.route("/") @app.route("/")
@app.route('/upload', methods=['GET','POST']) @app.route('/upload', methods=['GET', 'POST'])
def upload_file(): def upload_file():
if request.method == 'POST': if request.method == 'POST':
f = request.files['file'] f = request.files['file']
@ -34,42 +37,40 @@ def upload_file():
file_dir = config.uploads + '/' + filename file_dir = config.uploads + '/' + filename
check_create(file_dir) check_create(file_dir)
check_create(file_dir + '/' + time_uploaded) check_create(file_dir + '/' + time_uploaded)
f.save(config.uploads f.save(config.uploads
+ '/' + encode(filename) + '/' + encode(filename)
+ '/' + time_uploaded + '/' + time_uploaded
+ '/' + filename) + '/' + filename)
return render_template('upload_success.html', return render_template('upload_success.html',
link = filename + '?version=' + time_uploaded, link=filename + '?version=' + time_uploaded,
port = config.port, port=config.port,
) )
else: else:
return render_template('upload.html', return render_template('upload.html', port=config.port)
port = config.port,
)
@app.route('/qrcodes/<filename>') @app.route('/qrcodes/<filename>')
def qrcodes(filename): def qrcodes(filename):
link = 'http://{0}:{1}/download/{2}?version={3}'.format( link = 'http://{0}:{1}/download/{2}?version={3}'.format(
config.domain config.domain, config.port, encode(filename),
, config.port request.args.get('version')
, encode(filename)
, request.args.get('version')
) )
image = qrcode.make(link) image = qrcode.make(link)
with open(config.qrcodes + '/' + filename+ '.png', mode = 'bw+') as file: with open(config.qrcodes + '/' + filename + '.png', mode='bw+') as file:
image.save(file) image.save(file)
return send_from_directory(config.qrcodes, filename + '.png', as_attachment=True) return send_from_directory(config.qrcodes, filename + '.png',
as_attachment=True)
@app.route('/qr/<filename>') @app.route('/qr/<filename>')
def qr(filename=None): def qr(filename=None):
return render_template('qr.html' return render_template('qr.html', imagesrc='http://{0}:{1}/qrcodes/{2}?'
, imagesrc = 'http://{0}:{1}/qrcodes/{2}?version={3}'.format( 'version={3}'.format(
config.domain config.domain, config.port, filename,
, config.port request.args.get('version')
, filename )
, request.args.get('version') )
)
)
@app.route('/download/<filename>') @app.route('/download/<filename>')
def download(filename=None): def download(filename=None):
@ -80,18 +81,22 @@ def download(filename=None):
existing_versions = glob.glob(directory_base + '*') existing_versions = glob.glob(directory_base + '*')
existing_versions.sort() existing_versions.sort()
latest_version = existing_versions[-1] latest_version = existing_versions[-1]
directory = latest_version directory = latest_version
return flask.send_from_directory(directory, decode(filename),
as_attachment=True)
return flask.send_from_directory(directory, decode(filename), as_attachment=True)
@app.route('/css/<filename>') @app.route('/css/<filename>')
def css(filename=None): def css(filename=None):
return flask.send_from_directory('css', filename) return flask.send_from_directory('css', filename)
@app.route('/fonts/<filename>') @app.route('/fonts/<filename>')
def fonts(filename=None): def fonts(filename=None):
return flask.send_from_directory('fonts', filename) return flask.send_from_directory('fonts', filename)
if __name__ == '__main__': if __name__ == '__main__':
app.debug = True app.debug = True
app.run() app.run()