Pokemon-Terminal/scripter.py

104 lines
3.7 KiB
Python

# 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())