Pokemon-Terminal/scripter.py

# Used for creating, running and analyzing applescript and bash scripts.

import os


def __terminal_script(pokemon):
    # Create the content for script that will change the terminal background image.
    content = "tell application \"iTerm\"\n"
    content += "\ttell current session of current window\n"
    content += "\t\tset background image to \"" + pokemon.get_path() + "\"\n"
    content += "\tend tell\n"
    content += "end tell"
    return content


def __wallpaper_script(pokemon):
    # Create the content for the script that will change the wallpaper.
    content = "tell application \"System Events\"\n"
    content += "\ttell current desktop\n"
    content += "\t\tset picture to \"" + pokemon.get_path() + "\"\n"
    content += "\tend tell\n"
    content += "end tell"
    return content


def __create_terminal_script(pokemon):
    # Create and save the script for changing the terminal background image.
    content = __terminal_script(pokemon)
    file = open(os.get_exec_path()[0] + "/./Scripts/background.scpt", "wb")
    file.write(bytes(content, 'UTF-8'))
    file.close()


def __create_wallpaper_script(pokemon):
    # Create and save the script for changing the wallpaper.
    content = __wallpaper_script(pokemon)
    file = open(os.get_exec_path()[0] + "/./Scripts/wallpaper.scpt", "wb")
    file.write(bytes(content, 'UTF-8'))
    file.close()


def __create_terminal_bash():
    # Create and save the run.sh that will execute the AppleScript if the correct run.sh doesn't already exist.
    content = "#!/bin/bash\n" + "osascript " + os.get_exec_path()[0] + "/./Scripts/background.scpt"
    if open(os.get_exec_path()[0] + "/./Scripts/run.sh", 'r').read() == content:
        return
    file = open(os.get_exec_path()[0] + "/./Scripts/run.sh", 'wb')
    file.write(bytes(content, 'UTF-8'))
    file.close()


def __create_wallpaper_bash():
    # Create and save the run.sh that will execute the AppleScript if the correct run.sh doesn't already exist.
    content = "#!/bin/bash\n" + "osascript " + os.get_exec_path()[0] + "/./Scripts/wallpaper.scpt"
    if open(os.get_exec_path()[0] + "/./Scripts/run.sh", 'r').read() == content:
        return
    file = open(os.get_exec_path()[0] + "/./Scripts/run.sh", 'wb')
    file.write(bytes(content, 'UTF-8'))
    file.close()


def change_terminal(pokemon):
    # Create, save and run the bash script to change the terminal background.
    __create_terminal_script(pokemon)
    __create_terminal_bash()
    os.system(os.get_exec_path()[0] + "/./Scripts/run.sh")


def change_wallpaper(pokemon):
    # Create, save and run the bash script to change the wallpaper.
    __create_wallpaper_script(pokemon)
    __create_wallpaper_bash()
    os.system(os.get_exec_path()[0] + "/./Scripts/run.sh")


def determine_terminal_pokemon(db):
    # Print the current Pokemon that is being used as the terminal background.
    __determine_pokemon(db, "background.scpt")


def determine_wallpaper_pokemon(db):
    # Print the current Pokemon that is being used the wallpaper.
    __determine_pokemon(db, "wallpaper.scpt")


def __determine_pokemon(db, script_name):
    # Helper method to get the current Pokemon that is in the specified script.
    path = os.get_exec_path()[0] + "/Scripts/" + script_name
    try:
        content = open(path, "r+").readlines()
    except FileNotFoundError:
        print("Missing File: ", path)
        return

    try:
        split = content[2].split('/')
        image_name = split[-1]  # The content after the final slash.
        image_name = image_name[:-6]  # Remove the .png and quotation at the end.
    except IndexError:
        print("Corrupt file:", path)
        return

    pokemon = db.get_pokemon(image_name)
    print(pokemon.get_id(), pokemon.get_name().capitalize())