Skip to content

Commit 4ace0d3

Browse files
authored
fix deprecation warning for __version__ (#543)
1 parent a33e151 commit 4ace0d3

File tree

2 files changed

+16
-1
lines changed

2 files changed

+16
-1
lines changed

tests/test_config.py

+15-1
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,31 @@
11
import json
2+
import sys
23
from datetime import date
34
from datetime import timedelta
45

56
import pytest
67
from dateutil.relativedelta import relativedelta
7-
from flask import __version__ as flask_version
88
from flask import Flask
99

1010
from flask_jwt_extended import JWTManager
1111
from flask_jwt_extended.config import config
1212
from flask_jwt_extended.internal_utils import JSONEncoder
1313

1414

15+
def get_package_version(package_name):
16+
if sys.version_info >= (3, 8):
17+
# Use importlib.metadata for Python 3.8 and newer
18+
import importlib.metadata
19+
20+
return importlib.metadata.version(package_name)
21+
else:
22+
# Use pkg_resources for older versions
23+
import pkg_resources
24+
25+
return pkg_resources.get_distribution(package_name).version
26+
27+
28+
flask_version = get_package_version("flask")
1529
flask_version_tuple = tuple(map(int, flask_version.split(".")))
1630

1731

tox.ini

+1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ deps =
2727
python-dateutil
2828
types-python-dateutil
2929
mypy
30+
types-setuptools
3031

3132
[testenv:coverage]
3233
commands =

0 commit comments

Comments
 (0)