Added updater functions and corrected URLs

0xInfection · Jan 17, 2020 · 25ddab6 · 25ddab6
1 parent fe580c3
commit 25ddab6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/xsrfprobe/core/updater.py b/xsrfprobe/core/updater.py
@@ -11,17 +11,18 @@
 
 import os
 from requests import get
+from xsrfprobe import __version__
 from xsrfprobe.core.colors import *
 
 def updater():
     '''
     Function to update XSRFProbe seamlessly.
     '''
     print(GR+'Checking for updates...')
-    vno = get('https://github.com/raw/0xInfection/XSRFProbe/master/files/VersionNum').text
+    vno = get('https://github.com/raw/0xInfection/XSRFProbe/master/xsrfprobe/files/VersionNum').text
     print(GR+'Version on GitHub: '+color.CYAN+vno.strip())
-    print(GR+'Version You Have : '+color.CYAN+open('files/VersionNum').read())
-    if vno != open('files/VersionNum').read():
+    print(GR+'Version You Have : '+color.CYAN+__version__)
+    if vno != __version__:
         print(G+'A new version of XSRFProbe is available!')
         current_path = os.getcwd().split('/') # if you know it, you know it
         folder = current_path[-1] # current directory name