Hopiu/djLint
1
0
Fork 0
mirror of https://github.com/Hopiu/djLint.git synced 2026-04-19 12:50:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Correctly initialize and deinitialize colorama

Browse source 
djlint is invoked with __name__ == "djlint" (it must be, or the
relative imports would fail), so the if __name__ == "__main__" block
does nothing.  These calls need to be made from main() itself.

The colorama_text decorator makes this easy.

Signed-off-by: Anders Kaseorg <andersk@mit.edu>
This commit is contained in:
Anders Kaseorg 2022-03-17 13:07:17 -07:00
parent f9ff3477e1
commit beefd04471
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/djlint/__init__.py
View file

@ -11,7 +11,7 @@ from typing import Dict, List, Optional
import click
from click import echo
from colorama import Fore, Style, deinit, init
from colorama import Fore, Style, colorama_text
from tqdm import tqdm
from .lint import lint_file
@ -94,6 +94,7 @@ from .src import get_src
is_flag=True,
help="Return errors as warnings.",
)
@colorama_text(autoreset=True)
def main(
src: List[str],
extension: str,
@ -244,9 +245,3 @@ def process(config: Config, this_file: Path) -> Dict:
output["lint_message"] = lint_file(config, this_file)
return output
if __name__ == "__main__":
init(autoreset=True)
main()
deinit()