You've already forked essentialsubmit
mirror of
https://github.com/nh-server/essentialsubmit.git
synced 2026-04-30 11:25:38 -07:00
ManiacOfGitHub
104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
from flask import Flask, request, make_response, render_template, send_from_directory, url_for
|
|
from flask_httpauth import HTTPBasicAuth
|
|
from configparser import ConfigParser
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
class AdminEntry:
|
|
def __init__(self, name, sdserial, nandserial, twlnserial, secinfoserial, date):
|
|
self.name = name
|
|
self.retrieveurl = url_for('retrieve', username=name)
|
|
self.deleteurl = url_for('delete', username=name)
|
|
self.sdserial = sdserial
|
|
self.nandserial = nandserial
|
|
self.twlnserial = twlnserial
|
|
self.secinfoserial = secinfoserial
|
|
self.date = date
|
|
|
|
def get_essential_list():
|
|
essential_entries = []
|
|
for file in os.listdir("essentials"):
|
|
if not file.endswith(".txt"):
|
|
try:
|
|
with open(f"essentials/{file}.serials.txt", "r") as f:
|
|
arr = f.readlines()
|
|
essential_entries.append(AdminEntry(file[10:].split('.exefs')[0], arr[0], arr[1], arr[2], arr[3], arr[4]))
|
|
except FileNotFoundError:
|
|
essential_entries.append(AdminEntry(file[10:].split('.exefs')[0], "", "", "", "", ""))
|
|
return essential_entries
|
|
|
|
if not "config.ini" in os.listdir():
|
|
print("Configuration doesn't exist")
|
|
sys.exit(4)
|
|
|
|
config = ConfigParser()
|
|
config.read("config.ini")
|
|
config_username = config["Access_Credentials"]["username"]
|
|
config_password = config["Access_Credentials"]["password"]
|
|
if config_password == "<password here>":
|
|
print("Password not set (too dangerous)")
|
|
sys.exit(4)
|
|
|
|
if not "essentials" in os.listdir():
|
|
print("Create an \"essentials\" folder")
|
|
sys.exit(4)
|
|
|
|
app = Flask(__name__)
|
|
auth = HTTPBasicAuth()
|
|
|
|
@app.route('/favicon.ico')
|
|
def favicon():
|
|
return send_from_directory(os.path.join(app.root_path, 'static'),
|
|
'favicon.ico', mimetype='image/vnd.microsoft.icon')
|
|
|
|
@auth.verify_password
|
|
def verify_password(username, password):
|
|
if config_password == password and config_username == username:
|
|
return username
|
|
|
|
@app.route('/submit', methods=['POST'])
|
|
def get_submission():
|
|
curtime = int(time.time())
|
|
print(f"essential.exefs submitted by {request.form['discordhandle']}")
|
|
with open(f"essentials/essential_{request.form['discordhandle']}.exefs", "wb") as f:
|
|
f.write(request.files['file'].read())
|
|
|
|
with open(f"essentials/essential_{request.form['discordhandle']}.exefs.serials.txt", "w") as f:
|
|
f.write(f"{request.form['sd']}\n{request.form['nand']}\n{request.form['twln']}\n{request.form['secinfo']}\n{curtime}")
|
|
|
|
return "<p>Hi</p>"
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return render_template('index.html')
|
|
|
|
@app.route('/admin')
|
|
@auth.login_required
|
|
def admin():
|
|
return render_template('admin.html', entries=get_essential_list())
|
|
|
|
@app.route('/retrieve', methods=['GET'])
|
|
@auth.login_required
|
|
def retrieve():
|
|
name = request.args['username']
|
|
print(f"{name} retrieved by {request.remote_addr}")
|
|
with open(f"essentials/essential_{name}.exefs", "rb") as f:
|
|
essential = f.read()
|
|
response = make_response(essential)
|
|
response.headers.set('Content-Type', 'application/octet-stream')
|
|
response.headers.set('Content-Disposition', 'attachment', filename=f'essential_{name}.exefs')
|
|
return response
|
|
|
|
@app.route('/delete')
|
|
@auth.login_required
|
|
def delete():
|
|
name = request.args['username']
|
|
print(f"{name} deleted by {request.remote_addr}")
|
|
os.remove(f"essentials/essential_{name}.exefs")
|
|
try:
|
|
os.remove(f"essentials/essential_{name}.exefs.serials.txt")
|
|
except FileNotFoundError:
|
|
pass
|
|
return "Deleted successfully. <a href=\"/admin\">Return to Admin Panel</a>"
|