70 lines
1.8 KiB
Python
Executable File
70 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
The end user script for the modpack.
|
|
"""
|
|
import os
|
|
import json
|
|
import urllib.parse
|
|
import urllib.request
|
|
|
|
import config_slave
|
|
|
|
def sync(delete=False):
|
|
"""
|
|
Synchronize the local `mods/` folder with the remote server.
|
|
"""
|
|
remote_mods =urllib.request.urlopen(config_slave.remote_addr+'/get').read()
|
|
remote_mods = json.loads(remote_mods)
|
|
if os.path.isfile('modlist.json'):
|
|
with open('modlist.json', 'r') as file:
|
|
local_mods = json.loads(file.read())
|
|
else:
|
|
local_mods = os.listdir('mods')
|
|
|
|
local_mods_del = [mod for mod in local_mods if mod not in remote_mods]
|
|
remote_mods_get = [mod for mod in remote_mods if mod not in local_mods]
|
|
|
|
for mod in local_mods_del:
|
|
if delete:
|
|
os.remove(os.path.join('mods', mod))
|
|
else:
|
|
os.makedirs('mods_old', exist_ok=True)
|
|
os.rename(os.path.join('mods', mod), os.path.join('mods_old', mod))
|
|
print(f"Deleted {mod}")
|
|
|
|
for mod in remote_mods_get:
|
|
url = config_slave.remote_addr + '/mods/' + urllib.parse.quote(mod)
|
|
res = urllib.request.urlopen(url)
|
|
data = res.read()
|
|
with open(os.path.join('mods', mod), 'wb') as file:
|
|
file.write(data)
|
|
print(f"Downloaded {mod}")
|
|
with open('modlist.json', 'w') as file:
|
|
file.write(json.dumps(remote_mods))
|
|
print("Success!")
|
|
|
|
if __name__ == "__main__":
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="Script for synchronizing the local mods/ folder "
|
|
"with the remote server."
|
|
)
|
|
parser.add_argument(
|
|
'action',
|
|
choices=['sync'],
|
|
default='sync',
|
|
help="What action to perform."
|
|
)
|
|
parser.add_argument(
|
|
'--delete',
|
|
action='store_true',
|
|
help="By default, old mods are moved to a mods_old/ folder "
|
|
"instead of being deleted. Checking this flag will delete "
|
|
"them instead."
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
if args.action == 'sync':
|
|
sync(args.delete)
|