codeboje.de

#!/usr/bin/env python  
# -*- coding: utf-8 -*-  
import sys, os, wx  
from wx.lib.mixins.listctrl import CheckListCtrlMixin  
from stat import *  

ID_DELETE=100  

def listDrives():  
    if sys.platform == 'win32':  
        try:  
            import win32api,string  
            drives=win32api.GetLogicalDriveStrings()  
            drives=string.splitfields(drives,'\000')  
            return drives  
        except: # in case PyWin is not present we do it the classical way  
            drives = []  
            for i in range(ord('a'), ord('z')+1):  
                drive = chr(i)  
                if(os.path.exists(drive +":\\")):  
                    drives.append(drive+":\\")  
            return drives  


def listFreeSpace(drive):  
    try:  
        if sys.platform == 'win32':  
            import win32api,string  
            info=win32api.GetDiskFreeSpaceEx(drive)  
            return info  
    except: # most probably pywin is not present  
        print "Unexpected error:", sys.exc_info()  

def listFilesByName(listOfFiles,directory,files):  
    for fileName in files:  
        filepath = os.path.join(directory,fileName)  
        if os.path.isfile(filepath):  
            filesize = os.stat(filepath)[ST_SIZE]  
            listOfFiles[filepath] = filesize  

def scan(directories):   
    listOfFiles = {}  
    for directory in directories:   
        sys.stdout.write('Scanning directory '+directory+'...\n')           
        directory = os.path.abspath(directory)  
        os.path.walk(directory,listFilesByName,listOfFiles)         
    sys.stdout.write('Done scanning all dirs. Now sorting and building GUI.\n')  
    allFiles = listOfFiles.items()  
    allFiles.sort(list_compare)  
    sortedList = allFiles  
    return sortedList  

def list_compare(listX, listY):  
    if listX[1]>listY[1]:  
        return -1  
    elif listX[1]==listY[1]:  
        return 0  
    else: # x<y  
        return 1  

class CheckListCtrl(wx.ListCtrl, CheckListCtrlMixin):  
    def __init__(self, parent):  
        wx.ListCtrl.__init__(self, parent, -1, style=wx.LC_REPORT)  
        CheckListCtrlMixin.__init__(self)  
        self.log = sys.stdout  
        self.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnItemActivated)  

    def OnItemActivated(self, evt):  
        self.ToggleItem(evt.m_itemIndex)  

class FilesFrame(wx.Frame):       
    def __init__(self, parent, files):   
        wx.Frame.__init__(   
            self, parent, -1, "HDD Clean", size=(640,480)   
            )   
        p = wx.Panel(self, -1, style=0)   
        b = wx.Button(p, -1, "Delete...")   
        b.SetDefault()   
        self.Bind(wx.EVT_BUTTON, self.OnDelete, b)   
        self.files=files  
        self.list = CheckListCtrl(p)  
        sizer = wx.BoxSizer(wx.VERTICAL)  
        sizer.Add(b, 0, wx.EXPAND)  
        sizer.Add(self.list, 1, wx.EXPAND)  
        p.SetSizer(sizer)  

        self.list.InsertColumn(0, "Delete")  
        self.list.InsertColumn(1, "Size",wx.LIST_FORMAT_RIGHT)  
        self.list.InsertColumn(2, "File")  
        self.list.InsertColumn(3, "Type")  


        for i in self.files:  
            index = self.list.InsertStringItem(sys.maxint,"")  
            self.list.SetStringItem(index, 1, str(i[1]))  
            self.list.SetStringItem(index, 2, str(i[0]))  
            indexForFileType=str(i[0]).rfind(".")+1  
            filetype="None"  
            if indexForFileType!=0 and (len(str(i[0]))-indexForFileType<10):  
                filetype=str(i[0])[indexForFileType:]  
            self.list.SetStringItem(index, 3, filetype)  

        self.list.SetColumnWidth(0, wx.LIST_AUTOSIZE)  
        self.list.SetColumnWidth(1, wx.LIST_AUTOSIZE)  
        self.list.SetColumnWidth(2, wx.LIST_AUTOSIZE)  
        self.list.SetColumnWidth(3, wx.LIST_AUTOSIZE)  

        self.Show(1)  

    def OnDelete(self, evt):   
        items=self.list.GetItemCount()  
        allItems = range(0,items)  
        allItems.reverse()  
        for item in allItems:  
            if self.list.IsChecked(item):  
                listItem=self.list.GetItem(item,col=2)  
                self.list.DeleteItem(item)  
                os.remove(listItem.GetText())  

myapp=wx.PySimpleApp()  
files={}  
if len(sys.argv)>1:  
    files=scan(sys.argv[1:])  
else:  
    allDrives = listDrives()  
    if allDrives==None:  
        files = scan("/")  
    else:  
        files= scan(allDrives)  
if len(files)>0:  
    FilesFrame(None,files)  
    myapp.MainLoop()  
else:  
    print "No files found, exiting !"