overwrought_master.py

@@ -251,34 +251,22 @@ def post_pack():
 	res.raise_for_status()
 	mods_remote = res.json()
 	mods = [mod for mod in mods if mod.filename not in mods_remote]
-
+	files = {}
 	for mod in mods:
-		files = {}
 		files[mod.filename] = open(os.path.join('mods', mod.filename), 'rb')
-		res = requests.post(
-			url=config_master.remote_addr + '/post',
-			data=data,
-			files=files
-		)
-		res.raise_for_status()
-		files[mod.filename].close()
-		print(f"{mod.filename} uploaded.")
-
-
-	files = {}
-	files['modpack.db'] = open('modpack.db', 'rb')
 	files[config_master.modpack_name + '.html'] = open(
-		config_master.modpack_name +'.html', 'rb')
+		config_master.modpack_name, 'rb')
+
 	res = requests.post(
 		url=config_master.remote_addr + '/post',
 		data=data,
 		files=files
 	)
 	res.raise_for_status()
+
 	for file in files.values():
 		file.close()
-	print("modpack.db uploaded.")
+
-	print(f"{config_master.modpack_name}.html uploaded.")
 	print('Success!')
 
 

overwrought_server.py

@@ -2,7 +2,6 @@
 """
 The overwrought server, for exchanging mods from master to slave recipients.
 """
-import os
 import json
 import sqlite3
 import threading
@@ -14,22 +13,6 @@ import config_server
 
 app = Flask(__name__)
 app.config['db_lock'] = threading.Lock()
-os.makedirs('mods', exist_ok=True)
-
-def init_db():
-	"""
-	If no database is found, a dummy database is created to allow the
-	initial modpack upload to go smoothly.
-	"""
-	con = sqlite3.connect('modpack.db')
-	cur = con.cursor()
-	try:
-		cur.execute("SELECT filename FROM mod").fetchone()
-	except sqlite3.OperationalError:
-		cur.execute("CREATE TABLE mod(filename TEXT)")
-		con.commit()
-	con.close()
-init_db()
 
 @app.route('/')
 def index():
@@ -64,20 +47,17 @@ def post():
 	if not argon2.verify(password, config_server.pw_hash):
 		abort(401)
 
-	files = request.files.to_dict()
-	if 'modpack.db' in files:
 	db = request.files.get('database')
 	app.config.get('db_lock').acquire()
 	db.save('modpack.db')
 	app.config.get('db_lock').release()
-		files.pop('modpack.db')
 
-	if config_server.modpack_name + '.html' in files:
 	summary = request.files.get(config_server.modpack_name + '.html')
 	summary.save(config_server.modpack_name + '.html')
-		files.pop(config_server.modpack_name + '.html')
 
-	for fname, file in files.items():
+	for fname, file in request.files.items():
+		if not fname.endswith('.jar'):
+			continue
 		file.save(os.path.join(config_server.mods_dir, fname))
 
 def generate_hash(password):