From 22f25ed331cd3ce3988313bafcd083a1c31c99cd Mon Sep 17 00:00:00 2001
From: 68cdrBxM8YdoJ <68cdrBxM8YdoJ@gmail.com>
Date: Thu, 23 Apr 2020 17:53:27 +0900
Subject: [PATCH] Change check_update func

When using a pre-release, the previous release
version always appears as the new update.
---
 AV_Data_Capture.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/AV_Data_Capture.py b/AV_Data_Capture.py
index ebb0322..207ccb9 100755
--- a/AV_Data_Capture.py
+++ b/AV_Data_Capture.py
@@ -2,14 +2,20 @@ import argparse
 from core import *
 
 
-def check_update(current_version):
+def check_update(local_version):
     data = json.loads(get_html("https://api.github.com/repos/yoshiko2/AV_Data_Capture/releases/latest"))
 
-    remote_version = data["tag_name"]
+    try:
+        remote = float(data["tag_name"])
+        local = float(local_version)
+    except ValueError:
+        print("[-] Check update failed! Skipped.")
+        return
+
     download_url = data["html_url"]
 
-    if current_version != remote_version:
-        line1 = "* New update " + remote_version + " *"
+    if local < remote:
+        line1 = "* New update " + str(remote) + " *"
         print("[*]" + line1.center(54))
         print("[*]" + "↓ Download ↓".center(54))
         print("[*] " + download_url)