hopefully securely fixed updater

tdolist · Nov 8, 2015 · ce310af · ce310af
1 parent 6275a1c
commit ce310af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tdo/todolist/maintenance.py b/tdo/todolist/maintenance.py
@@ -50,14 +50,14 @@ def update():
         with zipfile.ZipFile('dl.zip', "r") as z:
             z.extractall(".")
         os.remove('dl.zip')
-        newdir = ''
+        newdir = None
         for entry in os.listdir():
             if re.match(r'{user}\-tdo\-.*'.format(user=user), entry):
                 newdir = entry
         os.chdir(newdir)
         subprocess.call(['python3', './setup.py', 'install'])
         os.chdir('..')
-        if newdir != '':
+        if newdir is not None:
             shutil.rmtree(newdir)
 
     except AttributeError: