From decb56d2041b6a2ef73ff6ef938f70d94cbe5a31 Mon Sep 17 00:00:00 2001 From: Yuanchen Wang Date: Tue, 30 Jan 2024 20:57:00 +1100 Subject: [PATCH] fixed the issue for checking flask version in python3.7 --- tests/test_config.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/tests/test_config.py b/tests/test_config.py index 2326f244..fa938795 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -1,5 +1,5 @@ -import importlib import json +import sys from datetime import date from datetime import timedelta @@ -12,7 +12,20 @@ from flask_jwt_extended.internal_utils import JSONEncoder -flask_version = importlib.metadata.version("flask") +def get_package_version(package_name): + if sys.version_info >= (3, 8): + # Use importlib.metadata for Python 3.8 and newer + import importlib.metadata + + return importlib.metadata.version(package_name) + else: + # Use pkg_resources for older versions + import pkg_resources + + return pkg_resources.get_distribution(package_name).version + + +flask_version = get_package_version("flask") flask_version_tuple = tuple(map(int, flask_version.split(".")))