From 27ca36687aed8d5ae36a7712567201b87bf2be02 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Fri, 13 Sep 2019 17:08:27 +0300 Subject: [PATCH] Add support for upcoming trains-server --- trains/backend_api/session/session.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/trains/backend_api/session/session.py b/trains/backend_api/session/session.py index b69ff4c1..b3e50d8c 100644 --- a/trains/backend_api/session/session.py +++ b/trains/backend_api/session/session.py @@ -485,6 +485,16 @@ class Session(TokenManager): return urlunparse(parsed) + @classmethod + def check_min_api_version(cls, min_api_version): + """ + Return True if Session.api_version is greater or equal >= to min_api_version + """ + def version_tuple(v): + v = tuple(map(int, (v.split(".")))) + return v + (0,) * max(0, 3 - len(v)) + return version_tuple(cls.api_version) >= version_tuple(str(min_api_version)) + def _do_refresh_token(self, old_token, exp=None): """ TokenManager abstract method implementation. Here we ignore the old token and simply obtain a new token.