sb3Unpacker.py

"""
This program generates a dictionary and objects based on the project.json file.
It also loads the project file set in main.py.

======= CLASS INFO =======
The various files with classes are used by this program and the correct data is
set. Those are then used to build the project in main.py.
"""
import zipfile as zf
import json
import config
import target, costume, sound, block, monitor  # , broadcast
from pathlib import Path
import io
import pygame
import scratch
import sys
import os
import config
import i18n

i18n.set("locale", config.language)
i18n.set("filename_format", "{locale}.{format}")
i18n.load_path.append("lang/")
_ = i18n.t


if not config.enableDebugMessages:
    sys.stderr = open(os.devnull, "w")
if not config.enableTerminalOutput:
    sys.stdout = open(os.devnull, "w")


def sb3Unpack(sb3):
    # If project does not exist, quit with exit code 1
    if not Path(sb3).exists():
        raise FileNotFoundError(_("project-file-not-found"))

    print(_("loading-project"))
    project = zf.ZipFile(sb3, "r")
    projectJSON = json.loads(project.read("project.json"))
    targets = []

    # Generate the dictionary based on the contents of project.json
    for targetObj in projectJSON["targets"]:
        t = target.Target()

        # Set sprite values
        if "x" in targetObj:
            t.x = targetObj["x"]
            t.y = targetObj["y"]
            t.direction = targetObj["direction"]
            t.size = targetObj["size"]
            t.rotationStyle = targetObj["rotationStyle"]
        t.currentCostume = targetObj["currentCostume"]
        t.isStage = targetObj["isStage"]
        t.layerOrder = targetObj["layerOrder"]
        t.name = targetObj["name"]

        # Get costumes
        for costumeObj in targetObj["costumes"]:
            c = costume.Costume()
            if "md5ext" in costumeObj:
                c.md5ext = costumeObj["md5ext"]
            c.rotationCenterX, c.rotationCenterY = costumeObj["rotationCenterX"], costumeObj["rotationCenterY"]
            c.dataFormat = costumeObj["dataFormat"]
            c.offset = pygame.math.Vector2(c.rotationCenterX, c.rotationCenterY)
            c.file = project.read(costumeObj["assetId"] + "." + costumeObj["dataFormat"])
            c.name = costumeObj["name"]
            if costumeObj["dataFormat"] != "svg":
                c.bitmapResolution = int(costumeObj["bitmapResolution"])
            else:
                c.bitmapResolution = 1
            t.costumes.append(c)

        # Get sounds
        for soundObj in targetObj["sounds"]:
            s = sound.Sound()
            if "md5ext" in soundObj:
                s.md5ext = soundObj["md5ext"]
            s.dataFormat = soundObj["dataFormat"]
            s.rate = soundObj["rate"]
            s.sampleCount = soundObj["sampleCount"]
            s.file = project.read(soundObj["assetId"] + "." + soundObj["dataFormat"])
            s.name = soundObj["name"]
            t.sounds.append(s)

        # Get variables
        for variable in targetObj["variables"]:
            t.variables[variable] = targetObj["variables"][variable]

        # Set blocks to their correct values
        for blockId, blockObj in targetObj["blocks"].items():
            b = block.Block()
            b.blockID = blockId
            b.opcode = blockObj["opcode"]
            b.next = blockObj["next"]
            b.parent = blockObj["parent"]
            b.shadow = blockObj["shadow"]
            b.topLevel = blockObj["topLevel"]
            b.inputs = blockObj["inputs"]
            b.fields = blockObj["fields"]
            if b.opcode == "procedures_call":
                b.proccode = blockObj["mutation"]["proccode"]
                b.warp = blockObj["mutation"]["warp"]
            b.blockRan = False
            b.target = t
            t.blocks[blockId] = b
        for blockId, blockObj in targetObj["blocks"].items():
            b = t.blocks[blockId]
            if b.topLevel:
                b.top = b
            elif b.parent:
                b.top = t.blocks[b.parent].top
            else:
                b.top = None
        targets.append(t)

    return targets, project