Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Improvements to version info message #7594

Merged
merged 2 commits into from Sep 25, 2023
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
35 changes: 24 additions & 11 deletions pydantic/version.py
Expand Up @@ -8,9 +8,9 @@


def version_short() -> str:
"""Return the major.minor part of Pydantic version.
"""Return the `major.minor` part of Pydantic version.

It return '2.1' if Pydantic version is '2.1.1'.
It returns '2.1' if Pydantic version is '2.1.1'.
"""
return '.'.join(VERSION.split('.')[:2])

Expand All @@ -19,18 +19,31 @@ def version_info() -> str:
"""Return complete version information for Pydantic and its dependencies."""
import platform
import sys
from importlib import import_module
from pathlib import Path

import pydantic_core._pydantic_core as pdc

optional_deps = []
for p in 'devtools', 'email-validator', 'typing-extensions':
try:
import_module(p.replace('-', '_'))
except ImportError: # pragma: no cover
continue
optional_deps.append(p)
if sys.version_info >= (3, 8):
import importlib.metadata as importlib_metadata
else:
import importlib_metadata

# get data about packages that are closely related to pydantic, use pydantic or often conflict with pydantic
package_names = {
'email-validator',
'fastapi',
'mypy',
'pydantic-extra-types',
'pydantic-settings',
'pyright',
'typing_extensions',
}
related_packages = []

for dist in importlib_metadata.distributions():
name = dist.metadata['Name']
if name in package_names:
related_packages.append(f'{name}-{dist.version}')

info = {
'pydantic version': VERSION,
Expand All @@ -39,7 +52,7 @@ def version_info() -> str:
'install path': Path(__file__).resolve().parent,
'python version': sys.version,
'platform': platform.platform(),
'optional deps. installed': optional_deps,
'related packages': ' '.join(related_packages),
}
return '\n'.join('{:>30} {}'.format(k + ':', str(v).replace('\n', ' ')) for k, v in info.items())

Expand Down