import re from hashlib import md5 from typing import TYPE_CHECKING, Any, Callable, List, Optional, Type, TypeVar from django_components.util.nanoid import generate if TYPE_CHECKING: from django_components.component import Component T = TypeVar("T") # Based on nanoid implementation from # https://github.com/puyuan/py-nanoid/tree/99e5b478c450f42d713b6111175886dccf16f156/nanoid def gen_id() -> str: """Generate a unique ID that can be associated with a Node""" # Alphabet is only alphanumeric. Compared to the default alphabet used by nanoid, # we've omitted `-` and `_`. # With this alphabet, at 6 chars, the chance of collision is 1 in 3.3M. # See https://zelark.github.io/nano-id-cc/ return generate( "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ", size=6, ) def find_last_index(lst: List, predicate: Callable[[Any], bool]) -> Any: for r_idx, elem in enumerate(reversed(lst)): if predicate(elem): return len(lst) - 1 - r_idx return -1 def is_str_wrapped_in_quotes(s: str) -> bool: return s.startswith(('"', "'")) and s[0] == s[-1] and len(s) >= 2 def is_identifier(value: Any) -> bool: if not isinstance(value, str): return False if not value.isidentifier(): return False return True def any_regex_match(string: str, patterns: List[re.Pattern]) -> bool: return any(p.search(string) is not None for p in patterns) def no_regex_match(string: str, patterns: List[re.Pattern]) -> bool: return all(p.search(string) is None for p in patterns) # See https://stackoverflow.com/a/2020083/9788634 def get_import_path(cls_or_fn: Type[Any]) -> str: """ Get the full import path for a class or a function, e.g. `"path.to.MyClass"` """ module = cls_or_fn.__module__ if module == "builtins": return cls_or_fn.__qualname__ # avoid outputs like 'builtins.str' return module + "." + cls_or_fn.__qualname__ def default(val: Optional[T], default: T) -> T: return val if val is not None else default def get_last_index(lst: List, key: Callable[[Any], bool]) -> Optional[int]: for index, item in enumerate(reversed(lst)): if key(item): return len(lst) - 1 - index return None def is_nonempty_str(txt: Optional[str]) -> bool: return txt is not None and bool(txt.strip()) def hash_comp_cls(comp_cls: Type["Component"]) -> str: full_name = get_import_path(comp_cls) comp_cls_hash = md5(full_name.encode()).hexdigest()[0:6] return comp_cls.__name__ + "_" + comp_cls_hash