public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update config.py to load config file manually.

Browse source
This commit is contained in:
GZTime 2021-04-09 21:32:52 +08:00
parent 41120b096e
commit 65d0826b91
1 changed files with 36 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
manimlib/config.py
View file

@ -130,7 +130,11 @@ def parse_cli():
) )
parser.add_argument( parser.add_argument(
"--video_dir", "--video_dir",
help="directory to write video", help="Directory to write video",
)
parser.add_argument(
"--config_file",
help="Path to the custom configuration file",
) )
args = parser.parse_args() args = parser.parse_args()
return args return args
@ -155,17 +159,19 @@ def get_module(file_name):
spec.loader.exec_module(module) spec.loader.exec_module(module)
return module return module
__config_file__ = "custom_config.yml"
def get_custom_config(): def get_custom_config():
filename = "custom_config.yml" global __config_file__
global_defaults_file = os.path.join(get_manim_dir(), "manimlib", "default_config.yml") global_defaults_file = os.path.join(get_manim_dir(), "manimlib", "default_config.yml")
if os.path.exists(global_defaults_file): if os.path.exists(global_defaults_file):
with open(global_defaults_file, "r") as file: with open(global_defaults_file, "r") as file:
config = yaml.safe_load(file) config = yaml.safe_load(file)
if os.path.exists(filename): if os.path.exists(__config_file__):
with open(filename, "r") as file: with open(__config_file__, "r") as file:
local_defaults = yaml.safe_load(file) local_defaults = yaml.safe_load(file)
if local_defaults: if local_defaults:
config = merge_dicts_recursively( config = merge_dicts_recursively(
@ -173,22 +179,40 @@ def get_custom_config():
local_defaults, local_defaults,
) )
else: else:
with open(filename, "r") as file: with open(__config_file__, "r") as file:
config = yaml.safe_load(file) config = yaml.safe_load(file)
return config return config
def get_configuration(args): def get_configuration(args):
local_config_file = "custom_config.yml" global __config_file__
# ensure __config_file__ always exists
if args.config_file is not None:
if not os.path.exists(args.config_file):
print("There is no configuration file detected.")
print(f"Copy default configuration file to {args.config_file}.")
if sys.platform == 'win32':
os.system(f"copy default_config.yml {args.config_file}")
elif sys.platform == 'linux2':
os.system(f"cp default_config.yml {args.config_file}")
else:
print("Please create the file manually.")
print("Read configuration from default_config.yml.")
else:
__config_file__ = args.config_file
global_defaults_file = os.path.join(get_manim_dir(), "manimlib", "default_config.yml") global_defaults_file = os.path.join(get_manim_dir(), "manimlib", "default_config.yml")
if not (os.path.exists(global_defaults_file) or os.path.exists(local_config_file)):
if not (os.path.exists(global_defaults_file) or os.path.exists(__config_file__)):
print("There is no configuration file detected. Initial configuration:\n") print("There is no configuration file detected. Initial configuration:\n")
init_customization() init_customization()
elif not os.path.exists(local_config_file):
print(f"""Warning: Using the default configuration file, which you can modify in {global_defaults_file} elif not os.path.exists(__config_file__):
If you want to create a local configuration file, you can create a file named {local_config_file}, or run manimgl --config print(f"Warning: Using the default configuration file, which you can modify in {global_defaults_file}")
""") print(f"If you want to create a local configuration file, you can create a file named {__config_file__}, or run manimgl --config")
custom_config = get_custom_config() custom_config = get_custom_config()
write_file = any([args.write_file, args.open, args.finder]) write_file = any([args.write_file, args.open, args.finder])