mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-28 05:07:14 +00:00
chore: try pack templates in langbot/
This commit is contained in:
+1
-1
@@ -8,7 +8,7 @@ recursive-include pkg *.py
|
|||||||
recursive-include libs *.py
|
recursive-include libs *.py
|
||||||
|
|
||||||
# Include templates and resources
|
# Include templates and resources
|
||||||
recursive-include templates *
|
recursive-include langbot/templates *
|
||||||
recursive-include res *
|
recursive-include res *
|
||||||
|
|
||||||
# Include compiled frontend files (will be added during build)
|
# Include compiled frontend files (will be added during build)
|
||||||
|
|||||||
+17
-20
@@ -1,6 +1,6 @@
|
|||||||
import os
|
import os
|
||||||
import shutil
|
|
||||||
import yaml
|
import yaml
|
||||||
|
import importlib.resources as resources
|
||||||
|
|
||||||
from .. import model as file_model
|
from .. import model as file_model
|
||||||
|
|
||||||
@@ -11,29 +11,27 @@ class YAMLConfigFile(file_model.ConfigFile):
|
|||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
config_file_name: str,
|
config_file_name: str,
|
||||||
template_file_name: str = None,
|
template_resource_name: str = None,
|
||||||
template_data: dict = None,
|
template_data: dict = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
self.config_file_name = config_file_name
|
self.config_file_name = config_file_name
|
||||||
self.template_file_name = template_file_name
|
self.template_resource_name = template_resource_name
|
||||||
self.template_data = template_data
|
self.template_data = template_data
|
||||||
|
|
||||||
def _get_template_path(self) -> str:
|
|
||||||
"""Get the actual path to the template file, handling package installation"""
|
|
||||||
if self.template_file_name is None:
|
|
||||||
return None
|
|
||||||
|
|
||||||
from ...utils import paths as path_utils
|
|
||||||
return path_utils.get_resource_path(self.template_file_name)
|
|
||||||
|
|
||||||
def exists(self) -> bool:
|
def exists(self) -> bool:
|
||||||
return os.path.exists(self.config_file_name)
|
return os.path.exists(self.config_file_name)
|
||||||
|
|
||||||
|
async def get_template_file_str(self) -> str:
|
||||||
|
if self.template_resource_name is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
with resources.path('langbot.templates', self.template_resource_name) as path:
|
||||||
|
return path.open('r', encoding='utf-8').read()
|
||||||
|
|
||||||
async def create(self):
|
async def create(self):
|
||||||
template_path = self._get_template_path()
|
if await self.get_template_file_str() is not None:
|
||||||
|
with open(self.config_file_name, 'w', encoding='utf-8') as f:
|
||||||
if template_path is not None:
|
f.write(await self.get_template_file_str())
|
||||||
shutil.copyfile(template_path, self.config_file_name)
|
|
||||||
elif self.template_data is not None:
|
elif self.template_data is not None:
|
||||||
with open(self.config_file_name, 'w', encoding='utf-8') as f:
|
with open(self.config_file_name, 'w', encoding='utf-8') as f:
|
||||||
yaml.dump(self.template_data, f, indent=4, allow_unicode=True)
|
yaml.dump(self.template_data, f, indent=4, allow_unicode=True)
|
||||||
@@ -44,11 +42,10 @@ class YAMLConfigFile(file_model.ConfigFile):
|
|||||||
if not self.exists():
|
if not self.exists():
|
||||||
await self.create()
|
await self.create()
|
||||||
|
|
||||||
template_path = self._get_template_path()
|
template_file_str = await self.get_template_file_str()
|
||||||
|
|
||||||
if template_path is not None:
|
if template_file_str is not None:
|
||||||
with open(template_path, 'r', encoding='utf-8') as f:
|
self.template_data = yaml.load(template_file_str, Loader=yaml.FullLoader)
|
||||||
self.template_data = yaml.load(f, Loader=yaml.FullLoader)
|
|
||||||
|
|
||||||
with open(self.config_file_name, 'r', encoding='utf-8') as f:
|
with open(self.config_file_name, 'r', encoding='utf-8') as f:
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -63,6 +63,7 @@ async def load_python_module_config(config_name: str, template_name: str, comple
|
|||||||
async def load_json_config(
|
async def load_json_config(
|
||||||
config_name: str,
|
config_name: str,
|
||||||
template_name: str = None,
|
template_name: str = None,
|
||||||
|
template_resource_name: str = None,
|
||||||
template_data: dict = None,
|
template_data: dict = None,
|
||||||
completion: bool = True,
|
completion: bool = True,
|
||||||
) -> ConfigManager:
|
) -> ConfigManager:
|
||||||
@@ -85,6 +86,7 @@ async def load_json_config(
|
|||||||
async def load_yaml_config(
|
async def load_yaml_config(
|
||||||
config_name: str,
|
config_name: str,
|
||||||
template_name: str = None,
|
template_name: str = None,
|
||||||
|
template_resource_name: str = None,
|
||||||
template_data: dict = None,
|
template_data: dict = None,
|
||||||
completion: bool = True,
|
completion: bool = True,
|
||||||
) -> ConfigManager:
|
) -> ConfigManager:
|
||||||
|
|||||||
Reference in New Issue
Block a user