css/custom_css.py
Jorge Maldonado Ventura 2e91cf3d78 Small fix
2017-02-15 13:23:16 +01:00

103 lines
2.8 KiB
Python

#!/usr/bin/env python3
# pelican-css: embed custom CSS easily
# Copyright (C) 2017 Jorge Maldonado Ventura
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Embed CSS files for Pelican
===========================
This plugin allows you to easily embed CSS files in the header of individual
articles. The CSS files are embedded using the HTML <link> tag inside the
<head> tag.
"""
import os
import shutil
from pelican import signals
def format_css(gen, metastring, formatter):
"""
Create a list of URL-formatted script/style tags
Parameters
----------
gen: generator
Pelican Generator
metastring: string
metadata['scripts'] or metadata['styles']
formatter: string
String format for output.
Output
------
List of formatted strings
"""
metalist = metastring.replace(' ', '').split(',')
if gen.settings['RELATIVE_URLS']:
site_url = '..'
else:
site_url = gen.settings['SITEURL']
return [formatter.format(site_url, x) for x in metalist]
def copy_resources(src, dest, file_list):
"""
Copy files from content folder to output folder
Parameters
----------
src: string
Content folder path
dest: string,
Output folder path
file_list: list
List of files to be transferred
Output
------
Copies files from content to output
"""
if not os.path.exists(dest):
os.makedirs(dest)
for file_ in file_list:
file_src = os.path.join(src, file_)
shutil.copy2(file_src, dest)
def add_tags(gen, metadata):
"""
It will add the CSS to the article
"""
if 'css' in metadata.keys():
style = '<link rel="stylesheet" href="{0}/css/{1}" type="text/css">'
metadata['styles'] = format_css(gen, metadata['css'], style)
def move_resources(gen):
"""
Move CSS files from css folder to output folder
"""
css_files = gen.get_files('css', extensions='css')
css_dest = os.path.join(gen.output_path, 'css')
copy_resources(gen.path, css_dest, css_files)
def register():
"""
Plugin registration
"""
signals.article_generator_context.connect(add_tags)
signals.page_generator_context.connect(add_tags)
signals.article_generator_finalized.connect(move_resources)