Files
drath.cc/app/main.py
2023-06-28 20:16:39 +02:00

57 lines
1.6 KiB
Python

from flask import Flask, render_template, current_app as this, Response
from google.cloud import storage
from flask_minify import Minify
app = Flask(__name__.split('.')[0])
app.jinja_env.trim_blocks = True
app.jinja_env.lstrip_blocks = True
Minify(app=app, html=True, js=True, cssless=True)
def humanSize(size: int) -> str:
prefix = [" ", " K", " M", " G", " T"]
i = 0
while size > 1024:
size /= 1024
i += 1
size = str(round(size,3-len(str(int(size)))))
if size.endswith(".0"): size = size.replace(".0","")
return size+prefix[i]+"B"
def getDownloads(bucket_name: str) -> list:
""" takes the name of a google storage bucket and returns a list of objects on it """
storage_client = storage.Client.from_service_account_json(json_credentials_path="acc.json")
blobs = []
for blob in storage_client.list_blobs(bucket_name, prefix='dload'):
if blob.size == 0: continue
blobs.append({"name": blob.name.replace("dload/",""), "size": humanSize(blob.size), "url": blob.public_url})
storage_client.close()
return blobs
@app.route('/')
def main():
return render_template("index.html")
@app.route('/codecalc')
def calc():
return render_template("code-calc.html")
@app.route('/secret')
def secret():
imgUrl = "/static/images/kroniicopter.webp"
return render_template("secret.html", imgUrl = imgUrl)
@app.route('/tests')
def test():
return render_template("test.html")
@app.route('/downloads')
def dload():
return render_template("downloads.htm", users = getDownloads("drath-public"))
if __name__ == '__main__':
app.run(host='0.0.0.0', port=8080, debug=True)