mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
30 lines
688 B
Python
30 lines
688 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
|