[pmachines.git] / lib / build / screenshots.py

from os import system
from glob import glob
from importlib import import_module
from inspect import isclass
from multiprocessing import Pool
from pmachines.scene import Scene


def do_screenshot(cls):
    system('python main.py --screenshot ' + cls.__name__)


def bld_screenshots():
    scene_classes = []
    for _file in glob('pmachines/scenes/*.py'):
        _fn = _file.replace('.py', '').replace('/', '.')
        for member in import_module(_fn).__dict__.values():
            if isclass(member) and issubclass(member, Scene) and \
                    member != Scene:
                scene_classes += [member]
    with Pool() as p:
        p.map(do_screenshot, scene_classes)
Commit	Line	Data
63e7aeb2	1	from os import system
a747111f FC	2	from glob import glob
	3	from importlib import import_module
	4	from inspect import isclass
	5	from multiprocessing import Pool
	6	from pmachines.scene import Scene
	7
	8
	9	def do_screenshot(cls):
	10	system('python main.py --screenshot ' + cls.__name__)
63e7aeb2 FC	11
	12
	13	def bld_screenshots():
a747111f FC	14	scene_classes = []
	15	for _file in glob('pmachines/scenes/*.py'):
	16	_fn = _file.replace('.py', '').replace('/', '.')
	17	for member in import_module(_fn).__dict__.values():
	18	if isclass(member) and issubclass(member, Scene) and \
	19	member != Scene:
	20	scene_classes += [member]
	21	with Pool() as p:
	22	p.map(do_screenshot, scene_classes)