ModpackUpdater/Backend.py

211 lines
6.5 KiB
Python
Raw Normal View History

2024-06-20 08:30:57 -06:00
from ftplib import FTP_TLS as FTP
import io
import json
import tempfile
import subprocess
import time
import os
import datetime
import zipfile
2024-06-20 16:09:18 -06:00
from customtkinter import *
import time
import math
2024-06-20 08:30:57 -06:00
def perform_installation(instance_name, prism_command, prism_instance_path, pack):
try:
ftp = FTP("gitea.piwalker.net")
ftp.login()
ftp.prot_p()
ftp.cwd(pack)
2024-06-20 08:30:57 -06:00
# Fetching versions.json from FTP
bio = io.BytesIO()
ftp.retrbinary("RETR versions.json", bio.write)
bio.seek(0)
versions = json.load(bio)
bio.close()
2024-06-20 08:30:57 -06:00
# Checking current version
version = "0.0.0"
version_file_path = os.path.join(prism_instance_path, instance_name, ".minecraft", "version.txt")
if os.path.exists(version_file_path):
with open(version_file_path, 'r') as fp:
version = fp.readline().rstrip()
2024-06-20 08:30:57 -06:00
# Checking if update is needed
if version != versions[-1]["Version"]:
print(f"Current version: {version}")
print(f"Latest version: {versions[-1]['Version']}")
2024-06-20 08:30:57 -06:00
# Downloading modpack
with tempfile.TemporaryDirectory() as temp_dir:
modpack_file_path = os.path.join(temp_dir, instance_name + ".mrpack")
with open(modpack_file_path, 'wb') as modpack:
2024-06-20 16:09:18 -06:00
ftpDownload(ftp, versions[-1]["File"], modpack)
# Running PrismLauncher with modpack
subprocess.Popen([prism_command, '-I', modpack_file_path])
2024-06-20 08:30:57 -06:00
# Waiting for installation to complete
while True:
time.sleep(5)
if os.path.exists(version_file_path):
with open(version_file_path, 'r') as fp:
version = fp.readline().rstrip()
if version == versions[-1]["Version"]:
break
2024-06-20 16:09:18 -06:00
except Exception as e:
print(e)
print("Unable to check for updates. Modpack may be out of date")
finally:
# Launching PrismLauncher with the instance
subprocess.run([prism_command, '-l', instance_name])
2024-06-20 08:30:57 -06:00
def upload_pack(username, password, version_tag, fileName, pack):
2024-06-20 08:30:57 -06:00
ftp = FTP("gitea.piwalker.net", username, password)
ftp.prot_p()
ftp.cwd(pack)
2024-06-20 08:30:57 -06:00
bio = io.BytesIO()
ftp.retrbinary("RETR versions.json", bio.write)
bio.seek(0)
versions = json.load(bio)
bio.close()
time = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
versions.append({"Version":version_tag, "Date":time, "File":"Versions/"+pack+time+".mrpack"})
with zipfile.ZipFile(fileName, 'r') as zin:
zipbytes = io.BytesIO()
zout = zipfile.ZipFile(zipbytes, 'w')
for item in zin.infolist():
buffer = zin.read(item.filename)
if item.filename != "overrides/version.txt":
zout.writestr(item, buffer)
zout.writestr("overrides/version.txt", version_tag)
zout.close()
zipbytes.seek(0)
ftp.storbinary("STOR "+versions[len(versions)-1]["File"], zipbytes)
2024-06-20 08:30:57 -06:00
bio = io.BytesIO()
bio.write(json.dumps(versions).encode())
bio.seek(0)
ftp.storbinary("STOR versions.json", bio)
bio.close()
modpackUpdate(pack, ftp)
2024-06-20 08:30:57 -06:00
ftp.close()
def getModpacks():
ftp = FTP("gitea.piwalker.net")
ftp.login()
ftp.prot_p()
bio = io.BytesIO()
ftp.retrbinary("RETR modpacks.json", bio.write)
bio.seek(0)
ftp.close()
return json.load(bio)
def uploadModpacks(modpacks, ftp):
ftp.cwd("/ftp");
bio = io.BytesIO()
bio.write(json.dumps(modpacks).encode())
bio.seek(0)
ftp.storbinary("STOR modpacks.json", bio)
bio.close()
def modpackUpdate(id, ftp):
modpacks = getModpacks()
time = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
for modpack in modpacks:
if modpack["id"] == id:
modpack["last-updated"] = time
uploadModpacks(modpacks, ftp);
def createModpack(id, name, username, password):
modpacks = getModpacks()
time = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
modpacks.append({"id":id, "name":name, "last-updated":time})
ftp = FTP("gitea.piwalker.net", username, password)
ftp.prot_p()
uploadModpacks(modpacks,ftp)
ftp.mkd(id)
ftp.cwd(id)
versions = []
bio = io.BytesIO()
bio.write(json.dumps(versions).encode())
bio.seek(0)
ftp.storbinary("STOR versions.json", bio)
ftp.mkd("Versions")
ftp.close()
2024-06-20 16:09:18 -06:00
def deleteModpack(username, password, id):
ftp = FTP("gitea.piwalker.net", username, password)
ftp.prot_p()
modpacks = getModpacks()
for pack in modpacks:
if pack["id"] == id:
modpacks.remove(pack)
break
uploadModpacks(modpacks, ftp)
deleteFolder(ftp, id)
def deleteFolder(ftp, path):
print("Deleting folder: "+path)
ftp.cwd(path)
for item in ftp.nlst():
try:
print("deleting file: "+path)
ftp.delete(item)
except:
deleteFolder(ftp, item)
ftp.cwd("..")
ftp.rmd(path)
def ftpDownload(ftp, file, stream):
#create tkinter window
print("downloading file: "+file)
dialog = CTk()
dialog.attributes('-type', 'dialog')
dialog.title("Downloading Modpack")
set_appearance_mode("dark")
set_default_color_theme("blue")
2024-06-20 16:09:18 -06:00
global pbar
pbar = CTkProgressBar(master=dialog)
pbar.pack(padx=20, pady=20)
global progress
global ETA
progress = StringVar()
ETA = StringVar()
progress_label = CTkLabel(master=dialog, textvariable=progress)
progress_label.pack()
eta_label = CTkLabel(master=dialog, textvariable=ETA)
eta_label.pack()
size = ftp.size(file)
global total
total = 0
global start
global timer
timer = 0
start = time.time()
dialog.update()
2024-06-20 16:09:18 -06:00
def downloadCallback(data):
global total
global pbar
global start
global progress
global ETA
global timer
stream.write(data)
total += len(data)
if time.time() - timer >= 1:
progress.set(str(round(total/1048576, 1))+" MB / "+str(round(size/1048576, 1))+" MB @ " + str(round((total/1048576)/(time.time()-start), 3))+" MB/s")
time_left = (size-total)/(total/(time.time()-start))
ETA.set("ETA: " + str(datetime.timedelta(seconds=math.ceil(time_left))))
pbar.set(total/size)
timer = time.time()
2024-06-20 16:17:24 -06:00
dialog.update()
2024-06-20 16:09:18 -06:00
ftp.retrbinary("RETR " + file, downloadCallback)
2024-06-20 16:55:21 -06:00
dialog.destroy()