File size: 2,100 Bytes
e020674 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import os
import re
from colorama import init, Fore, Style
from .paths import DataFlowPath
from .copy_funcs import copy_files_without_recursion, copy_file, copy_files_recursively
def _copy_scripts():
target_dir = os.getcwd()
if not os.path.exists(target_dir):
os.makedirs(target_dir)
# script_path = DataFlowPath.get_dataflow_scripts_dir()
copy_files_recursively(DataFlowPath.get_dataflow_scripts_dir(), target_dir)
def _copy_pipelines():
target_dir = os.getcwd()
if not os.path.exists(target_dir):
os.makedirs(target_dir)
copy_files_recursively(DataFlowPath.get_dataflow_pipelines_dir(), target_dir)
# Copy pipelines
def _copy_playground():
target_dir = os.getcwd()
if not os.path.exists(target_dir):
os.makedirs(target_dir)
copy_files_recursively(DataFlowPath.get_dataflow_playground_dir(), target_dir)
def _copy_agentscript():
target_dir = os.getcwd()
if not os.path.exists(target_dir):
os.makedirs(target_dir)
copy_files_recursively(DataFlowPath.get_dataflow_agentscript_dir(), target_dir)
def _copy_examples():
target_dir = os.path.join(os.getcwd(), "example_data")
if not os.path.exists(target_dir):
os.makedirs(target_dir)
copy_files_recursively(DataFlowPath.get_dataflow_example_dir(), target_dir)
def cli_init(subcommand):
print(f'{Fore.GREEN}Initializing in current working directory...{Style.RESET_ALL}')
# base initialize that only contain default scripts
if subcommand == "base":
_copy_pipelines()
_copy_examples()
_copy_playground()
_copy_agentscript()
# if subcommand == "model_zoo":
# _copy_train_scripts()
# _copy_demo_runs()
# _copy_demo_configs()
# _copy_dataset_json()
# # base initialize that only contain default scripts
# if subcommand == "backbone":
# _copy_train_scripts()
# _copy_demo_runs()
# _copy_demo_configs()
# _copy_dataset_json()
# print(f'{Fore.GREEN}Successfully initialized IMDLBenCo scripts.{Style.RESET_ALL}') |