scritcher/scritcher/executer.py

import os
import shlex
import tempfile
import shutil

from pathlib import Path

from .utils import load_file_path
from .error import InterpreterError


class Interpreter:
    """Interpreter for scritcher instructions/statements."""
    def __init__(self):
        self.orig_path = None
        self.loaded_path = None

    def _cmd_noop(self):
        pass

    def _cmd_load(self, loadpath: str):
        source_path = load_file_path(loadpath)
        self.orig_path = source_path

        # create a temporary file to hold bmp data
        handle, bmp_path = tempfile.mkstemp('.bmp')
        os.close(handle)

        self.loaded_path = Path(bmp_path)

    def _cmd_quicksave(self):
        suffix = self.orig_path.suffix
        name = self.orig_path.name.replace(suffix, '')
        parent_folder = self.orig_path.parents[0]

        # we need to search all files that match the pattern NAME_g*
        index = 0
        for glitched_out in parent_folder.glob(f'{name}*'):
            try:
                idx = int(glitched_out.name.strip(name + '_g')[0])
                index = idx
            except (IndexError, ValueError):
                continue

        # create our next glitched path
        out_path = shutil.copyfile(
            self.loaded_path, parent_folder / f'{name}_g{index + 1}.raw')
        print('saved to', out_path)


    def run(self, line: str):
        """Run a single line."""
        print(f'running {line!r}')

        args = shlex.split(line)
        command = args[0]

        if command != 'load' and self.loaded_path is None:
            print('warn: no file loaded.')

        try:
            method = getattr(self, f'_cmd_{command}')
        except AttributeError:
            raise InterpreterError(f'Command {command!r} not found')

        method(*args[1:])