41 lines
865 B
Python
41 lines
865 B
Python
"""
|
|
setup or cleanup enviroment file
|
|
usage: rvc env [create / cleanup]
|
|
Default: [nowDir/.env]
|
|
|
|
"""
|
|
|
|
import os
|
|
|
|
import click
|
|
|
|
|
|
@click.group()
|
|
def env():
|
|
pass
|
|
|
|
|
|
@env.command()
|
|
def create():
|
|
env_file_path = os.path.join(os.getcwd(), ".env")
|
|
|
|
if not os.path.exists(env_file_path):
|
|
default_values = {
|
|
"weight_root": "",
|
|
"weight_uvr5_root": "",
|
|
"index_root": "",
|
|
"rmvpe_root": "",
|
|
"hubert_path": "",
|
|
"save_uvr_path": "",
|
|
"TEMP": "",
|
|
"pretrained": "",
|
|
}
|
|
|
|
with open(env_file_path, "w") as env_file:
|
|
for key, value in default_values.items():
|
|
env_file.write(f"{key}={value}\n")
|
|
|
|
click.echo(f"{env_file_path} created successfully.")
|
|
else:
|
|
click.echo(f"{env_file_path} already exists, no change")
|