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 | |
a0462193 | 6 | from game.scene import Scene |
a747111f FC |
7 | |
8 | ||
9 | def do_screenshot(cls): | |
10 | system('python main.py --screenshot ' + cls.__name__) | |
63e7aeb2 FC |
11 | |
12 | ||
13 | def bld_screenshots(): | |
a747111f | 14 | scene_classes = [] |
a0462193 | 15 | for _file in glob('game/scenes/*.py'): |
a747111f FC |
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) |