methodwebscan 0.0.17__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- methodwebscan-0.0.17/PKG-INFO +25 -0
- methodwebscan-0.0.17/README.md +0 -0
- methodwebscan-0.0.17/pyproject.toml +56 -0
- methodwebscan-0.0.17/src/methodwebscan/__init__.py +71 -0
- methodwebscan-0.0.17/src/methodwebscan/core/__init__.py +25 -0
- methodwebscan-0.0.17/src/methodwebscan/core/datetime_utils.py +28 -0
- methodwebscan-0.0.17/src/methodwebscan/core/pydantic_utilities.py +249 -0
- methodwebscan-0.0.17/src/methodwebscan/core/serialization.py +254 -0
- methodwebscan-0.0.17/src/methodwebscan/py.typed +0 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/__init__.py +54 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/common/__init__.py +5 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/common/tls_version.py +7 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/__init__.py +10 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/certificate.py +39 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/fingerprint_report.py +31 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/http_headers.py +33 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/public_key_algorithm.py +5 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/signature_algorithm.py +25 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fingerprint/tls_info.py +23 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fuzzpath/__init__.py +6 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fuzzpath/fuzz_path_report.py +25 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/fuzzpath/url_details.py +19 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/__init__.py +10 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/graph_ql_data.py +20 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/graph_ql_field.py +17 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/graph_ql_query.py +18 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/graph_ql_schema.py +18 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/graph_ql_schema_data.py +18 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/graphql/graph_ql_type.py +21 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/__init__.py +23 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/api_type.py +5 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/o_auth_flow.py +22 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/o_auth_flows.py +27 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/route.py +31 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/routes_report.py +37 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/security_requirement.py +17 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/security_scheme.py +35 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/security_scheme_name.py +3 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/routes/security_scheme_type.py +7 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/webpagecapture/__init__.py +6 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/webpagecapture/webpage_capture_report.py +19 -0
- methodwebscan-0.0.17/src/methodwebscan/resources/webpagecapture/webpage_screenshot_report.py +19 -0
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
Metadata-Version: 2.1
|
|
2
|
+
Name: methodwebscan
|
|
3
|
+
Version: 0.0.17
|
|
4
|
+
Summary:
|
|
5
|
+
Requires-Python: >=3.8,<4.0
|
|
6
|
+
Classifier: Intended Audience :: Developers
|
|
7
|
+
Classifier: Operating System :: MacOS
|
|
8
|
+
Classifier: Operating System :: Microsoft :: Windows
|
|
9
|
+
Classifier: Operating System :: OS Independent
|
|
10
|
+
Classifier: Operating System :: POSIX
|
|
11
|
+
Classifier: Operating System :: POSIX :: Linux
|
|
12
|
+
Classifier: Programming Language :: Python
|
|
13
|
+
Classifier: Programming Language :: Python :: 3
|
|
14
|
+
Classifier: Programming Language :: Python :: 3.8
|
|
15
|
+
Classifier: Programming Language :: Python :: 3.9
|
|
16
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
17
|
+
Classifier: Programming Language :: Python :: 3.11
|
|
18
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
19
|
+
Classifier: Topic :: Software Development :: Libraries :: Python Modules
|
|
20
|
+
Classifier: Typing :: Typed
|
|
21
|
+
Requires-Dist: pydantic (>=1.9.2)
|
|
22
|
+
Requires-Dist: pydantic-core (>=2.18.2,<3.0.0)
|
|
23
|
+
Description-Content-Type: text/markdown
|
|
24
|
+
|
|
25
|
+
|
|
File without changes
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
[tool.poetry]
|
|
2
|
+
name = "methodwebscan"
|
|
3
|
+
version = "v0.0.17"
|
|
4
|
+
description = ""
|
|
5
|
+
readme = "README.md"
|
|
6
|
+
authors = []
|
|
7
|
+
keywords = []
|
|
8
|
+
|
|
9
|
+
classifiers = [
|
|
10
|
+
"Intended Audience :: Developers",
|
|
11
|
+
"Programming Language :: Python",
|
|
12
|
+
"Programming Language :: Python :: 3",
|
|
13
|
+
"Programming Language :: Python :: 3.8",
|
|
14
|
+
"Programming Language :: Python :: 3.9",
|
|
15
|
+
"Programming Language :: Python :: 3.10",
|
|
16
|
+
"Programming Language :: Python :: 3.11",
|
|
17
|
+
"Programming Language :: Python :: 3.12",
|
|
18
|
+
"Operating System :: OS Independent",
|
|
19
|
+
"Operating System :: POSIX",
|
|
20
|
+
"Operating System :: MacOS",
|
|
21
|
+
"Operating System :: POSIX :: Linux",
|
|
22
|
+
"Operating System :: Microsoft :: Windows",
|
|
23
|
+
"Topic :: Software Development :: Libraries :: Python Modules",
|
|
24
|
+
"Typing :: Typed"
|
|
25
|
+
]
|
|
26
|
+
packages = [
|
|
27
|
+
{ include = "methodwebscan", from = "src"}
|
|
28
|
+
]
|
|
29
|
+
|
|
30
|
+
[tool.poetry.dependencies]
|
|
31
|
+
python = "^3.8"
|
|
32
|
+
pydantic = ">= 1.9.2"
|
|
33
|
+
pydantic-core = "^2.18.2"
|
|
34
|
+
|
|
35
|
+
[tool.poetry.dev-dependencies]
|
|
36
|
+
mypy = "1.0.1"
|
|
37
|
+
pytest = "^7.4.0"
|
|
38
|
+
pytest-asyncio = "^0.23.5"
|
|
39
|
+
python-dateutil = "^2.9.0"
|
|
40
|
+
types-python-dateutil = "^2.9.0.20240316"
|
|
41
|
+
ruff = "^0.5.6"
|
|
42
|
+
|
|
43
|
+
[tool.pytest.ini_options]
|
|
44
|
+
testpaths = [ "tests" ]
|
|
45
|
+
asyncio_mode = "auto"
|
|
46
|
+
|
|
47
|
+
[tool.mypy]
|
|
48
|
+
plugins = ["pydantic.mypy"]
|
|
49
|
+
|
|
50
|
+
[tool.ruff]
|
|
51
|
+
line-length = 120
|
|
52
|
+
|
|
53
|
+
|
|
54
|
+
[build-system]
|
|
55
|
+
requires = ["poetry-core"]
|
|
56
|
+
build-backend = "poetry.core.masonry.api"
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
|
2
|
+
|
|
3
|
+
from .resources import (
|
|
4
|
+
ApiType,
|
|
5
|
+
Certificate,
|
|
6
|
+
FingerprintReport,
|
|
7
|
+
FuzzPathReport,
|
|
8
|
+
GraphQlData,
|
|
9
|
+
GraphQlField,
|
|
10
|
+
GraphQlQuery,
|
|
11
|
+
GraphQlSchema,
|
|
12
|
+
GraphQlSchemaData,
|
|
13
|
+
GraphQlType,
|
|
14
|
+
HttpHeaders,
|
|
15
|
+
OAuthFlow,
|
|
16
|
+
OAuthFlows,
|
|
17
|
+
PublicKeyAlgorithm,
|
|
18
|
+
Route,
|
|
19
|
+
RoutesReport,
|
|
20
|
+
SecurityRequirement,
|
|
21
|
+
SecurityScheme,
|
|
22
|
+
SecuritySchemeName,
|
|
23
|
+
SecuritySchemeType,
|
|
24
|
+
SignatureAlgorithm,
|
|
25
|
+
TlsInfo,
|
|
26
|
+
TlsVersion,
|
|
27
|
+
UrlDetails,
|
|
28
|
+
WebpageCaptureReport,
|
|
29
|
+
WebpageScreenshotReport,
|
|
30
|
+
common,
|
|
31
|
+
fingerprint,
|
|
32
|
+
fuzzpath,
|
|
33
|
+
graphql,
|
|
34
|
+
routes,
|
|
35
|
+
webpagecapture,
|
|
36
|
+
)
|
|
37
|
+
|
|
38
|
+
__all__ = [
|
|
39
|
+
"ApiType",
|
|
40
|
+
"Certificate",
|
|
41
|
+
"FingerprintReport",
|
|
42
|
+
"FuzzPathReport",
|
|
43
|
+
"GraphQlData",
|
|
44
|
+
"GraphQlField",
|
|
45
|
+
"GraphQlQuery",
|
|
46
|
+
"GraphQlSchema",
|
|
47
|
+
"GraphQlSchemaData",
|
|
48
|
+
"GraphQlType",
|
|
49
|
+
"HttpHeaders",
|
|
50
|
+
"OAuthFlow",
|
|
51
|
+
"OAuthFlows",
|
|
52
|
+
"PublicKeyAlgorithm",
|
|
53
|
+
"Route",
|
|
54
|
+
"RoutesReport",
|
|
55
|
+
"SecurityRequirement",
|
|
56
|
+
"SecurityScheme",
|
|
57
|
+
"SecuritySchemeName",
|
|
58
|
+
"SecuritySchemeType",
|
|
59
|
+
"SignatureAlgorithm",
|
|
60
|
+
"TlsInfo",
|
|
61
|
+
"TlsVersion",
|
|
62
|
+
"UrlDetails",
|
|
63
|
+
"WebpageCaptureReport",
|
|
64
|
+
"WebpageScreenshotReport",
|
|
65
|
+
"common",
|
|
66
|
+
"fingerprint",
|
|
67
|
+
"fuzzpath",
|
|
68
|
+
"graphql",
|
|
69
|
+
"routes",
|
|
70
|
+
"webpagecapture",
|
|
71
|
+
]
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
|
2
|
+
|
|
3
|
+
from .datetime_utils import serialize_datetime
|
|
4
|
+
from .pydantic_utilities import (
|
|
5
|
+
IS_PYDANTIC_V2,
|
|
6
|
+
UniversalBaseModel,
|
|
7
|
+
UniversalRootModel,
|
|
8
|
+
parse_obj_as,
|
|
9
|
+
universal_field_validator,
|
|
10
|
+
universal_root_validator,
|
|
11
|
+
update_forward_refs,
|
|
12
|
+
)
|
|
13
|
+
from .serialization import FieldMetadata
|
|
14
|
+
|
|
15
|
+
__all__ = [
|
|
16
|
+
"FieldMetadata",
|
|
17
|
+
"IS_PYDANTIC_V2",
|
|
18
|
+
"UniversalBaseModel",
|
|
19
|
+
"UniversalRootModel",
|
|
20
|
+
"parse_obj_as",
|
|
21
|
+
"serialize_datetime",
|
|
22
|
+
"universal_field_validator",
|
|
23
|
+
"universal_root_validator",
|
|
24
|
+
"update_forward_refs",
|
|
25
|
+
]
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
|
2
|
+
|
|
3
|
+
import datetime as dt
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
def serialize_datetime(v: dt.datetime) -> str:
|
|
7
|
+
"""
|
|
8
|
+
Serialize a datetime including timezone info.
|
|
9
|
+
|
|
10
|
+
Uses the timezone info provided if present, otherwise uses the current runtime's timezone info.
|
|
11
|
+
|
|
12
|
+
UTC datetimes end in "Z" while all other timezones are represented as offset from UTC, e.g. +05:00.
|
|
13
|
+
"""
|
|
14
|
+
|
|
15
|
+
def _serialize_zoned_datetime(v: dt.datetime) -> str:
|
|
16
|
+
if v.tzinfo is not None and v.tzinfo.tzname(None) == dt.timezone.utc.tzname(None):
|
|
17
|
+
# UTC is a special case where we use "Z" at the end instead of "+00:00"
|
|
18
|
+
return v.isoformat().replace("+00:00", "Z")
|
|
19
|
+
else:
|
|
20
|
+
# Delegate to the typical +/- offset format
|
|
21
|
+
return v.isoformat()
|
|
22
|
+
|
|
23
|
+
if v.tzinfo is not None:
|
|
24
|
+
return _serialize_zoned_datetime(v)
|
|
25
|
+
else:
|
|
26
|
+
local_tz = dt.datetime.now().astimezone().tzinfo
|
|
27
|
+
localized_dt = v.replace(tzinfo=local_tz)
|
|
28
|
+
return _serialize_zoned_datetime(localized_dt)
|
|
@@ -0,0 +1,249 @@
|
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
|
2
|
+
|
|
3
|
+
# nopycln: file
|
|
4
|
+
import datetime as dt
|
|
5
|
+
import typing
|
|
6
|
+
from collections import defaultdict
|
|
7
|
+
|
|
8
|
+
import typing_extensions
|
|
9
|
+
|
|
10
|
+
import pydantic
|
|
11
|
+
|
|
12
|
+
from .datetime_utils import serialize_datetime
|
|
13
|
+
from .serialization import convert_and_respect_annotation_metadata
|
|
14
|
+
|
|
15
|
+
IS_PYDANTIC_V2 = pydantic.VERSION.startswith("2.")
|
|
16
|
+
|
|
17
|
+
if IS_PYDANTIC_V2:
|
|
18
|
+
# isort will try to reformat the comments on these imports, which breaks mypy
|
|
19
|
+
# isort: off
|
|
20
|
+
from pydantic.v1.datetime_parse import ( # type: ignore # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
21
|
+
parse_date as parse_date,
|
|
22
|
+
)
|
|
23
|
+
from pydantic.v1.datetime_parse import ( # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
24
|
+
parse_datetime as parse_datetime,
|
|
25
|
+
)
|
|
26
|
+
from pydantic.v1.json import ( # type: ignore # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
27
|
+
ENCODERS_BY_TYPE as encoders_by_type,
|
|
28
|
+
)
|
|
29
|
+
from pydantic.v1.typing import ( # type: ignore # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
30
|
+
get_args as get_args,
|
|
31
|
+
)
|
|
32
|
+
from pydantic.v1.typing import ( # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
33
|
+
get_origin as get_origin,
|
|
34
|
+
)
|
|
35
|
+
from pydantic.v1.typing import ( # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
36
|
+
is_literal_type as is_literal_type,
|
|
37
|
+
)
|
|
38
|
+
from pydantic.v1.typing import ( # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
39
|
+
is_union as is_union,
|
|
40
|
+
)
|
|
41
|
+
from pydantic.v1.fields import ModelField as ModelField # type: ignore # pyright: ignore[reportMissingImports] # Pydantic v2
|
|
42
|
+
else:
|
|
43
|
+
from pydantic.datetime_parse import parse_date as parse_date # type: ignore # Pydantic v1
|
|
44
|
+
from pydantic.datetime_parse import parse_datetime as parse_datetime # type: ignore # Pydantic v1
|
|
45
|
+
from pydantic.fields import ModelField as ModelField # type: ignore # Pydantic v1
|
|
46
|
+
from pydantic.json import ENCODERS_BY_TYPE as encoders_by_type # type: ignore # Pydantic v1
|
|
47
|
+
from pydantic.typing import get_args as get_args # type: ignore # Pydantic v1
|
|
48
|
+
from pydantic.typing import get_origin as get_origin # type: ignore # Pydantic v1
|
|
49
|
+
from pydantic.typing import is_literal_type as is_literal_type # type: ignore # Pydantic v1
|
|
50
|
+
from pydantic.typing import is_union as is_union # type: ignore # Pydantic v1
|
|
51
|
+
|
|
52
|
+
# isort: on
|
|
53
|
+
|
|
54
|
+
|
|
55
|
+
T = typing.TypeVar("T")
|
|
56
|
+
Model = typing.TypeVar("Model", bound=pydantic.BaseModel)
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
def parse_obj_as(type_: typing.Type[T], object_: typing.Any) -> T:
|
|
60
|
+
dealiased_object = convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
|
|
61
|
+
if IS_PYDANTIC_V2:
|
|
62
|
+
adapter = pydantic.TypeAdapter(type_) # type: ignore # Pydantic v2
|
|
63
|
+
return adapter.validate_python(dealiased_object)
|
|
64
|
+
else:
|
|
65
|
+
return pydantic.parse_obj_as(type_, dealiased_object)
|
|
66
|
+
|
|
67
|
+
|
|
68
|
+
def to_jsonable_with_fallback(
|
|
69
|
+
obj: typing.Any, fallback_serializer: typing.Callable[[typing.Any], typing.Any]
|
|
70
|
+
) -> typing.Any:
|
|
71
|
+
if IS_PYDANTIC_V2:
|
|
72
|
+
from pydantic_core import to_jsonable_python
|
|
73
|
+
|
|
74
|
+
return to_jsonable_python(obj, fallback=fallback_serializer)
|
|
75
|
+
else:
|
|
76
|
+
return fallback_serializer(obj)
|
|
77
|
+
|
|
78
|
+
|
|
79
|
+
class UniversalBaseModel(pydantic.BaseModel):
|
|
80
|
+
if IS_PYDANTIC_V2:
|
|
81
|
+
model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
|
|
82
|
+
protected_namespaces=(),
|
|
83
|
+
json_encoders={dt.datetime: serialize_datetime},
|
|
84
|
+
) # type: ignore # Pydantic v2
|
|
85
|
+
else:
|
|
86
|
+
|
|
87
|
+
class Config:
|
|
88
|
+
smart_union = True
|
|
89
|
+
json_encoders = {dt.datetime: serialize_datetime}
|
|
90
|
+
|
|
91
|
+
def json(self, **kwargs: typing.Any) -> str:
|
|
92
|
+
kwargs_with_defaults: typing.Any = {
|
|
93
|
+
"by_alias": True,
|
|
94
|
+
"exclude_unset": True,
|
|
95
|
+
**kwargs,
|
|
96
|
+
}
|
|
97
|
+
if IS_PYDANTIC_V2:
|
|
98
|
+
return super().model_dump_json(**kwargs_with_defaults) # type: ignore # Pydantic v2
|
|
99
|
+
else:
|
|
100
|
+
return super().json(**kwargs_with_defaults)
|
|
101
|
+
|
|
102
|
+
def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
|
|
103
|
+
"""
|
|
104
|
+
Override the default dict method to `exclude_unset` by default. This function patches
|
|
105
|
+
`exclude_unset` to work include fields within non-None default values.
|
|
106
|
+
"""
|
|
107
|
+
# Note: the logic here is multi-plexed given the levers exposed in Pydantic V1 vs V2
|
|
108
|
+
# Pydantic V1's .dict can be extremely slow, so we do not want to call it twice.
|
|
109
|
+
#
|
|
110
|
+
# We'd ideally do the same for Pydantic V2, but it shells out to a library to serialize models
|
|
111
|
+
# that we have less control over, and this is less intrusive than custom serializers for now.
|
|
112
|
+
if IS_PYDANTIC_V2:
|
|
113
|
+
kwargs_with_defaults_exclude_unset: typing.Any = {
|
|
114
|
+
**kwargs,
|
|
115
|
+
"by_alias": True,
|
|
116
|
+
"exclude_unset": True,
|
|
117
|
+
"exclude_none": False,
|
|
118
|
+
}
|
|
119
|
+
kwargs_with_defaults_exclude_none: typing.Any = {
|
|
120
|
+
**kwargs,
|
|
121
|
+
"by_alias": True,
|
|
122
|
+
"exclude_none": True,
|
|
123
|
+
"exclude_unset": False,
|
|
124
|
+
}
|
|
125
|
+
dict_dump = deep_union_pydantic_dicts(
|
|
126
|
+
super().model_dump(**kwargs_with_defaults_exclude_unset), # type: ignore # Pydantic v2
|
|
127
|
+
super().model_dump(**kwargs_with_defaults_exclude_none), # type: ignore # Pydantic v2
|
|
128
|
+
)
|
|
129
|
+
|
|
130
|
+
else:
|
|
131
|
+
_fields_set = self.__fields_set__
|
|
132
|
+
|
|
133
|
+
fields = _get_model_fields(self.__class__)
|
|
134
|
+
for name, field in fields.items():
|
|
135
|
+
if name not in _fields_set:
|
|
136
|
+
default = _get_field_default(field)
|
|
137
|
+
|
|
138
|
+
# If the default values are non-null act like they've been set
|
|
139
|
+
# This effectively allows exclude_unset to work like exclude_none where
|
|
140
|
+
# the latter passes through intentionally set none values.
|
|
141
|
+
if default != None:
|
|
142
|
+
_fields_set.add(name)
|
|
143
|
+
|
|
144
|
+
kwargs_with_defaults_exclude_unset_include_fields: typing.Any = {
|
|
145
|
+
"by_alias": True,
|
|
146
|
+
"exclude_unset": True,
|
|
147
|
+
"include": _fields_set,
|
|
148
|
+
**kwargs,
|
|
149
|
+
}
|
|
150
|
+
|
|
151
|
+
dict_dump = super().dict(**kwargs_with_defaults_exclude_unset_include_fields)
|
|
152
|
+
|
|
153
|
+
return convert_and_respect_annotation_metadata(object_=dict_dump, annotation=self.__class__, direction="write")
|
|
154
|
+
|
|
155
|
+
|
|
156
|
+
def deep_union_pydantic_dicts(
|
|
157
|
+
source: typing.Dict[str, typing.Any], destination: typing.Dict[str, typing.Any]
|
|
158
|
+
) -> typing.Dict[str, typing.Any]:
|
|
159
|
+
for key, value in source.items():
|
|
160
|
+
if isinstance(value, dict):
|
|
161
|
+
node = destination.setdefault(key, {})
|
|
162
|
+
deep_union_pydantic_dicts(value, node)
|
|
163
|
+
else:
|
|
164
|
+
destination[key] = value
|
|
165
|
+
|
|
166
|
+
return destination
|
|
167
|
+
|
|
168
|
+
|
|
169
|
+
if IS_PYDANTIC_V2:
|
|
170
|
+
|
|
171
|
+
class V2RootModel(UniversalBaseModel, pydantic.RootModel): # type: ignore # Pydantic v2
|
|
172
|
+
pass
|
|
173
|
+
|
|
174
|
+
UniversalRootModel: typing_extensions.TypeAlias = V2RootModel # type: ignore
|
|
175
|
+
else:
|
|
176
|
+
UniversalRootModel: typing_extensions.TypeAlias = UniversalBaseModel # type: ignore
|
|
177
|
+
|
|
178
|
+
|
|
179
|
+
def encode_by_type(o: typing.Any) -> typing.Any:
|
|
180
|
+
encoders_by_class_tuples: typing.Dict[typing.Callable[[typing.Any], typing.Any], typing.Tuple[typing.Any, ...]] = (
|
|
181
|
+
defaultdict(tuple)
|
|
182
|
+
)
|
|
183
|
+
for type_, encoder in encoders_by_type.items():
|
|
184
|
+
encoders_by_class_tuples[encoder] += (type_,)
|
|
185
|
+
|
|
186
|
+
if type(o) in encoders_by_type:
|
|
187
|
+
return encoders_by_type[type(o)](o)
|
|
188
|
+
for encoder, classes_tuple in encoders_by_class_tuples.items():
|
|
189
|
+
if isinstance(o, classes_tuple):
|
|
190
|
+
return encoder(o)
|
|
191
|
+
|
|
192
|
+
|
|
193
|
+
def update_forward_refs(model: typing.Type["Model"]) -> None:
|
|
194
|
+
if IS_PYDANTIC_V2:
|
|
195
|
+
model.model_rebuild(raise_errors=False) # type: ignore # Pydantic v2
|
|
196
|
+
else:
|
|
197
|
+
model.update_forward_refs()
|
|
198
|
+
|
|
199
|
+
|
|
200
|
+
# Mirrors Pydantic's internal typing
|
|
201
|
+
AnyCallable = typing.Callable[..., typing.Any]
|
|
202
|
+
|
|
203
|
+
|
|
204
|
+
def universal_root_validator(
|
|
205
|
+
pre: bool = False,
|
|
206
|
+
) -> typing.Callable[[AnyCallable], AnyCallable]:
|
|
207
|
+
def decorator(func: AnyCallable) -> AnyCallable:
|
|
208
|
+
if IS_PYDANTIC_V2:
|
|
209
|
+
return pydantic.model_validator(mode="before" if pre else "after")(func) # type: ignore # Pydantic v2
|
|
210
|
+
else:
|
|
211
|
+
return pydantic.root_validator(pre=pre)(func) # type: ignore # Pydantic v1
|
|
212
|
+
|
|
213
|
+
return decorator
|
|
214
|
+
|
|
215
|
+
|
|
216
|
+
def universal_field_validator(field_name: str, pre: bool = False) -> typing.Callable[[AnyCallable], AnyCallable]:
|
|
217
|
+
def decorator(func: AnyCallable) -> AnyCallable:
|
|
218
|
+
if IS_PYDANTIC_V2:
|
|
219
|
+
return pydantic.field_validator(field_name, mode="before" if pre else "after")(func) # type: ignore # Pydantic v2
|
|
220
|
+
else:
|
|
221
|
+
return pydantic.validator(field_name, pre=pre)(func) # type: ignore # Pydantic v1
|
|
222
|
+
|
|
223
|
+
return decorator
|
|
224
|
+
|
|
225
|
+
|
|
226
|
+
PydanticField = typing.Union[ModelField, pydantic.fields.FieldInfo]
|
|
227
|
+
|
|
228
|
+
|
|
229
|
+
def _get_model_fields(
|
|
230
|
+
model: typing.Type["Model"],
|
|
231
|
+
) -> typing.Mapping[str, PydanticField]:
|
|
232
|
+
if IS_PYDANTIC_V2:
|
|
233
|
+
return model.model_fields # type: ignore # Pydantic v2
|
|
234
|
+
else:
|
|
235
|
+
return model.__fields__ # type: ignore # Pydantic v1
|
|
236
|
+
|
|
237
|
+
|
|
238
|
+
def _get_field_default(field: PydanticField) -> typing.Any:
|
|
239
|
+
try:
|
|
240
|
+
value = field.get_default() # type: ignore # Pydantic < v1.10.15
|
|
241
|
+
except:
|
|
242
|
+
value = field.default
|
|
243
|
+
if IS_PYDANTIC_V2:
|
|
244
|
+
from pydantic_core import PydanticUndefined
|
|
245
|
+
|
|
246
|
+
if value == PydanticUndefined:
|
|
247
|
+
return None
|
|
248
|
+
return value
|
|
249
|
+
return value
|