blob: 6c1106d77f493bab6b22aa4e9b81d50ad8ac563a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
# Copyright 2019 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from distutils import version # pylint: disable=no-name-in-module
def RequireVersion(module, min_version, max_version=None):
"""Ensure that an imported module's version is within a required range.
Version strings are parsed with LooseVersion, so versions like "1.8.0rc1"
(default numpy on macOS Sierra) and "2.4.13.2" (a version of OpenCV 2.x)
are allowed.
Args:
module: An already imported python module.
min_version: The module must have this or a higher version.
max_version: Optional, the module should not have this or a higher version.
Raises:
ImportError if the module's __version__ is not within the allowed range.
"""
module_version = version.LooseVersion(module.__version__)
min_version = version.LooseVersion(str(min_version))
valid_version = min_version <= module_version
if max_version is not None:
max_version = version.LooseVersion(str(max_version))
valid_version = valid_version and (module_version < max_version)
wants_version = 'at or above %s and below %s' % (min_version, max_version)
else:
wants_version = '%s or higher' % min_version
if not valid_version:
raise ImportError('%s has version %s, but version %s is required' % (
module.__name__, module_version, wants_version))
|
