lib/build/screenshots.py

   1 from os import system
   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__)
  11
  12
  13 def bld_screenshots():
  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)