From b08c1f6699da163a0b631bff9cb5b74a0b8d3fd5 Mon Sep 17 00:00:00 2001
From: Donald Cheng Hong Zou <donaldzou@live.hk>
Date: Thu, 2 Sep 2021 21:56:50 -0400
Subject: [PATCH] Updated requirement.txt and wgd.sh

---
 src/requirements.txt | 10 +++++-----
 src/wgd.sh           |  5 +++++
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/requirements.txt b/src/requirements.txt
index 511897c..58178ca 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -1,5 +1,5 @@
-Flask==2.0.1
-tinydb==4.3.0
-ifcfg==0.21
-icmplib==3.0.1
-flask-qrcode==3.0.0
\ No newline at end of file
+Flask
+tinydb
+ifcfg
+icmplib
+flask-qrcode
\ No newline at end of file
diff --git a/src/wgd.sh b/src/wgd.sh
index acf0dad..97a9a3d 100755
--- a/src/wgd.sh
+++ b/src/wgd.sh
@@ -53,9 +53,14 @@ update_wgd() {
     printf "%s\n" "$dashes"
     printf "| Shutting down Wireguard Dashboard...                     |\n"
     printf "%s\n" "$dashes"
+    stop_wgd
     printf "| Downloading %s from GitHub...                            |\n" "$new_ver"
     printf "%s\n" "$dashes"
+    git stash > /dev/null 2>&1
     git pull https://github.com/donaldzou/wireguard-dashboard.git $new_ver --force >  /dev/null 2>&1
+    printf "%s\n" "$dashes"
+    printf "| Installing latest Python dependencies                    |\n"
+    python3 -m pip install -r requirements.txt
     printf "| Update Successfully!                                     |\n"
     printf "%s\n" "$dashes"
     printf "| Now you can start the dashboard with >> sh wgd.sh start  |\n"