64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
import os
|
|
from bottle import route, request, static_file, run, redirect, HTTPResponse, hook
|
|
from random import randint
|
|
from os import remove
|
|
|
|
from convert import convert_dhl, convert_hermes
|
|
|
|
@route('/')
|
|
def root():
|
|
return static_file('index.html', root='templates')
|
|
|
|
@route('/dhl', method='POST')
|
|
def dhl():
|
|
upload = request.files.get('upload')
|
|
if upload.content_type != "application/pdf":
|
|
return HTTPResponse(status=400, body="Bad file")
|
|
|
|
id = randint(100000,999999)
|
|
upload.save(f"./uploads/{id}.pdf")
|
|
convert_dhl(id)
|
|
print(f"converted {upload.filename} as {id}.pdf")
|
|
# delete upload file
|
|
remove(f"./uploads/{id}.pdf")
|
|
|
|
return redirect(f"/download/{id}")
|
|
|
|
@route('/hermes', method='POST')
|
|
def hermes():
|
|
upload = request.files.get('upload')
|
|
if upload.content_type != "application/pdf":
|
|
return HTTPResponse(status=400, body="Bad file")
|
|
|
|
id = randint(100000,999999)
|
|
upload.save(f"./uploads/{id}.pdf")
|
|
convert_hermes(id)
|
|
print(f"converted {upload.filename} as {id}.pdf")
|
|
# delete upload file
|
|
remove(f"./uploads/{id}.pdf")
|
|
|
|
return redirect(f"/download/{id}")
|
|
|
|
@route('/download/<id>')
|
|
def download(id):
|
|
filename = f"{id}.pdf"
|
|
return static_file(filename, root='downloads', download=filename)
|
|
|
|
@hook('after_request')
|
|
def delFiles():
|
|
if request.path.startswith("/download/"):
|
|
id = request.path.split("/")[2]
|
|
try:
|
|
remove(f"./downloads/{id}.pdf")
|
|
except FileNotFoundError:
|
|
return
|
|
|
|
@route('/static/<filename:path>')
|
|
def send_static(filename):
|
|
return static_file(filename, root='static/')
|
|
|
|
if __name__ == '__main__':
|
|
for p in ['./uploads', './downloads']:
|
|
if not os.path.exists(p):
|
|
os.makedirs(p)
|
|
run(host='0.0.0.0', port=8080)
|