remove uuid.txt if there is no file in the directory

1 files changed, 7 insertions, 0 deletions

diff --git a/scan.py b/scan.py
index 3b47df1..5eee604 100755
--- a/scan.py
+++ b/scan.py
@@ -37,8 +37,10 @@ def main():
         for root, dirs, files in os.walk(START_DIR):
             if not len(files):
                 continue
+
             files, uuids = partition(lambda f: '/uuid.txt' in f,
                                     (os.path.join(root, f) for f in files))
+
             if len(uuids) == 1:
                 my_uuid = open(uuids[0]).readline().strip()
             else:
@@ -47,6 +49,11 @@ def main():
                 for f in uuids:
                     os.unlink(f)
                 my_uuid = create_uuid_txt(root)
+
+            if len(files) == 0:
+                os.unlink(os.path.join(root, 'uuid.txt'))
+                continue
+
             db.append((my_uuid, [(f, os.stat(f).st_size) for f in files]))
 
         pickle.dump(db, out)