Skip to content

Commit 82d2b25

Browse files
authored
Avoid importing packaging or pkg_resources for version validation (#948)
Importing the `pkg_resources` module has high memory and startup time cost. A recent change in 102e01c already avoided it for loading extensions, but it's still used for validating that __version__ is correctly formatted. It is possible to avoid it by installing the `packaging` package, but that adds a dependency for something quite trivial. Instead, remove the validation and add tests which check the output is as expected. Since `setuptools` is no longer required at runtime, remove it from `install_required`.
1 parent 7daa674 commit 82d2b25

File tree

4 files changed

+38
-18
lines changed

4 files changed

+38
-18
lines changed

docs/change_log/index.md

+2
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@ Under development: version 3.2.2 (a bug-fix release).
1010
* Correctly report if an extension raises a `TypeError` (#939).
1111
* Raise a `KeyError` when attempting to delete a nonexistent key from the
1212
extension registry (#939).
13+
* Remove import of `packaging` (or `pkg_resources` fallback) entirely.
14+
* Remove `setuptools` as a run-time dependency (`install_required`).
1315

1416
Feb 12, 2020: Released version 3.2.1 (a bug-fix release).
1517

markdown/__meta__.py

+11-17
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,6 @@
1919
License: BSD (see LICENSE.md for details).
2020
"""
2121

22-
try:
23-
import packaging.version
24-
except ImportError:
25-
from pkg_resources.extern import packaging
26-
2722
# __version_info__ format:
2823
# (major, minor, patch, dev/alpha/beta/rc/final, #)
2924
# (1, 1, 2, 'dev', 0) => "1.1.2.dev0"
@@ -34,22 +29,21 @@
3429
__version_info__ = (3, 2, 1, 'final', 0)
3530

3631

37-
def _get_version(): # pragma: no cover
32+
def _get_version(version_info):
3833
" Returns a PEP 440-compliant version number from version_info. "
39-
assert len(__version_info__) == 5
40-
assert __version_info__[3] in ('dev', 'alpha', 'beta', 'rc', 'final')
34+
assert len(version_info) == 5
35+
assert version_info[3] in ('dev', 'alpha', 'beta', 'rc', 'final')
4136

42-
parts = 2 if __version_info__[2] == 0 else 3
43-
v = '.'.join(map(str, __version_info__[:parts]))
37+
parts = 2 if version_info[2] == 0 else 3
38+
v = '.'.join(map(str, version_info[:parts]))
4439

45-
if __version_info__[3] == 'dev':
46-
v += '.dev' + str(__version_info__[4])
47-
elif __version_info__[3] != 'final':
40+
if version_info[3] == 'dev':
41+
v += '.dev' + str(version_info[4])
42+
elif version_info[3] != 'final':
4843
mapping = {'alpha': 'a', 'beta': 'b', 'rc': 'rc'}
49-
v += mapping[__version_info__[3]] + str(__version_info__[4])
44+
v += mapping[version_info[3]] + str(version_info[4])
5045

51-
# Ensure version is valid and normalized
52-
return str(packaging.version.Version(v))
46+
return v
5347

5448

55-
__version__ = _get_version()
49+
__version__ = _get_version(__version_info__)

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def get_version():
8888
license='BSD License',
8989
packages=['markdown', 'markdown.extensions'],
9090
python_requires='>=3.5',
91-
install_requires=['setuptools >= 36', "importlib_metadata;python_version<'3.8'"],
91+
install_requires=["importlib_metadata;python_version<'3.8'"],
9292
extras_require={
9393
'testing': [
9494
'coverage',

tests/test_meta.py

+24
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
import unittest
2+
from markdown.__meta__ import _get_version, __version__
3+
4+
5+
class TestVersion(unittest.TestCase):
6+
7+
def test_get_version(self):
8+
"""Test that _get_version formats __version_info__ as required by PEP 440."""
9+
10+
self.assertEqual(_get_version((1, 1, 2, 'dev', 0)), "1.1.2.dev0")
11+
self.assertEqual(_get_version((1, 1, 2, 'alpha', 1)), "1.1.2a1")
12+
self.assertEqual(_get_version((1, 2, 0, 'beta', 2)), "1.2b2")
13+
self.assertEqual(_get_version((1, 2, 0, 'rc', 4)), "1.2rc4")
14+
self.assertEqual(_get_version((1, 2, 0, 'final', 0)), "1.2")
15+
16+
def test__version__IsValid(self):
17+
"""Test that __version__ is valid and normalized."""
18+
19+
try:
20+
import packaging.version
21+
except ImportError:
22+
from pkg_resources.extern import packaging
23+
24+
self.assertEqual(__version__, str(packaging.version.Version(__version__)))

0 commit comments

Comments
 (0)