mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
25 lines
649 B
Python
25 lines
649 B
Python
"""Main entry point for the documentation build process"""
|
|
|
|
import os
|
|
|
|
|
|
def define_env(env):
|
|
"""Define custom environment variables for the documentation build process"""
|
|
|
|
@env.macro
|
|
def listimages(subdir):
|
|
"""Return a listing of all asset files in the provided subdir"""
|
|
here = os.path.dirname(__file__)
|
|
|
|
directory = os.path.join(here, 'docs', 'assets', 'images', subdir)
|
|
|
|
assets = []
|
|
|
|
allowed = ['.png', '.jpg']
|
|
|
|
for asset in os.listdir(directory):
|
|
if any(asset.endswith(x) for x in allowed):
|
|
assets.append(os.path.join(subdir, asset))
|
|
|
|
return assets
|