rox/shell/directory.py

   1 """
   2 @copyright: (C) 2008, Thomas Leonard
   3 @see: U{http://roscidus.com}
   4 """
   5 from __future__ import with_statement
   6 import gobject, stat
   7 import _gio as gio
   8 import gtk
   9 import weakref
  10 from logging import warn
  11
  12 gtk_theme = gtk.icon_theme_get_default()
  13
  14 icon_size = 48
  15
  16 def get_themed_icon(name, icon_size):
  17         try:
  18                 return gtk_theme.load_icon('text-x-generic', icon_size, 0)
  19         except gobject.GError:
  20                 return None
  21
  22 icon_text_plain = get_themed_icon('text-x-generic', icon_size)
  23 icon_dir = get_themed_icon('folder', icon_size)
  24
  25 def sort_key(name, info):
  26         t = 9 - int(info.get_file_type())
  27         return str(t) + name.lower()
  28
  29 COLOUR_EXEC   = '#008000'
  30 COLOUR_NORMAL = '#000000'
  31 COLOUR_DIR    = '#0000a0'
  32 COLOUR_ERROR  = '#800000'
  33
  34 QUERY = 'standard::*,unix::mode,thumbnail::path'
  35
  36 def is_hidden(name):
  37         return name.startswith('.')
  38
  39 class ErrorInfo:
  40         def __init__(self, name):
  41                 self.name = name
  42
  43         def get_name(self):
  44                 return self.name
  45
  46         def get_file_type(self):
  47                 return gio.FILE_TYPE_UNKNOWN
  48
  49 def make_error_row(name, ex):
  50         info = ErrorInfo(name)
  51         return [name, None, info, sort_key(name, info), COLOUR_ERROR]
  52
  53 def make_row(info):
  54         name = info.get_name()
  55         if is_hidden(name):
  56                 return None
  57
  58         pixbuf = None
  59         thumbnail = info.get_attribute_byte_string('thumbnail::path')
  60         if thumbnail:
  61                 try:
  62                         loader = gtk.gdk.PixbufLoader('png')
  63                         try:
  64                                 with open(thumbnail) as stream:
  65                                         loader.write(stream.read())
  66                         finally:
  67                                 loader.close()
  68                         pixbuf = loader.get_pixbuf()
  69                         assert pixbuf, "Failed to load PNG thumbnail"
  70                 except Exception, ex:
  71                         warn("Failed to load cached PNG thumbnail icon: %s", ex)
  72         else:
  73                 icon = info.get_icon()
  74                 if icon:
  75                         gtkicon_info = gtk_theme.choose_icon(icon.get_names(), icon_size, 0)
  76                         if gtkicon_info:
  77                                 gtkicon_info.get_filename()
  78                                 pixbuf = gtkicon_info.load_icon()
  79         mode = info.get_attribute_uint32('unix::mode')
  80         if stat.S_ISDIR(mode):
  81                 colour = COLOUR_DIR
  82         elif mode & 0111:
  83                 colour = COLOUR_EXEC
  84         else:
  85                 colour = COLOUR_NORMAL
  86         return [name, pixbuf, info, sort_key(name, info), colour]
  87
  88 class DirModel:
  89         NAME = 0
  90         PIXBUF = 1
  91         INFO = 2
  92         SORT = 3                # Type-then-name
  93         COLOUR = 4
  94
  95         model = None
  96         monitor = None
  97
  98         def __init__(self, file):
  99                 assert isinstance(file, gio.File), file
 100                 self.file = file
 101                 self.users = []
 102
 103         def __del__(self):
 104                 assert not self.users, "DirModel garbage collected while still monitoring! Monitored by " + str(self.users)
 105
 106         def add_ref(self, user):
 107                 user = id(user)
 108
 109                 if not self.users:
 110                         # First user...
 111                         assert not self.monitor
 112                         assert not self.model
 113                         self.model = gtk.ListStore(str, gtk.gdk.Pixbuf, object, str, str)
 114
 115                         # Should really be the view that sorts, but see GTK bug #523724
 116                         self.model.set_sort_column_id(DirModel.SORT, gtk.SORT_ASCENDING)
 117
 118                         self.monitor = self.file.monitor_directory(0)
 119                         weakself = weakref.ref(self)
 120                         self.monitor.connect('changed', lambda *args: weakself().contents_changed(*args))
 121                         self.build_contents()
 122
 123                 self.users.append(user)
 124
 125         def del_ref(self, user):
 126                 user = id(user)
 127                 self.users.remove(user)
 128
 129                 if not self.users:
 130                         # Last user...
 131                         self.model.clear()
 132                         self.model = None
 133                         self.monitor.cancel()
 134                         self.monitor = None
 135
 136         def contents_changed(self, monitor, this_file, other_file, event_type):
 137                 name = this_file.get_basename()
 138                 if is_hidden(name):
 139                         return
 140                 #print "build", monitor, this_file.get_basename(), other_file, event_type
 141                 if event_type == gio.FILE_MONITOR_EVENT_CREATED:
 142                         #print "Create", name
 143                         i = self.model.append(None)
 144                 elif event_type in (gio.FILE_MONITOR_EVENT_DELETED, gio.FILE_MONITOR_EVENT_CHANGED, gio.FILE_MONITOR_EVENT_ATTRIBUTE_CHANGED):
 145                         i = self.model.get_iter_root()
 146                         while i:
 147                                 if self.model[i][DirModel.NAME] == name:
 148                                         break
 149                                 i = self.model.iter_next(i)
 150                         else:
 151                                 #print "Note: deleted unknown file", name
 152                                 self.build_contents()
 153                 elif event_type == gio.FILE_MONITOR_EVENT_CHANGES_DONE_HINT:
 154                         return
 155                 else:
 156                         print "Unknown event type", event_type, name
 157                         self.build_contents()
 158
 159                 if event_type in (gio.FILE_MONITOR_EVENT_CREATED, gio.FILE_MONITOR_EVENT_CHANGED, gio.FILE_MONITOR_EVENT_ATTRIBUTE_CHANGED):
 160                         #print "Update", name
 161                         try:
 162                                 info = this_file.query_info(QUERY, 0)
 163                         except gobject.GError, ex:
 164                                 self.model[i] = make_error_row(name, ex)
 165                         else:
 166                                 self.model[i] = make_row(info)
 167                 elif event_type == gio.FILE_MONITOR_EVENT_DELETED:
 168                         #print "Remove", name
 169                         self.model.remove(i)
 170                 else:
 171                         assert False, "Unreachable! " + str(event_type)
 172
 173         def build_contents(self):
 174                 m = self.model
 175                 name_to_iter = {}
 176
 177                 i = m.get_iter_root()
 178                 while i:
 179                         name_to_iter[m[i][DirModel.NAME]] = i
 180                         i = m.iter_next(i)
 181
 182                 self.error = None
 183
 184                 files = []
 185                 try:
 186                         e = self.file.enumerate_children(QUERY, 0)
 187                 except gobject.GError, ex:
 188                         self.error = ex
 189                 else:
 190                         if e:
 191                                 while True:
 192                                         info = e.next_file()
 193                                         if not info:
 194                                                 break
 195                                         row = make_row(info)
 196                                         if row:
 197                                                 files.append(row)
 198
 199                 # name_to_iter contains the old contents
 200                 # files contains the desired contents
 201
 202                 new = []
 203                 for row in files:
 204                         name = row[DirModel.NAME]
 205
 206                         i = name_to_iter.get(name, None)
 207                         if i:
 208                                 m[i] = row
 209                                 del name_to_iter[name]
 210                         else:
 211                                 new.append(row)
 212
 213                 # name_to_iter contains old contents that are no longer desired
 214                 # new contains rows that are desired but were not present
 215
 216                 i = m.get_iter_root()
 217                 while i:
 218                         if m[i][DirModel.NAME] in name_to_iter:
 219                                 if not m.remove(i):
 220                                         i = None
 221                         else:
 222                                 i = m.iter_next(i)
 223
 224                 for row in new:
 225                         i = m.append(None)
 226                         m[i] = row
 227
 228 dirs = weakref.WeakValueDictionary()
 229 def get_dir_model(file):
 230         # Might want some caching here...
 231         uri = file.get_uri()
 232         try:
 233                 return dirs[uri]
 234         except KeyError:
 235                 dm = DirModel(file)
 236                 dirs[uri] = dm
 237                 return dm