diff options
Diffstat (limited to 'python/deprecation.py')
| -rw-r--r-- | python/deprecation.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/python/deprecation.py b/python/deprecation.py index fecb3fbc..e6c5a482 100644 --- a/python/deprecation.py +++ b/python/deprecation.py @@ -9,6 +9,7 @@ # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the # License for the specific language governing permissions and limitations # under the License. +import re import collections import functools import textwrap @@ -30,6 +31,15 @@ __all__ = ["deprecated", "message_location", "fail_if_not_removed", message_location = "bottom" +# This is a patch applied to this library to allow specifying a deprecation version +def parse_version(version: str) -> tuple: + match = re.match(r"^(?P<major>\d+)\.(?P<minor>\d+)(\.(?P<patch>\d+))?", version) + if match is None: + return (0,0,0) + groups = match.groupdict() + return (int(groups['major']), int(groups['minor']), int(groups['patch'] or '0')) + + class DeprecatedWarning(DeprecationWarning): """A warning class for deprecated methods @@ -165,6 +175,15 @@ def deprecated(deprecated_in=None, removed_in=None, current_version=None, is_unsupported = True else: is_deprecated = True + elif current_version: + current_version = parse_version(current_version) + + if (removed_in + and current_version >= parse_version(removed_in)): + is_unsupported = True + elif (deprecated_in + and current_version >= parse_version(deprecated_in)): + is_deprecated = True else: # If we can't actually calculate that we're in a period of # deprecation...well, they used the decorator, so it's deprecated. |