94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
Create image galleries with GalleriaJS
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
from typing import Dict, Any
|
||
|
|
from sphinx.application import Sphinx
|
||
|
|
from sphinx.environment import BuildEnvironment
|
||
|
|
from sphinx.util.osutil import ensuredir, copyfile
|
||
|
|
from PIL import Image
|
||
|
|
|
||
|
|
from . import directive
|
||
|
|
from . import collector
|
||
|
|
|
||
|
|
__version_info__ = (1, 0, 0)
|
||
|
|
__version__ = '.'.join([str(val) for val in __version_info__])
|
||
|
|
|
||
|
|
|
||
|
|
def copy_images_files(app: Sphinx, env: BuildEnvironment) -> None:
|
||
|
|
|
||
|
|
if hasattr(env, 'galleria') and env.galleria:
|
||
|
|
for source in env.galleria:
|
||
|
|
relpath = os.path.relpath(source, env.srcdir)
|
||
|
|
dest = os.path.join(app.outdir, relpath)
|
||
|
|
ensuredir(os.path.dirname(dest))
|
||
|
|
copyfile(source, dest)
|
||
|
|
|
||
|
|
if hasattr(env, 'galleriathumbs') and env.galleriathumbs:
|
||
|
|
for thumb in env.galleriathumbs:
|
||
|
|
relpath = os.path.relpath(thumb, env.srcdir)
|
||
|
|
dest = os.path.join(app.outdir, relpath)
|
||
|
|
basename, ext = os.path.splitext(dest)
|
||
|
|
thumbsize = basename.split('-')[-1].split('x')
|
||
|
|
thumbsize = [int(size) for size in thumbsize]
|
||
|
|
original = '.'.join(basename.split('.')[:-1]) + ext
|
||
|
|
dest = basename + '.jpg'
|
||
|
|
ensuredir(os.path.dirname(dest))
|
||
|
|
|
||
|
|
with Image.open(original) as im:
|
||
|
|
im.thumbnail(thumbsize)
|
||
|
|
im.save(dest, "JPEG")
|
||
|
|
|
||
|
|
|
||
|
|
def install_static_files(app: Sphinx, env: BuildEnvironment) -> None:
|
||
|
|
path = os.path.join(
|
||
|
|
app.builder.outdir,
|
||
|
|
'_static',
|
||
|
|
'sphinxgalleria'
|
||
|
|
)
|
||
|
|
source = os.path.join(
|
||
|
|
os.path.dirname(__file__),
|
||
|
|
'static',
|
||
|
|
'sphinxgalleria',
|
||
|
|
)
|
||
|
|
|
||
|
|
statics = []
|
||
|
|
for dirpath, dirs, files in os.walk(source):
|
||
|
|
source_path = os.path.relpath(dirpath, source)
|
||
|
|
for static in files:
|
||
|
|
statics.append(os.path.join(source_path, static))
|
||
|
|
|
||
|
|
for static in statics:
|
||
|
|
static_path = os.path.join(path, static)
|
||
|
|
source_path = os.path.join(source, static)
|
||
|
|
if not os.path.exists(os.path.dirname(static_path)):
|
||
|
|
ensuredir(os.path.dirname(static_path))
|
||
|
|
|
||
|
|
copyfile(source_path, static_path)
|
||
|
|
|
||
|
|
if static.endswith('.js'):
|
||
|
|
app.add_js_file(os.path.join('sphinxgalleria', static))
|
||
|
|
elif static.endswith('.css'):
|
||
|
|
app.add_css_file(os.path.join('sphinxgalleria', static))
|
||
|
|
|
||
|
|
|
||
|
|
def setup(app: Sphinx) -> Dict[str, Any]:
|
||
|
|
app.add_node(
|
||
|
|
directive.galleria,
|
||
|
|
html=(directive.html_visit_galleria, None),
|
||
|
|
epub=(directive.html_visit_galleria, None),
|
||
|
|
latex=(directive.latex_visit_galleria, None),
|
||
|
|
texinfo=(directive.texinfo_visit_galleria, None),
|
||
|
|
text=(directive.text_visit_galleria, None),
|
||
|
|
man=(directive.man_visit_galleria, None),
|
||
|
|
gemini=(directive.gemini_visit_galleria, None),
|
||
|
|
)
|
||
|
|
app.add_directive('galleria', directive.GalleriaDirective)
|
||
|
|
app.add_env_collector(collector.GalleriaCollector)
|
||
|
|
app.connect('env-updated', install_static_files)
|
||
|
|
app.connect('env-updated', copy_images_files)
|
||
|
|
return {'version': __version__}
|