mirror of
https://github.com/dobin/avred
synced 2026-06-08 13:54:13 +00:00
Dobin Rutishauser
55 lines
1.7 KiB
Python
Executable File
55 lines
1.7 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import os
|
|
import argparse
|
|
from flask import Flask
|
|
|
|
from config import Config
|
|
from app.views import views
|
|
from app.views_upload import views_upload
|
|
from app.views_auth import views_auth, login_manager
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--listenip', type=str, help='IP to listen on', default="0.0.0.0")
|
|
parser.add_argument('--listenport', type=int, help='Port to listen on', default=5000)
|
|
parser.add_argument('--debug', action='store_true', help='Debug', default=False)
|
|
args = parser.parse_args()
|
|
|
|
config = Config()
|
|
config.load()
|
|
root_folder = os.path.dirname(__file__)
|
|
app_folder = os.path.join(root_folder, 'app')
|
|
|
|
app = Flask(__name__,
|
|
static_folder=os.path.join(app_folder, 'static'),
|
|
template_folder=os.path.join(app_folder, 'templates')
|
|
)
|
|
|
|
app.config['UPLOAD_FOLDER'] = os.path.join(app_folder, 'upload')
|
|
app.config['EXAMPLE_FOLDER'] = os.path.join(app_folder, 'examples')
|
|
|
|
app.config['TEMPLATES_AUTO_RELOAD'] = True
|
|
app.config['SECRET_KEY'] = os.urandom(24)
|
|
app.config['SESSION_TYPE'] = 'filesystem'
|
|
app.config['AVRED_SERVERS'] = config.get('server')
|
|
app.config['PASSWORD'] = config.get('password')
|
|
app.config['WebMaxFileSizeMb'] = config.get('WebMaxFileSizeMb')
|
|
app.config['AVRED_SCANNER'] = os.path.join(root_folder, 'avred.py')
|
|
app.config['ALLOWED_EXTENSIONS'] = { 'exe', 'dll', 'ps1', 'docm', 'bin', 'lnk' }
|
|
|
|
app.config.from_prefixed_env()
|
|
|
|
for key in ('UPLOAD_FOLDER', 'EXAMPLE_FOLDER', 'ALLOWED_EXTENSIONS',):
|
|
print("{}: {}".format(key, app.config[key]))
|
|
print("")
|
|
|
|
app.register_blueprint(views)
|
|
app.register_blueprint(views_upload)
|
|
app.register_blueprint(views_auth)
|
|
|
|
login_manager.init_app(app)
|
|
|
|
app.run(host=args.listenip, port=args.listenport, debug=args.debug)
|