Cool_Tools/mkdocs-material
1
0
Fork 0
mirror of https://github.com/squidfunk/mkdocs-material.git synced 2024-11-12 10:00:52 +01:00
Code Issues Releases Wiki Activity

Added support for theme.custom_dir in social plugin

Browse Source
This commit is contained in:
Kamil Krzyśków 2023-04-27 02:38:46 +02:00
parent 2bb853687d
commit 2e37c10d53
2 changed files with 52 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
material/plugins/social/plugin.py
View File

@ -68,6 +68,7 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
def __init__(self): def __init__(self):
self._executor = concurrent.futures.ThreadPoolExecutor(4) self._executor = concurrent.futures.ThreadPoolExecutor(4)
self.custom_dir = None
# Retrieve configuration # Retrieve configuration
def on_config(self, config): def on_config(self, config):
@ -112,6 +113,13 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
# Retrieve color overrides # Retrieve color overrides
self.color = { **self.color, **self.config.cards_color } self.color = { **self.color, **self.config.cards_color }
# Retrieve custom_dir path
for user_config in config.user_configs:
custom_dir = user_config.get("theme", {}).get("custom_dir")
if custom_dir:
self.custom_dir = custom_dir
break
# Retrieve logo and font # Retrieve logo and font
self._resized_logo_promise = self._executor.submit(self._load_resized_logo, config) self._resized_logo_promise = self._executor.submit(self._load_resized_logo, config)
self.font = self._load_font(config) self.font = self._load_font(config)
@ -344,8 +352,15 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
if "logo" in theme: if "logo" in theme:
_, extension = os.path.splitext(theme["logo"]) _, extension = os.path.splitext(theme["logo"])
# Load SVG and convert to PNG
path = os.path.join(config.docs_dir, theme["logo"]) path = os.path.join(config.docs_dir, theme["logo"])
# Allow users to put the logo inside their custom_dir (theme["logo"] case)
if self.custom_dir:
custom_dir_logo = os.path.join(self.custom_dir, theme["logo"])
if os.path.exists(custom_dir_logo):
path = custom_dir_logo
# Load SVG and convert to PNG
if extension == ".svg": if extension == ".svg":
return self._load_logo_svg(path) return self._load_logo_svg(path)
@ -353,10 +368,11 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
return Image.open(path).convert("RGBA") return Image.open(path).convert("RGBA")
# Handle icons # Handle icons
logo = "material/library"
icon = theme["icon"] or {} icon = theme["icon"] or {}
if "logo" in icon and icon["logo"]: if "logo" in icon and icon["logo"]:
logo = icon["logo"] logo = icon["logo"]
else:
logo = "material/library"
# Resolve path of package # Resolve path of package
base = os.path.abspath(os.path.join( base = os.path.abspath(os.path.join(
@ -364,8 +380,15 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
"../.." "../.."
)) ))
# Load icon data and fill with color
path = f"{base}/.icons/{logo}.svg" path = f"{base}/.icons/{logo}.svg"
# Allow users to put the logo inside their custom_dir (theme["icon"]["logo"] case)
if self.custom_dir:
custom_dir_logo = os.path.join(self.custom_dir, ".icons", f"{logo}.svg")
if os.path.exists(custom_dir_logo):
path = custom_dir_logo
# Load icon data and fill with color
return self._load_logo_svg(path, self.color["text"]) return self._load_logo_svg(path, self.color["text"])
# Load SVG file and convert to PNG # Load SVG file and convert to PNG

29
src/plugins/social/plugin.py
View File

@ -68,6 +68,7 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
def __init__(self): def __init__(self):
self._executor = concurrent.futures.ThreadPoolExecutor(4) self._executor = concurrent.futures.ThreadPoolExecutor(4)
self.custom_dir = None
# Retrieve configuration # Retrieve configuration
def on_config(self, config): def on_config(self, config):
@ -112,6 +113,13 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
# Retrieve color overrides # Retrieve color overrides
self.color = { **self.color, **self.config.cards_color } self.color = { **self.color, **self.config.cards_color }
# Retrieve custom_dir path
for user_config in config.user_configs:
custom_dir = user_config.get("theme", {}).get("custom_dir")
if custom_dir:
self.custom_dir = custom_dir
break
# Retrieve logo and font # Retrieve logo and font
self._resized_logo_promise = self._executor.submit(self._load_resized_logo, config) self._resized_logo_promise = self._executor.submit(self._load_resized_logo, config)
self.font = self._load_font(config) self.font = self._load_font(config)
@ -344,8 +352,15 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
if "logo" in theme: if "logo" in theme:
_, extension = os.path.splitext(theme["logo"]) _, extension = os.path.splitext(theme["logo"])
# Load SVG and convert to PNG
path = os.path.join(config.docs_dir, theme["logo"]) path = os.path.join(config.docs_dir, theme["logo"])
# Allow users to put the logo inside their custom_dir (theme["logo"] case)
if self.custom_dir:
custom_dir_logo = os.path.join(self.custom_dir, theme["logo"])
if os.path.exists(custom_dir_logo):
path = custom_dir_logo
# Load SVG and convert to PNG
if extension == ".svg": if extension == ".svg":
return self._load_logo_svg(path) return self._load_logo_svg(path)
@ -353,10 +368,11 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
return Image.open(path).convert("RGBA") return Image.open(path).convert("RGBA")
# Handle icons # Handle icons
logo = "material/library"
icon = theme["icon"] or {} icon = theme["icon"] or {}
if "logo" in icon and icon["logo"]: if "logo" in icon and icon["logo"]:
logo = icon["logo"] logo = icon["logo"]
else:
logo = "material/library"
# Resolve path of package # Resolve path of package
base = os.path.abspath(os.path.join( base = os.path.abspath(os.path.join(
@ -364,8 +380,15 @@ class SocialPlugin(BasePlugin[SocialPluginConfig]):
"../.." "../.."
)) ))
# Load icon data and fill with color
path = f"{base}/.icons/{logo}.svg" path = f"{base}/.icons/{logo}.svg"
# Allow users to put the logo inside their custom_dir (theme["icon"]["logo"] case)
if self.custom_dir:
custom_dir_logo = os.path.join(self.custom_dir, ".icons", f"{logo}.svg")
if os.path.exists(custom_dir_logo):
path = custom_dir_logo
# Load icon data and fill with color
return self._load_logo_svg(path, self.color["text"]) return self._load_logo_svg(path, self.color["text"])
# Load SVG file and convert to PNG # Load SVG file and convert to PNG