mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-05-24 01:12:54 -04:00
c4540f1395
* add dependency injection for get_repositories * convert events api to controller * update generic typing * add abstract controllers * update test naming * migrate admin services to controllers * add additional admin route tests * remove print * add public shared dependencies * add types * fix typo * add static variables for recipe json keys * add coverage gutters config * update controller routers * add generic success response * add category/tag/tool tests * add token refresh test * add coverage utilities * covert comments to controller * add todo * add helper properties * delete old service * update test notes * add unit test for pretty_stats * remove dead code from post_webhooks * update group routes to use controllers * add additional group test coverage * abstract common permission checks * convert ingredient parser to controller * update recipe crud to use controller * remove dead-code * add class lifespan tracker for debugging * convert bulk export to controller * migrate tools router to controller * update recipe share to controller * move customer router to _base * ignore prints in flake8 * convert units and foods to new controllers * migrate user routes to controllers * centralize error handling * fix invalid ref * reorder fields * update routers to share common handling * update tests * remove prints * fix cookbooks delete * fix cookbook get * add controller for mealplanner * cover report routes to controller * remove __future__ imports * remove dead code * remove all base_http children and remove dead code
102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
import re
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
from typing import Tuple
|
|
|
|
import black
|
|
import isort
|
|
from jinja2 import Template
|
|
|
|
|
|
def render_python_template(template_file: Path | str, dest: Path, data: dict) -> str:
|
|
"""Render and Format a Jinja2 Template for Python Code"""
|
|
if isinstance(template_file, Path):
|
|
tplt = Template(template_file.read_text())
|
|
else:
|
|
tplt = Template(template_file)
|
|
|
|
text = tplt.render(data=data)
|
|
text = black.format_str(text, mode=black.FileMode())
|
|
dest.write_text(text)
|
|
isort.file(dest)
|
|
|
|
|
|
@dataclass
|
|
class CodeSlicer:
|
|
start: int
|
|
end: int
|
|
|
|
indentation: str
|
|
text: list[str]
|
|
|
|
_next_line = None
|
|
|
|
def purge_lines(self) -> None:
|
|
start = self.start + 1
|
|
end = self.end
|
|
del self.text[start:end]
|
|
|
|
def push_line(self, string: str) -> None:
|
|
self._next_line = self._next_line or self.start + 1
|
|
print(self.indentation)
|
|
self.text.insert(self._next_line, self.indentation + string + "\n")
|
|
self._next_line += 1
|
|
|
|
|
|
def get_indentation_of_string(line: str, comment_char: str = "//") -> str:
|
|
return re.sub(rf"{comment_char}.*", "", line).removesuffix("\n")
|
|
|
|
|
|
def find_start_end(file_text: list[str], gen_id: str) -> Tuple[int, int]:
|
|
start = None
|
|
end = None
|
|
indentation = None
|
|
|
|
for i, line in enumerate(file_text):
|
|
if "CODE_GEN_ID:" in line and gen_id in line:
|
|
start = i
|
|
indentation = get_indentation_of_string(line)
|
|
if f"END: {gen_id}" in line:
|
|
end = i
|
|
|
|
if start is None or end is None:
|
|
raise Exception("Could not find start and end of code generation block")
|
|
|
|
if start > end:
|
|
raise Exception(f"Start ({start=}) of code generation block is after end ({end=})")
|
|
|
|
return start, end, indentation
|
|
|
|
|
|
def inject_inline(file_path: Path, key: str, code: list[str]) -> None:
|
|
"""Injects a list of strings into the file where the key is found in the format defined
|
|
by the code-generation. Strings are properly indented and a '\n' is added to the end of
|
|
each string.
|
|
|
|
Start -> 'CODE_GEN_ID: <key>'
|
|
End -> 'END: <key>'
|
|
|
|
If no 'CODE_GEN_ID: <key>' is found, and exception is raised
|
|
|
|
Args:
|
|
file_path (Path): Write to file
|
|
key (str): CODE_GEN_ID: <key>
|
|
code (list[str]): List of strings to inject.
|
|
|
|
"""
|
|
|
|
with open(file_path, "r") as f:
|
|
file_text = f.readlines()
|
|
|
|
start, end, indentation = find_start_end(file_text, key)
|
|
|
|
slicer = CodeSlicer(start, end, indentation, file_text)
|
|
|
|
slicer.purge_lines()
|
|
|
|
for line in code:
|
|
slicer.push_line(line)
|
|
|
|
with open(file_path, "w") as file:
|
|
file.writelines(slicer.text)
|