Added Settings menu

Author: coderSuren

components/convert.py

@@ -1,7 +1,17 @@
-def toGB(s):
-    return round(s/(1024*1024*1024),3)
-
-def toKB(s):
-    return round(s/(1024),3)
-def toMB(s):
-    return round(s/(1024*1024),3)
+def convert_bytes(bytes):
+   bytes = float(bytes)
+   if bytes >= 1099511627776:
+      terabytes = bytes / 1099511627776
+      size = '%.2fTb' % terabytes
+   elif bytes >= 1073741824:
+      gigabytes = bytes / 1073741824
+      size = '%.2fGb' % gigabytes
+   elif bytes >= 1048576:
+      megabytes = bytes / 1048576
+      size = '%.2fMb' % megabytes
+   elif bytes >= 1024:
+      kilobytes = bytes / 1024
+      size = '%.2fKb' % kilobytes
+   else:
+      size = '%.2fb' % bytes
+   return size

components/diskutil.py

@@ -1,6 +1,6 @@
 import wmi
 import os
-
+from convert import convert_bytes
 
 def get_disk_usage():
     key = wmi.WMI()
@@ -16,40 +16,26 @@ def get_disk_usage():
 
 
 
-def convert_bytes(bytes):
-   bytes = float(bytes)
-   if bytes >= 1099511627776:
-      terabytes = bytes / 1099511627776
-      size = '%.2fTb' % terabytes
-   elif bytes >= 1073741824:
-      gigabytes = bytes / 1073741824
-      size = '%.2fGb' % gigabytes
-   elif bytes >= 1048576:
-      megabytes = bytes / 1048576
-      size = '%.2fMb' % megabytes
-   elif bytes >= 1024:
-      kilobytes = bytes / 1024
-      size = '%.2fKb' % kilobytes
-   else:
-      size = '%.2fb' % bytes
-   return size
 
 def get_file_usage(dir):
     typesize = {}
+    typecount = {}
     try:
         for root, dirs, files in os.walk(dir):
             for file in files:
                 prefix, extension = os.path.splitext(file)
                 if extension not in typesize:
                     typesize[extension] = 0
+                    typecount[extension] = 0
                 typesize[extension] += os.stat(root + os.sep + file).st_size
+                typecount[extension] += 1
     except KeyboardInterrupt:
         pass
 
     # print(typesize)
     result =list(sorted(typesize.items(), key=lambda x:-x[1]))
-
+    
     for i in range(len(result)):
-        result[i] = [result[i][0], convert_bytes(result[i][1])]
+        result[i] = [result[i][0], convert_bytes(result[i][1]), typecount[result[i][0]]]
 
     return result

components/duplicates.py

@@ -1,7 +1,10 @@
 import os
 import sys
-import hashlib
 
+from hashlib import md5, blake2b, sha1, sha256, sha512, sha3_256, sha3_512, shake_128, shake_256
+from xxhash import xxh64, xxh128, xxh3_64, xxh3_128
+
+hashfunc="md5"
 
 def duplicates(folders):
     dup_size = {}
@@ -62,7 +65,38 @@ def join_dicts(dict1, dict2):
 
 def hashfile(path, blocksize=65536):
     file = open(path, 'rb')
-    hasher = hashlib.md5()
+    hasher = ""
+    match hashfunc:
+        case "md5":  
+            hasher = md5()
+        case "sha1":
+            hasher = sha1()
+        case "blake2b":
+            hasher = blake2b()
+        case "sha256":
+            hasher = sha256()
+        case "sha512":
+            hasher = sha512()
+        case "sha3_256":
+            hasher = sha3_256()
+        case "sha3_512":
+            hasher = sha3_512()
+        case "shake_128":
+            hasher = shake_128()
+        case "shake_256":
+            hasher = shake_256()
+        case "xxh64":
+            hasher = xxh64()
+        case "xxh128":
+            hasher = xxh128()
+        case "xxh3_64":
+            hasher = xxh3_64()
+        case "xxh3_128":
+            hasher = xxh3_128()
+        case _:
+            print("Invalid hash function")
+            sys.exit(1)
+    
     buf = file.read(blocksize)
     while len(buf) > 0:
         hasher.update(buf)
@@ -87,8 +121,11 @@ def print_results(dict1):
         print('No duplicate files found.')
 
 
-def find_duplicates(dir):
+def find_duplicates(dir, func):
     # dir=input("Enter the directory names to find for duplicates: ").split(" ")
+    global hashfunc 
+    hashfunc = func
+    print(hashfunc)
     a= duplicates([dir])
     return a
 

components/gui.py

@@ -1,20 +1,28 @@
 import glob
 import os
 import time
-from convert import toGB
-
-def infrequent_files(dir): 
+from convert import convert_bytes
+def infrequent_files(dir, days): 
+    today = time.strftime('%m/%d/%Y', time.gmtime())
     list_of_files = filter( os.path.isfile, glob.glob(dir + '/**/*') )
-    list_of_files = sorted( list_of_files, key = os.path.getmtime)
-
+    list_of_files = sorted( list_of_files, key = os.path.getatime)
     infrequent_files = []
     for file_path in list_of_files:
-        timestamp_str = time.strftime(  '%m/%d/%Y',
-                                    time.gmtime(os.path.getmtime(file_path))) 
-        file_size = os.path.getsize(file_path)
-        infrequent_files.append([file_path,file_size,timestamp_str]) 
+        timestamp_str = time.strftime(  '%m/%d/%Y', time.gmtime(os.path.getatime(file_path))) 
+        if compare_dates(timestamp_str, today, days):    
+            file_size = os.path.getsize(file_path)
+            infrequent_files.append([file_path,convert_bytes(file_size),timestamp_str]) 
     return infrequent_files
 
+def compare_dates(date1, date2, days):
+    date1 = time.strptime(date1, '%m/%d/%Y')
+    date2 = time.strptime(date2, '%m/%d/%Y')
+    difference = time.mktime(date2) - time.mktime(date1)
+    difference = difference / (24 * 60 * 60)
+    if difference > days:
+        return True
+    return False
+
 def delete_infrequent_files(files): 
     if files:
         for file in files:

components/infrequent.py

@@ -1,14 +1,15 @@
 import os
+from convert import convert_bytes
 
-def find_large_files(path):
-    # path=input("Enter the directory name to find large files: ")
+def find_large_files(path, file_size_limit):
+    limit=file_size_limit*1024*1024
     ans=[]
     if os.path.exists(path):
         for foldername, subfolders, filenames in os.walk(path):
             for filename in filenames:
                 size=os.path.getsize(os.path.join(foldername, filename))
-                if size>100000000: # 100 MB
-                    ans.append([foldername + '\\' + filename, size])
+                if size>limit: 
+                    ans.append([foldername + '\\' + filename, convert_bytes(size)])
 
         return ans
     else:

components/largefiles.py

@@ -9,17 +9,21 @@
 # create a class file that  imports functions from inported files and returns them in seperate functions
 
 class Scanner:
+    dataFormat = "bytes"
+    hashalgo = "md5"
+    file_size_limit = 100
+    time_limit = 30
     def __init__(self):
         super().__init__()
 
     def find_duplicates(self, folders):
-        return duplicates.find_duplicates(folders)
+        return duplicates.find_duplicates(folders, self.hashalgo)
 
     def remove_duplicates(self, folders):
         return duplicates.remove_duplicates(folders)
 
     def find_large_files(self, folder):
-        return largefiles.find_large_files(folder)
+        return largefiles.find_large_files(folder, self.file_size_limit)
 
     def find_disk_utilization(self):
         return diskutil.get_disk_usage()
@@ -28,7 +32,7 @@ def find_file_usage(self, folder):
         return diskutil.get_file_usage(folder)
 
     def show_infrequent_files(self, folder):
-        return infrequent.infrequent_files(folder)
+        return infrequent.infrequent_files(folder, self.time_limit)
 
     def delete_infrequent_files(self, files):
         return infrequent.delete_infrequent_files(files)