From c66e4a33e1a07c439f03fe47f146a6c6482bf6df Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Thu, 8 Feb 2024 06:38:42 +0100 Subject: Merging upstream version 21.0.1. Signed-off-by: Daniel Baumann --- sqlglot/helper.py | 22 +++++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) (limited to 'sqlglot/helper.py') diff --git a/sqlglot/helper.py b/sqlglot/helper.py index de737be..9799fe2 100644 --- a/sqlglot/helper.py +++ b/sqlglot/helper.py @@ -53,11 +53,13 @@ def seq_get(seq: t.Sequence[T], index: int) -> t.Optional[T]: @t.overload -def ensure_list(value: t.Collection[T]) -> t.List[T]: ... +def ensure_list(value: t.Collection[T]) -> t.List[T]: + ... @t.overload -def ensure_list(value: T) -> t.List[T]: ... +def ensure_list(value: T) -> t.List[T]: + ... def ensure_list(value): @@ -79,11 +81,13 @@ def ensure_list(value): @t.overload -def ensure_collection(value: t.Collection[T]) -> t.Collection[T]: ... +def ensure_collection(value: t.Collection[T]) -> t.Collection[T]: + ... @t.overload -def ensure_collection(value: T) -> t.Collection[T]: ... +def ensure_collection(value: T) -> t.Collection[T]: + ... def ensure_collection(value): @@ -232,7 +236,7 @@ def tsort(dag: t.Dict[T, t.Set[T]]) -> t.List[T]: for node, deps in tuple(dag.items()): for dep in deps: - if not dep in dag: + if dep not in dag: dag[dep] = set() while dag: @@ -316,6 +320,14 @@ def find_new_name(taken: t.Collection[str], base: str) -> str: return new +def is_int(text: str) -> bool: + try: + int(text) + return True + except ValueError: + return False + + def name_sequence(prefix: str) -> t.Callable[[], str]: """Returns a name generator given a prefix (e.g. a0, a1, a2, ... if the prefix is "a").""" sequence = count() -- cgit v1.2.3