sharebox/server.py at main · oddfellow/sharebox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/usr/bin/env python3
#        _                        _            ___  _  _
#   __ _| | _____  _____ ___   __| | ___ _ __ / _ \| || |
#  / _` | |/ _ \ \/ / __/ _ \ / _` |/ _ \ '__| | | | || |_
# | (_| | |  __/>  < (_| (_) | (_| |  __/ |  | |_| |__   _|
#  \__,_|_|\___/_/\_\___\___/ \__,_|\___|_|   \___/   |_|
#
# Copyright (c) 2021 alexcoder04 <https://github.com/alexcoder04>
#
# sharebox - upload and download files over http
# requires: flask, werkzeug, argparse
#

from flask import Flask, render_template, send_file, request, redirect
from werkzeug.utils import secure_filename
import argparse
import os
import sys
import time

app = Flask(__name__)

# main page, files overview
@app.route("/")
def index():
    files = []
    for f in os.listdir(app.config["SHARE_DIR"]):
        path = os.path.join(app.config["SHARE_DIR"], f)
        delete_after = app.config["KEEP"]
        if not os.path.isfile(path):
            continue
        if os.path.getmtime(path) + delete_after < time.time():
            os.remove(path)
            files.append({
                "name": f,
                "expires": 0,
                "deleted": True
                })
            continue
        files.append({
            "name": f,
            "expires": int(os.path.getmtime(path) + delete_after - time.time()),
            "deleted": False
            })
    return render_template("index.html", files=files)

# upload file via http post
@app.route("/upload", methods=["POST"])
def upload():
    if "upload" not in request.files:
        return "ERROR"
    files = request.files.getlist("upload")
    for f in files:
        if f.filename is None:
            continue
        fname = secure_filename(f.filename)
        if fname == "":
            return redirect("/error/No file selected".replace(" ", "%20"))
        f.save(os.path.join(app.config["SHARE_DIR"], fname))
    return redirect(f"/?f={[i.filename for i in files]}")

# download file
@app.route("/dl/<path:file_path>")
def download(file_path):
    requested_file = os.path.abspath(os.path.join(app.config["SHARE_DIR"], file_path))
    if os.path.getmtime(requested_file) + app.config["KEEP"] < time.time():
        os.remove(requested_file)
        return redirect("/error/This file was deleted".replace(" ", "%20"))
    if not requested_file.startswith(app.config["SHARE_DIR"]):
        return "ERROR"
    if not os.path.isfile(requested_file):
        return "ERROR"
    return send_file(requested_file)

@app.route("/error/<message>")
def error(message):
    return render_template("error.html", message=message)

# ifmain
if __name__ == "__main__":
    parser = argparse.ArgumentParser(
            description="upload and download files"
            )

    parser.add_argument("--debug", action="store_true", help="run flask in debug mode")
    parser.add_argument("--keep", default=(60*5), type=int, help="delete files after so many seconds")
    parser.add_argument("--port", default=8899, type=int, help="port to listen on")
    parser.add_argument("--share-dir", default="/tmp/sharebox", type=str, help="folder to share")

    args = parser.parse_args()

    app.config["SHARE_DIR"] = args.share_dir
    app.config["KEEP"] = args.keep

    if not os.path.isdir(app.config["SHARE_DIR"]):
        try:
            print("Warning: shared folder does not exist, creating...")
            os.mkdir(app.config["SHARE_DIR"])
        except Exception:
            print("Fatal error: shared folder could not be created")
            sys.exit(1)

    app.run(port=args.port, debug=args.debug, host="0.0.0.0")