#!/usr/bin/python# -*- coding: utf-8 -*-# utils.py#  #  Author: Laudeci Oliveira <laudeci@gmail.com>#          Rafael Proen莽a   <cypherbios@ubuntu.com># #  This program is free software; you can redistribute it and/or #  modify it under the terms of the GNU General Public License as #  published by the Free Software Foundation; either version 2 of the#  License, or (at your option) any later version.# #  This program is distributed in the hope that it will be useful,#  but WITHOUT ANY WARRANTY; without even the implied warranty of#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the#  GNU General Public License for more details.# #  You should have received a copy of the GNU General Public License#  along with this program; if not, write to the Free Software#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307#  USAimport apt_instimport apt_pkgimport gtkimport mathimport fnmatchimport osimport pygtkimport shutilimport urllibimport gnomevfsimport reimport tarfileimport bz2import configdef isValidFileName(filename):    try:        p=re.compile('[^0-9A-Za-z_\+\.,-]+')        if p.search(filename):            return False        else:            return True    except:        return False    def updateUI():    while gtk.events_pending(): gtk.main_iteration() #  Do case insensitive wildcard search on column 1.def TreeViewSearch( model, column, key, it, data):    key, title = key.lower(), model.get_value(it, column).lower()    return title.find(key) == -1        def getCDROM():    vol_monitor =  gnomevfs.VolumeMonitor()    volume = gnomevfs.Volume()    drives = []    for volume in vol_monitor.get_mounted_volumes():        if volume.get_hal_udi() != None:            vol_type = volume.get_device_type()            if vol_type == gnomevfs.DEVICE_TYPE_CDROM or vol_type == gnomevfs.DEVICE_TYPE_VIDEO_DVD:                drives.append(volume.get_drive())        return drives            def packageTooltip(widget, e, tooltips, cell, emptyText="no information"):     """ If emptyText is None, the cursor has to enter widget from a side that contains an item, otherwise no tooltip will be displayed. """     try:         (path, col, x, y) = widget.get_path_at_pos(int(e.x), int(e.y))         it = widget.get_model().get_iter(path)         value = widget.get_model().get_value(it, config.C_PKG)         tooltips.set_tip(widget, value.get_pkg_ShortDescription())         tooltips.enable()     except:         tooltips.set_tip(widget, emptyText)            def checkAccess(path):    """Checks file/folder permissions and returns True if the file/folder could be accessed"""    mRet = False    if pathExists(path):        if os.access(path, os.R_OK | os.W_OK | os.X_OK):            mRet = True        return mRetdef copyFile(source , destination):    try:        dest = os.path.dirname(destination)        if not pathExists(dest):            mkdir(dest)                shutil.copyfile(source , destination)    except Exception , e :        print str(e)def delFile( path):    """ Deletes a file if it exists"""    if fileExist(path):        os.remove(path)    def removePath( path):    """ Removes a directory path """    try:        shutil.rmtree(path, ignore_errors=True)    except:        pass        def pathExists( path):    """Returns True if a path exists."""    return os.path.isdir(path)    def fileExist( path):    """Returns True if a file exist."""    return os.path.isfile(path)    def mkdir( path, removeExisting = False):    """ Create a full path, directory by directory"""    #remove directiory if already exists    if removeExisting:        if pathExists(path):            removePath(path)                    dirs = path.split("/")    dir = ""            for dirname in dirs:        dir += dirname + "/"        if not os.path.isdir(dir) and len(dir)>0:            os.mkdir(dir)            def TimeFormat(value):    #get days        fDays = (value / 86400)    tModDay = math.modf(fDays)    Days = int(tModDay[1])    decimalDays =tModDay[0]        fTimeHour = decimalDays  * 24     # Extract decimal from integer part    tModHour = math.modf(fTimeHour)    nHour = int(tModHour[1])    fDecimalHour = tModHour[0]        # Transform decimal in minutes    fMinute = fDecimalHour*60    # Again, extract the decimal and the integer part    tModMinute = math.modf(fMinute)    nMinute = int(tModMinute[1])    fDecimalMinute = tModMinute[0]    # Transform decimal in seconds    fSecond = fDecimalMinute*60        # Again, extract the decimal and the integer part    tModSecond = math.modf(fSecond)    nSecond = int(tModSecond[1])            output=''    if Days >0:        output = '%id' % Days    if nHour >0:        output+='%ih' % nHour        if nMinute >0:        output+='%im' % nMinute        output+='%is' % nSecond        return outputdef fileSizeFormat( i):    """        Taken from jinja        Format the value like a 'human-readable' file size (i.e. 13 KB, 4.1 MB,        102 bytes, etc).    """    bytes = float(i)    if bytes < 1024:        return u"%d Byte%s" % (bytes, bytes != 1 and u's' or u'')    elif bytes < 1024 * 1024:        return u"%.1f KB" % (bytes / 1024)    elif bytes < 1024 * 1024 * 1024:        return u"%.1f MB" % (bytes / (1024 * 1024))    else:        return u"%.1f GB" % (bytes / (1024 * 1024 * 1024)) def openSelectDialog(title, filterName, pattern, selectMultiple = False):            file_open = gtk.FileChooserDialog(title=title,                           action=gtk.FILE_CHOOSER_ACTION_OPEN,                           buttons=(gtk.STOCK_CANCEL,                           gtk.RESPONSE_CANCEL,                           gtk.STOCK_OPEN, gtk.RESPONSE_OK))    filter = gtk.FileFilter()    filter.set_name(filterName)    filter.add_pattern(pattern)    file_open.add_filter(filter)    file_open.set_select_multiple(selectMultiple)            result = ""    path = ""    filename = ""    if file_open.run() == gtk.RESPONSE_OK:        file_open.hide()        results = file_open.get_filenames()        file_open.destroy()        return gtk.RESPONSE_OK, results    else:        file_open.destroy()        return gtk.RESPONSE_CANCEL, []        def getFilePathFromUrl( uri):     """ helper to get a useful path from a drop uri"""     path = urllib.url2pathname(uri) # escape special chars     path = path.strip('\r\n\x00') # remove \r\n and NULL     # get the path to file     if path.startswith('file:\\\\\\'): # windows         path = path[8:] # 8 is len('file:///')     elif path.startswith('file://'): # nautilus, rox         path = path[7:] # 7 is len('file://')     elif path.startswith('file:'): # xffm         path = path[5:] # 5 is len('file:')     return path          class cursorManager:	def __init__(self):		self.window = None	#procedure to change window cursor to ready cursor	def setBusy(self, window_main = None, flag = True):		""" Show a watch cursor if the app is busy for more than 0.3 sec.		Furthermore provide a loop to handle user interface events """		if window_main is None:			return		self.window = window_main.window		if flag == True:			try:				self.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.WATCH))			except:				print "An error occured changing cursor"		else:			self.window.set_cursor(None)			while gtk.events_pending():				gtk.main_iteration()		return    class ContextMenu(gtk.Menu):    def __init__(self, *args):        gtk.Menu.__init__(self)        self.menuItem = None             def addMenuItem(self,menuName,actionFunction= None):                if menuName == "-":            self.menuItem = gtk.SeparatorMenuItem()        else:             self.menuItem = gtk.MenuItem(menuName)            if actionFunction is not None :                self.menuItem.connect("activate", actionFunction)        self.menuItem.show()        self.append(self.menuItem)        return    class SystemInfo:    def __init__(self):        self.distro = ''        self.architecture = ''        self.codename = ''        self.release = ''        self.read_release_info()            def read_release_info(self):        self.get_infos()        self.architecture = self.__architecture()       def get_infos(self):          try:            p = open('/etc/lsb-release','r')            for line in p:                if 'DISTRIB_ID' in line:                    self.distro = line.split('=')[-1].replace('\n','').lower()                elif 'DISTRIB_CODENAME' in line:                    self.codename = line.split('=')[-1].replace('\n','').lower()                elif 'DISTRIB_RELEASE' in line:                    self.release = line.split('=')[-1].replace('\n','').lower()            p.close()        except:            pass    def __architecture(self):        """        Detect whether the architecture is x86/ppc/amd64        """        arch = os.uname()[-1]        if arch in ('ppc', 'ppc64'):            arch = 'powerpc'        elif arch in ('sparc32','sparc64','sparc'):            arch = 'sparc'        elif arch =='x86_64':            if self.distro in ('debian'):                arch = 'ia64'            else:                arch = 'amd64'        elif arch in ('i386','i686','i586','k7'):            arch = 'i386'        return arch        class RecursiveSearch:    def __init__(self, directory, pattern="*"):        self.stack = [directory]        self.pattern = pattern        self.files = []        self.index = 0    def __getitem__(self, index):        while 1:            try:                file = self.files[self.index]                self.index = self.index + 1            except IndexError:                # pop next directory from stack                self.directory = self.stack.pop()                self.files = os.listdir(self.directory)                self.index = 0            else:                # got a filename                fullname = os.path.join(self.directory, file)                if os.path.isdir(fullname) and not os.path.islink(fullname):                    self.stack.append(fullname)                if fnmatch.fnmatch(file, self.pattern):                    return fullname                    def compress(source, dest):    dest_ext = '.bz2'    arcname = os.path.basename (source)    dest_name = '%s%s' % (dest, dest_ext)    dest_path = os.path.join(dest, dest_name)    input = bz2.compress(file(source, 'r').read())    out = file(dest_path,'w')    out.write(input)    out.close()    return dest_path