46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
import os.path
|
|
import yaml
|
|
from glob import glob
|
|
from project.pages.utils import get_title_from_markdown, parse_content
|
|
|
|
|
|
def get_data_from_file(base_dir, filename):
|
|
with open(os.path.join(base_dir, 'data', filename)) as data_file:
|
|
return yaml.load(data_file) or {}
|
|
|
|
|
|
def generate_config(base_dir):
|
|
default = get_data_from_file(base_dir, 'context.yml')
|
|
page = get_data_from_file(base_dir, 'page_context.yml')
|
|
switcher = get_data_from_file(base_dir, 'path_switch.yml')
|
|
|
|
# Add projects config
|
|
default['projects'] = generate_projects(base_dir)
|
|
# Join projects config with it's page context
|
|
for i in range(len(default['projects'])):
|
|
project = default['projects'][i]
|
|
if project['path'] in page: # If there's a custom config
|
|
default['projects'][i] = dict(project, **page[project['path']])
|
|
|
|
return default, page, switcher
|
|
|
|
|
|
def generate_projects(base_dir):
|
|
projects_path = os.path.join(base_dir, 'templates/projects')
|
|
files = []
|
|
for path in glob(projects_path + '/*.*'):
|
|
filename = path.replace(projects_path, '')
|
|
if filename == '/all.html':
|
|
continue
|
|
with open(path) as f:
|
|
if filename.split('.')[1] == 'md':
|
|
parsed_content = parse_content(f.read(), filename.split('.')[1])
|
|
filename = get_title_from_markdown(parsed_content)
|
|
else:
|
|
filename = filename.split('.')[0]
|
|
files.append({
|
|
"name": filename,
|
|
"path": 'projects' + path.replace(projects_path, '').split('.')[0],
|
|
"url": '/projects' + path.replace(projects_path, '').split('.')[0],
|
|
})
|
|
return files
|