opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-02 11:03:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added support for Windows

Browse Source
This commit is contained in:
shahar603
2017-06-29 14:33:01 +03:00
committed by GitHub
parent 05db6174e9
commit 1467e02d18
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
extract_scene.py
View File

@ -7,6 +7,7 @@ import itertools as it
import inspect
import traceback
import imp
import os
from helpers import *
from scene import Scene
@ -151,14 +152,28 @@ def get_scene_classes(scene_names_to_classes, config):
return scene_names_to_classes.values()
return prompt_user_for_choice(scene_names_to_classes)
def get_module(file_name):
def get_module_windows(file_name):
module_name = file_name.replace(".py", "")
last_module = imp.load_module(".", *imp.find_module("."))
last_module = imp.load_module("__init__", *imp.find_module("__init__", ['.']))
for part in module_name.split(os.sep):
load_args = imp.find_module(part, last_module.__path__)
load_args = imp.find_module(part, [os.path.dirname(last_module.__file__)])
last_module = imp.load_module(part, *load_args)
return last_module
def get_module_posix(file_name):
module_name = file_name.replace(".py", "")
last_module = imp.load_module(".", *imp.find_module("."))
for part in module_name.split(os.sep):
load_args = imp.find_module(part, last_module.__path__)
last_module = imp.load_module(part, *load_args)
return last_module
def get_module(file_name):
if os.name == 'nt':
return get_module_windows(file_name)
return get_module_posix(file_name)
def main():
config = get_configuration(sys.argv)
module = get_module(config["file"])