dupeguru/base/py/app_se_cocoa.py

# Created By: Virgil Dupras
# Created On: 2009-05-24
# $Id$
# Copyright 2009 Hardcoded Software (http://www.hardcoded.net)
# 
# This software is licensed under the "HS" License as described in the "LICENSE" file, 
# which should be included with this package. The terms are also available at 
# http://www.hardcoded.net/licenses/hs_license

import logging

from AppKit import *

from hsfs.phys import Directory as DirectoryBase
from hsfs.phys.bundle import Bundle
from hsutil.path import Path
from hsutil.misc import extract
from hsutil.str import get_file_ext

from . import app_cocoa, data
from .directories import Directories as DirectoriesBase, STATE_EXCLUDED

if NSWorkspace.sharedWorkspace().respondsToSelector_('typeOfFile:error:'): # Only from 10.5
    def is_bundle(str_path):
        sw = NSWorkspace.sharedWorkspace()
        uti, error = sw.typeOfFile_error_(str_path)
        if error is not None:
            logging.warning(u'There was an error trying to detect the UTI of %s', str_path)
        return sw.type_conformsToType_(uti, 'com.apple.bundle') or sw.type_conformsToType_(uti, 'com.apple.package')
else: # Tiger
    def is_bundle(str_path): # just return a list of a few known bundle extensions.
        return get_file_ext(str_path) in ('app', 'pages', 'numbers')

class DGDirectory(DirectoryBase):
    def _create_sub_file(self, name, with_parent=True):
        if is_bundle(unicode(self.path + name)):
            parent = self if with_parent else None
            return Bundle(parent, name)
        else:
            return super(DGDirectory, self)._create_sub_file(name, with_parent)
    
    def _fetch_subitems(self):
        subdirs, subfiles = super(DGDirectory, self)._fetch_subitems()
        apps, normal_dirs = extract(lambda name: is_bundle(unicode(self.path + name)), subdirs)
        subfiles += apps
        return normal_dirs, subfiles
    

class Directories(DirectoriesBase):
    ROOT_PATH_TO_EXCLUDE = map(Path, ['/Library', '/Volumes', '/System', '/bin', '/sbin', '/opt', '/private', '/dev'])
    HOME_PATH_TO_EXCLUDE = [Path('Library')]
    def __init__(self):
        DirectoriesBase.__init__(self)
        self.dirclass = DGDirectory
    
    def _default_state_for_path(self, path):
        result = DirectoriesBase._default_state_for_path(self, path)
        if result is not None:
            return result
        if path in self.ROOT_PATH_TO_EXCLUDE:
            return STATE_EXCLUDED
        if path[:2] == Path('/Users') and path[3:] in self.HOME_PATH_TO_EXCLUDE:
            return STATE_EXCLUDED
    

class DupeGuru(app_cocoa.DupeGuru):
    def __init__(self):
        app_cocoa.DupeGuru.__init__(self, data, 'dupeGuru', appid=4)
        self.directories = Directories()