Utility.py

def init():
    global Loading
    Loading = False


def IsWritable(path):
    import os

    try:
        f = open(path + "/FOLON-Downgrader-TestFile", "w")
        f.write("Testing...")
        f.close()
        f = open(path + "/FOLON-Downgrader-TestFile", "r")
        f.read()
        f.close()
        os.remove(path + "/FOLON-Downgrader-TestFile")
        return True
    except:
        return False


def WhereSteam():
    import os
    from pathlib import Path

    home = str(Path.home())

    SteamLocations = []
    steamfounds = 0

    if os.path.isdir("C:/Program Files/Steam/steamapps/common"):
        steamfounds += 1
        SteamLocations.append("C:/Program Files/Steam/steamapps/common")
    if os.path.isdir("C:/Program Files (x86)/Steam/steamapps/common"):
        steamfounds += 1
        SteamLocations.append("C:/Program Files (x86)/Steam/steamapps/common")
    if os.path.isdir(
        f"{home}/.var/app/com.valvesoftware.Steam/.local/share/Steam/steamapps/common"
    ):
        steamfounds += 1
        SteamLocations.append(
            f"{home}/.var/app/com.valvesoftware.Steam/.local/share/Steam/steamapps/common"
        )
    if os.path.isdir(f"{home}/.steam/steam/steamapps/common"):
        steamfounds += 1
        SteamLocations.append(f"{home}/.steam/steam/steamapps/common")

    if steamfounds == 0:
        SteamLocations.append(home)
        return SteamLocations
    else:
        return SteamLocations


def resource_path(relative_path):
    import sys
    import os

    """ Get absolute path to resource, works for dev and for PyInstaller """
    base_path = getattr(sys, "_MEIPASS", os.path.dirname(os.path.abspath(__file__)))
    if IsWindows():
        return os.path.join(base_path, relative_path).replace("/", "\\")
    else:
        return os.path.join(base_path, relative_path)


def IsBundled():
    import sys

    if getattr(sys, "frozen", False) and hasattr(sys, "_MEIPASS"):
        return True
    else:
        return False


def IsWindows():
    import sys

    if hasattr(sys, "getwindowsversion"):
        return True
    else:
        return False


def CountFiles(Directory):
    import os

    try:
        Directory = str(Directory)
        if os.path.isdir(Directory):
            file_count = sum(len(files) for _, _, files in os.walk(Directory))
            return file_count
        else:
            return False
    except:
        return False


def Write_Settings(settings):
    import json
    import os

    if not os.path.isdir("FOLON-Downgrader-Files"):
        os.mkdir("FOLON-Downgrader-Files")

    with open("FOLON-Downgrader-Files/Settings.json", "w") as f:
        json.dump(settings, f)

    f = open("FOLON-Downgrader-Files/Settings.json")
    LocalSettings = json.load(f)
    f.close()
    if settings != LocalSettings:
        raise Exception("SETTINGS WRONG")


def Read_Settings():
    import json

    # Read Settings
    try:
        f = open("FOLON-Downgrader-Files/Settings.json")

        # returns JSON object as
        # a dictionary
        settings = json.load(f)

        # Closing file
        f.close()
    except:
        settings = {
            "LoginResult": "",
            "Steps": 4,
            "SteamPath": "",
        }

    return settings


def CleanUp(*args):
    import shutil
    import os
    from time import sleep

    Path = Read_Settings()["SteamPath"]

    if Path == "":
        Path = WhereSteam()[0]

    try:
        shutil.rmtree(Path + "/SteamFiles")
    except:
        pass
    try:
        shutil.rmtree("__pycache__")
    except:
        pass
    os.mkdir("AAAAAAAAAAAAAA")


def BlockUpdates():
    import os
    import stat

    for i in WhereSteam():
        try:
            FilePath = i.replace("/common", "/appmanifest_377160.acf")
            print(FilePath)
            os.chmod(FilePath, stat.S_IREAD | stat.S_IRGRP | stat.S_IROTH)
        except:
            pass


def oops(type, value, tb):
    import sys
    import traceback
    from tkinter.messagebox import showerror

    showerror("Error", "\n".join(traceback.format_exception(type, value, tb)))
    sys.exit(1)


def MoveFiles(srcdir, dstdir):
    from os import path, listdir, mkdir
    from shutil import move, rmtree

    print("Started:", srcdir)
    if not path.isdir(dstdir):
        mkdir(dstdir)

    for a in listdir(srcdir):
        if path.isdir(f"{srcdir}/{a}"):
            if not path.isdir(f"{dstdir}/{a}"):
                mkdir(f"{dstdir}/{a}")

            for b in listdir(f"{srcdir}/{a}"):
                if path.isdir(f"{srcdir}/{a}/{b}"):
                    if not path.isdir(f"{dstdir}/{a}/{b}"):
                        mkdir(f"{dstdir}/{a}/{b}")

                    for c in listdir(f"{srcdir}/{a}/{b}"):
                        move(
                            f"{srcdir}/{a}/{b}/{c}",
                            f"{dstdir}/{a}/{b}/{c}",
                        )
                        print(f"moved: {a}/{b}/{c}")
                else:
                    if not path.isdir(f"{dstdir}/{a}"):
                        mkdir(f"{dstdir}/{a}")
                    move(
                        f"{srcdir}/{a}/{b}",
                        f"{dstdir}/{a}/{b}",
                    )
                    print(f"moved: {a}/{b}")
        else:
            move(f"{srcdir}/{a}", f"{dstdir}/{a}")
        print("moved:", a)
    rmtree(srcdir)
    print("Finished:", srcdir)


def bytesto(bytes, to, bsize=1024):
    """convert bytes to megabytes, etc.
    sample code:
        print('mb= ' + str(bytesto(314575262000000, 'm')))
    sample output:
        mb= 300002347.946
    """

    a = {"k": 1, "m": 2, "g": 3, "t": 4, "p": 5, "e": 6}
    r = float(bytes)
    for i in range(a[to.lower()]):
        r = r / bsize

    return r


def IsBinaryAvilable(Binary):
    try:
        import subprocess

        subprocess.call(Binary)
    except:
        from tkinter import messagebox

        messagebox.showerror(
            "Missing Binary",
            f"Error: {Binary} is not available, please install it depending on OS",
        )


if __name__ == "__main__":
    print(WhereSteam())
    print(CountFiles("."))
    print(resource_path("."))
    print(IsBundled())
    print(IsWindows())