Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
add type hints
  • Loading branch information
Archmonger committed Oct 31, 2021
commit bc1666e4ed12d845136282f57b6c628a4bda89a6
9 changes: 5 additions & 4 deletions src/django_idom/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import re
from fnmatch import fnmatch
from importlib import import_module
from typing import Set

from django.template import engines
from django.utils.encoding import smart_str
Expand Down Expand Up @@ -64,7 +65,7 @@ def _get_loaders(self):
loaders.append(loader)
return loaders

def _get_paths(self):
def _get_paths(self) -> Set:
"""Obtains a set of all template directories."""
paths = set()
for loader in self._get_loaders():
Expand All @@ -79,7 +80,7 @@ def _get_paths(self):

return paths

def _get_templates(self, paths):
def _get_templates(self, paths: Set) -> Set:
"""Obtains a set of all HTML template paths."""
extensions = [".html"]
templates = set()
Expand All @@ -94,7 +95,7 @@ def _get_templates(self, paths):

return templates

def _get_components(self, templates):
def _get_components(self, templates: Set) -> Set:
"""Obtains a set of all IDOM components by parsing HTML templates."""
component_regex = re.compile(
r"{% *idom_component ((\"[^\"']*\")|('[^\"']*')).*%}"
Expand All @@ -114,7 +115,7 @@ def _get_components(self, templates):

return components

def _register_components(self, components):
def _register_components(self, components: Set) -> None:
"""Registers all IDOM components in an iterable."""
for component in components:
try:
Expand Down