Canvim / PyPen

pypen/__main__.py

import sys
import argparse
import time
from os import path, getcwd
from importlib import util as import_util
import pkg_resources

from pypen.settings import settings
from pypen.drawing.pypen_window import PyPenWindow

import pyglet

_argument_parser = argparse.ArgumentParser()

def space_print(msg=""):
    print()
    print(msg)
    print()

def print_help(msg=""):
    global _argument_parser
    if msg:
        space_print(msg)

    _argument_parser.print_help(sys.stderr)
    sys.exit(0)

def cli():
    global _argument_parser
    try:
        _argument_parser = argparse.ArgumentParser(description="The PyPen CLI for managing PyPen Sketches", prog="pypen")
        _argument_parser.add_argument(
            "-i",
            "--init",
            action="store_true",
            help="Flag to create a new PyPen sketch which imports pypen and contains default functions.")

        _argument_parser.add_argument("filename", nargs="?", help="The name/path of your PyPen Sketch.", default="")
        _argument_parser.add_argument("-f", "--fullscreen", action="store_true")
        _argument_parser.add_argument(
            "--timeout",
            help="Timeout in milliseconds. Window will automatically close after this timeout.",
            type=float,
            required=False,
            default=0.0)
        _argument_parser.add_argument("-v", "--version",
                                      help="Displays the currently installed PyPen's version.",
                                      action="version",
                                      version=pkg_resources.get_distribution("pypen").version)

        arguments = _argument_parser.parse_args()

        if len(sys.argv) == 1:
            print_help("It seems like you are running the pypen command without any arguments.\nHere is some help:")

        if not arguments.init and not arguments.filename:
            print_help("PyPen needs a path to a sketch in order to run it!\nPlease provide a path with 'pypen <path_to_sketch>'\nor run 'pypen --init <path_to_new_sketch>' to create a new one!")

        if arguments.init:
            path_to_new_sketch = arguments.filename if arguments.filename else settings.default_pypen_name
            init(path_to_new_sketch)

        if path.splitext(arguments.filename)[1] != ".py":
            arguments.filename = "{}.py".format(arguments.filename)

        main(arguments)
    except argparse.ArgumentError as error:
        print(str(error))
        sys.exit(1)


def init(path_to_new_file):
    if path.splitext(path_to_new_file)[1] != ".py":
        path_to_new_file += ".py"

    pypen_path = path.join(getcwd(), path_to_new_file)
    template = ""

    try:
        with open(path.join(path.realpath(__file__), "..", "pypen_template.py"), "r") as template_file:
            template = template_file.read()
    except EnvironmentError:
        # In case PyPen is installed in a directory that doesn't have read access,
        # it uses the below template instead of reading the template file
        template = """from pypen import *


def start():
    settings.fps = 60


def update():
    fill_screen("orange")
    rectangle(20, 20, 300, 400, "red")
"""

    try:
        with open(pypen_path, "w") as new_file:
            new_file.writelines(template)
    except EnvironmentError:
        space_print("Could not create file {}\nDo you have the right permissions?".format(pypen_path))
        sys.exit(1)

    space_print("'{0}' has been created!\nRun it using 'pypen {0}'".format(path_to_new_file))
    sys.exit(0)


def main(arguments):
    if not arguments.init:
        try:
            file_path = path.join(getcwd(), arguments.filename)
            spec = import_util.spec_from_file_location("", file_path)
            user_sketch = import_util.module_from_spec(spec)
            spec.loader.exec_module(user_sketch)
        except FileNotFoundError:
            print()
            print(f"Hmm, PyPen can't find '{arguments.filename}'.")
            print("Are you sure that's the right path?")
            print()
            sys.exit(1)

    try:
        user_sketch.TIME

The variable user_sketch does not seem to be defined in case BooleanNotNode on line 109 is False. Are you sure this can never be the case?

        user_sketch.T
        user_sketch.PI
    except AttributeError as error:
        print()
        print(f"It seems like you're not importing PyPen to your sketch '{arguments.filename}'")
        print("Import it by writing 'from pypen import *' at the very top!")
        print()

        print(error)
        sys.exit(1)

    try:
        user_sketch.start
    except AttributeError:
        user_sketch.settings._user_has_start = False

    try:
        user_sketch.update
    except AttributeError:
        user_sketch.settings._user_has_update = False

    if not user_sketch.settings._user_has_start and not user_sketch.settings._user_has_update:
        print()
        print(f"Your PyPen sketch '{arguments.filename}' appears to have neither a start() nor an update() function.")
        print("Try to add at least one of those and run again!")
        print()
        sys.exit(1)

    window_title = f"PyPen | {path.splitext(path.split(arguments.filename)[1])[0]}"
    pypen_window = PyPenWindow(user_sketch, window_title, arguments)

    pyglet.app.run()


if __name__ == "__main__":
    cli()