]> git.openstreetmap.org Git - nominatim.git/blob - nominatim/typing.py
Merge remote-tracking branch 'upstream/master' into collect_os_info.sh
[nominatim.git] / nominatim / typing.py
1 # SPDX-License-Identifier: GPL-2.0-only
2 #
3 # This file is part of Nominatim. (https://nominatim.org)
4 #
5 # Copyright (C) 2022 by the Nominatim developer community.
6 # For a full list of authors see the git log.
7 """
8 Type definitions for typing annotations.
9
10 Complex type definitions are moved here, to keep the source files readable.
11 """
12 from typing import Any, Union, Mapping, TypeVar, Sequence, TYPE_CHECKING
13
14 # Generics variable names do not confirm to naming styles, ignore globally here.
15 # pylint: disable=invalid-name,abstract-method,multiple-statements
16 # pylint: disable=missing-class-docstring,useless-import-alias
17
18 if TYPE_CHECKING:
19     import psycopg2.sql
20     import psycopg2.extensions
21     import psycopg2.extras
22     import os
23
24 StrPath = Union[str, 'os.PathLike[str]']
25
26 SysEnv = Mapping[str, str]
27
28 # psycopg2-related types
29
30 Query = Union[str, bytes, 'psycopg2.sql.Composable']
31
32 T_ResultKey = TypeVar('T_ResultKey', int, str)
33
34 class DictCursorResult(Mapping[str, Any]):
35     def __getitem__(self, x: Union[int, str]) -> Any: ...
36
37 DictCursorResults = Sequence[DictCursorResult]
38
39 T_cursor = TypeVar('T_cursor', bound='psycopg2.extensions.cursor')
40
41 # The following typing features require typing_extensions to work
42 # on all supported Python versions.
43 # Only require this for type checking but not for normal operations.
44
45 if TYPE_CHECKING:
46     from typing_extensions import (Protocol as Protocol,
47                                    Final as Final,
48                                    TypedDict as TypedDict)
49 else:
50     Protocol = object
51     Final = 'Final'
52     TypedDict = dict