From 8d67d68f26fac86236236a3dd4dceb3d073c290c Mon Sep 17 00:00:00 2001 From: Sunpy Date: Wed, 12 Feb 2020 13:32:55 +0100 Subject: [PATCH] Selectable shader module --- main.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index c1a75bb..c6d7f68 100644 --- a/main.py +++ b/main.py @@ -1,16 +1,22 @@ -# import shutil # shutil.get_terminal_size(fallback=(0, 0)) import os +import sys import math +from importlib import import_module from pixel import Screen -from shaders import eyeblower - WIDTH, HEIGHT = 100, 30 # Set to -1, -1 for dynamic resolution STD_HANDLE = 1 # Would want this to be 0, but it can cause issues +if len(sys.argv) < 2: # Randomly select a shader + from glob import glob + from random import choice + module = import_module( choice( glob("shaders/*.py") ).rstrip(".py").replace("\\", ".") ) +else: + module = import_module("shaders.%s" % sys.argv[1]) + screen = Screen(WIDTH, HEIGHT, STD_HANDLE, [ - eyeblower.main + module.main ]) screen.mainloop() \ No newline at end of file