mirror of
https://github.com/Hopiu/djLint.git
synced 2026-05-19 10:31:07 +00:00
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
|
"""Build src file list."""
|
||
|
|
import re
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from click import echo
|
||
|
|
from colorama import Fore
|
||
|
|
|
||
|
|
from .settings import Config
|
||
|
|
|
||
|
|
|
||
|
|
def get_src(src: List[Path], config: Config) -> List[Path]:
|
||
|
|
"""Get source files."""
|
||
|
|
paths = []
|
||
|
|
for item in src:
|
||
|
|
if Path.is_file(item) and no_pragma(config, item):
|
||
|
|
paths.append(item)
|
||
|
|
continue
|
||
|
|
|
||
|
|
# remove leading . from extension
|
||
|
|
extension = str(config.extension)
|
||
|
|
extension = extension[1:] if extension.startswith(".") else extension
|
||
|
|
|
||
|
|
paths.extend(
|
||
|
|
filter(
|
||
|
|
lambda x: not re.search(config.exclude, str(x), re.VERBOSE)
|
||
|
|
and no_pragma(config, x),
|
||
|
|
list(item.glob(f"**/*.{extension}")),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
if len(paths) == 0:
|
||
|
|
echo(Fore.BLUE + "No files to check! 😢")
|
||
|
|
|
||
|
|
return paths
|
||
|
|
|
||
|
|
|
||
|
|
html_patterns = [re.compile(r"<!--\s*djlint\:on\s*-->")]
|
||
|
|
django_jinja_patterns = [
|
||
|
|
re.compile(r"\{#\s*djlint\:on\s*#\}"),
|
||
|
|
re.compile(r"\{%\s*comment\s*%\}\s*djlint\:on\s*\{%\s*endcomment\s*%\}"),
|
||
|
|
]
|
||
|
|
nunjucks_patterns = [re.compile(r"\{#\s*djlint\:on\s*#\}")]
|
||
|
|
handlebars_patterns = [re.compile(r"\{\{!--\s*djlint\:on\s*--\}\}")]
|
||
|
|
golang_patterns = [re.compile(r"\{\{-?\s*/\*\s*djlint\:on\s*\*/\s*-?\}\}")]
|
||
|
|
|
||
|
|
|
||
|
|
def no_pragma(config: Config, this_file: Path) -> bool:
|
||
|
|
"""Verify there is no pragma present."""
|
||
|
|
if config.require_pragma is False:
|
||
|
|
return True
|
||
|
|
|
||
|
|
with this_file.open(encoding="utf8") as open_file:
|
||
|
|
first_line = open_file.readline()
|
||
|
|
|
||
|
|
pragma_patterns = {
|
||
|
|
"django": django_jinja_patterns + html_patterns,
|
||
|
|
"jinja": django_jinja_patterns + html_patterns,
|
||
|
|
"nunjucks": nunjucks_patterns + html_patterns,
|
||
|
|
"handlebars": handlebars_patterns + html_patterns,
|
||
|
|
"golang": golang_patterns + html_patterns,
|
||
|
|
"all": django_jinja_patterns
|
||
|
|
+ nunjucks_patterns
|
||
|
|
+ handlebars_patterns
|
||
|
|
+ golang_patterns
|
||
|
|
+ html_patterns,
|
||
|
|
}
|
||
|
|
|
||
|
|
return any(
|
||
|
|
re.match(pattern, first_line) for pattern in pragma_patterns[config.profile]
|
||
|
|
)
|