Commit 4dfe156
committed
Parse dlc versions
1 parent 0ba016c commit 4dfe156
1 file changed
Lines changed: 15 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -8,7 +8,10 @@ | |||
| 8 | 8 | # | |
| 9 | 9 | # Licensed under GNU Lesser General Public License v3.0 | |
| 10 | 10 | # | |
| 11 | + from typing import Tuple | ||
| 12 | + | ||
| 11 | 13 | from PySide6 import QtCore | |
| 14 | + import re | ||
| 12 | 15 | ||
| 13 | 16 | ||
| 14 | 17 | class Worker(QtCore.QObject): | |
@@ -38,6 +41,17 @@ def stop_thread(): | |||
| 38 | 41 | return worker, thread | |
| 39 | 42 | ||
| 40 | 43 | ||
| 44 | + def parse_version(version: str) -> Tuple[int, int, int]: | ||
| 45 | + """ | ||
| 46 | + Parses a version string into a tuple of (major, minor, patch). | ||
| 47 | + """ | ||
| 48 | + match = re.search(r"(\d+)\.(\d+)\.(\d+)", version) | ||
| 49 | + if match: | ||
| 50 | + return tuple(int(part) for part in match.groups()) | ||
| 51 | + else: | ||
| 52 | + raise ValueError(f"Invalid version format: {version}") | ||
| 53 | + | ||
| 54 | + | ||
| 41 | 55 | def is_latest_deeplabcut_version(): | |
| 42 | 56 | import json | |
| 43 | 57 | import urllib.request | |
@@ -46,4 +60,4 @@ def is_latest_deeplabcut_version(): | |||
| 46 | 60 | url = "https://pypi.org/pypi/deeplabcut/json" | |
| 47 | 61 | contents = urllib.request.urlopen(url).read() | |
| 48 | 62 | latest_version = json.loads(contents)["info"]["version"] | |
| 49 | - return VERSION == latest_version, latest_version | ||
| 63 | + return parse_version(VERSION) >= parse_version(latest_version), latest_version | ||
0 commit comments