def setup_project():
"""
Add a new project to the config file. Can also be used for updating an existing project.
"""
data = request.form
name = data['projectName']
path = data['path']
# Initialize project directory
if not os.path.exists(path):
os.mkdir(path)
# Update project config
try:
# Check for existing config file
config = utils.load_project_config(path)
old_name = config['name']
config['name'] = name
except FileNotFoundError:
# Setup new project config
config = {
'name': name,
'date_created': datetime.date.today().isoformat(),
'classes': {},
'use_gpu': False,
'temporal': False,
'video_recording': {
'countdown': 3,
'recording': 5,
},
}
old_name = None
utils.write_project_config(path, config)
# Setup directory structure
for split in SPLITS:
videos_dir = directories.get_videos_dir(path, split)
if not os.path.exists(videos_dir):
os.mkdir(videos_dir)
# Update overall projects config file
projects = utils.load_project_overview_config()
if old_name and old_name in projects:
del projects[old_name]
projects[name] = {
'path': path,
}
utils.write_project_overview_config(projects)
return redirect(url_for('project_details', project=name))
def remove_project(name):
"""
Remove a given project from the config file and reload the overview page.
"""
name = urllib.parse.unquote(name)
projects = utils.load_project_overview_config()
del projects[name]
utils.write_project_overview_config(projects)
return redirect(url_for('projects_overview'))