From ea38f95d6df738d94aced5e3f3cfa69525248e12 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Sun, 19 Feb 2017 12:57:55 -0800
Subject: [PATCH] add automatic upgrade feature

---
 anki_connect.py | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/anki_connect.py b/anki_connect.py
index 12b2da4..5e950a7 100644
--- a/anki_connect.py
+++ b/anki_connect.py
@@ -31,6 +31,7 @@ import socket
 API_VERSION = 2
 TICK_INTERVAL = 25
 URL_TIMEOUT = 10
+URL_UPGRADE = 'https://raw.githubusercontent.com/FooSoft/anki-connect/master/anki_connect.py'
 NET_ADDRESS = '127.0.0.1'
 NET_BACKLOG = 5
 NET_PORT = 8765
@@ -72,7 +73,7 @@ def makeStr(data):
     return data.decode('utf-8')
 
 
-def audioDownload(url):
+def download(url):
     try:
         resp = web.urlopen(url, timeout=URL_TIMEOUT)
     except web.URLError:
@@ -321,7 +322,7 @@ class AnkiBridge:
             return
 
         if params.audio is not None and len(params.audio.fields) > 0:
-            data = audioDownload(params.audio.url)
+            data = download(params.audio.url)
             if data is not None:
                 if params.audio.skipHash is None:
                     skip = False
@@ -435,7 +436,7 @@ class AnkiConnect:
             QMessageBox.critical(
                 self.anki.window(),
                 'AnkiConnect',
-                'Failed to listen on port {}...\nMake sure it is available and is not in use.'.format(NET_PORT)
+                'Failed to listen on port {}.\nMake sure it is available and is not in use.'.format(NET_PORT)
             )
 
     def advance(self):
@@ -490,6 +491,23 @@ class AnkiConnect:
         return results
 
 
+    def api_upgrade(self):
+        response = QMessageBox.question(
+            self.anki.window(),
+            'AnkiConnect',
+            'Upgrade to the latest version?',
+            QMessageBox.Yes | QMessageBox.No
+        )
+
+        if response == QMessageBox.Yes:
+            data = download(URL_UPGRADE)
+            if data is None:
+                QMessageBox.critical(self.anki.window, 'AnkiConnect', 'Failed to download latest version')
+            else:
+                with open(__file__, 'w') as fp:
+                    fp.write(data)
+
+
     def api_version(self):
         return API_VERSION