src/profile_window.py

   1 # -*- coding:utf-8 -*-
   2 ## src/profile_window.py
   3 ##
   4 ## Copyright (C) 2003-2010 Yann Leboulanger <asterix AT lagaule.org>
   5 ## Copyright (C) 2005-2006 Nikos Kouremenos <kourem AT gmail.com>
   6 ## Copyright (C) 2006-2008 Jean-Marie Traissard <jim AT lapin.org>
   7 ##
   8 ## This file is part of Gajim.
   9 ##
  10 ## Gajim is free software; you can redistribute it and/or modify
  11 ## it under the terms of the GNU General Public License as published
  12 ## by the Free Software Foundation; version 3 only.
  13 ##
  14 ## Gajim is distributed in the hope that it will be useful,
  15 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17 ## GNU General Public License for more details.
  18 ##
  19 ## You should have received a copy of the GNU General Public License
  20 ## along with Gajim. If not, see <http://www.gnu.org/licenses/>.
  21 ##
  22
  23 # THIS FILE IS FOR **OUR** PROFILE (when we edit our INFO)
  24
  25 import gtk
  26 import gobject
  27 import base64
  28 import mimetypes
  29 import os
  30 import time
  31
  32 import gtkgui_helpers
  33 import dialogs
  34 import vcard
  35
  36 from common import gajim
  37 from common import ged
  38
  39
  40 class ProfileWindow:
  41     """
  42     Class for our information window
  43     """
  44
  45     def __init__(self, account):
  46         self.xml = gtkgui_helpers.get_gtk_builder('profile_window.ui')
  47         self.window = self.xml.get_object('profile_window')
  48         self.window.set_transient_for(gajim.interface.roster.window)
  49         self.progressbar = self.xml.get_object('progressbar')
  50         self.statusbar = self.xml.get_object('statusbar')
  51         self.context_id = self.statusbar.get_context_id('profile')
  52
  53         self.account = account
  54         self.jid = gajim.get_jid_from_account(account)
  55
  56         self.dialog = None
  57         self.avatar_mime_type = None
  58         self.avatar_encoded = None
  59         self.message_id = self.statusbar.push(self.context_id,
  60             _('Retrieving profile...'))
  61         self.update_progressbar_timeout_id = gobject.timeout_add(100,
  62             self.update_progressbar)
  63         self.remove_statusbar_timeout_id = None
  64
  65         # Create Image for avatar button
  66         image = gtk.Image()
  67         self.xml.get_object('PHOTO_button').set_image(image)
  68         self.xml.connect_signals(self)
  69         gajim.ged.register_event_handler('vcard-published', ged.GUI1,
  70             self._nec_vcard_published)
  71         gajim.ged.register_event_handler('vcard-not-published', ged.GUI1,
  72             self._nec_vcard_not_published)
  73         gajim.ged.register_event_handler('vcard-received', ged.GUI1,
  74             self._nec_vcard_received)
  75         self.window.show_all()
  76
  77     def update_progressbar(self):
  78         self.progressbar.pulse()
  79         return True # loop forever
  80
  81     def remove_statusbar(self, message_id):
  82         self.statusbar.remove_message(self.context_id, message_id)
  83         self.remove_statusbar_timeout_id = None
  84
  85     def on_profile_window_destroy(self, widget):
  86         if self.update_progressbar_timeout_id is not None:
  87             gobject.source_remove(self.update_progressbar_timeout_id)
  88         if self.remove_statusbar_timeout_id is not None:
  89             gobject.source_remove(self.remove_statusbar_timeout_id)
  90         gajim.ged.remove_event_handler('vcard-published', ged.GUI1,
  91             self._nec_vcard_published)
  92         gajim.ged.remove_event_handler('vcard-not-published', ged.GUI1,
  93             self._nec_vcard_not_published)
  94         gajim.ged.remove_event_handler('vcard-received', ged.GUI1,
  95             self._nec_vcard_received)
  96         del gajim.interface.instances[self.account]['profile']
  97         if self.dialog: # Image chooser dialog
  98             self.dialog.destroy()
  99
 100     def on_profile_window_key_press_event(self, widget, event):
 101         if event.keyval == gtk.keysyms.Escape:
 102             self.window.destroy()
 103
 104     def on_clear_button_clicked(self, widget):
 105         # empty the image
 106         button = self.xml.get_object('PHOTO_button')
 107         image = button.get_image()
 108         image.set_from_pixbuf(None)
 109         button.hide()
 110         text_button = self.xml.get_object('NOPHOTO_button')
 111         text_button.show()
 112         self.avatar_encoded = None
 113         self.avatar_mime_type = None
 114
 115     def on_set_avatar_button_clicked(self, widget):
 116         def on_ok(widget, path_to_file):
 117             must_delete = False
 118             filesize = os.path.getsize(path_to_file) # in bytes
 119             invalid_file = False
 120             msg = ''
 121             if os.path.isfile(path_to_file):
 122                 stat = os.stat(path_to_file)
 123                 if stat[6] == 0:
 124                     invalid_file = True
 125                     msg = _('File is empty')
 126             else:
 127                 invalid_file = True
 128                 msg = _('File does not exist')
 129             if not invalid_file and filesize > 16384: # 16 kb
 130                 try:
 131                     pixbuf = gtk.gdk.pixbuf_new_from_file(path_to_file)
 132                     # get the image at 'notification size'
 133                     # and hope that user did not specify in ACE crazy size
 134                     scaled_pixbuf = gtkgui_helpers.get_scaled_pixbuf(pixbuf,
 135                             'tooltip')
 136                 except gobject.GError, msg: # unknown format
 137                     # msg should be string, not object instance
 138                     msg = str(msg)
 139                     invalid_file = True
 140             if invalid_file:
 141                 if True: # keep identation
 142                     dialogs.ErrorDialog(_('Could not load image'), msg)
 143                     return
 144             if filesize > 16384:
 145                 if scaled_pixbuf:
 146                     path_to_file = os.path.join(gajim.TMP,
 147                             'avatar_scaled.png')
 148                     scaled_pixbuf.save(path_to_file, 'png')
 149                     must_delete = True
 150
 151             fd = open(path_to_file, 'rb')
 152             data = fd.read()
 153             pixbuf = gtkgui_helpers.get_pixbuf_from_data(data)
 154             try:
 155                 # rescale it
 156                 pixbuf = gtkgui_helpers.get_scaled_pixbuf(pixbuf, 'vcard')
 157             except AttributeError: # unknown format
 158                 dialogs.ErrorDialog(_('Could not load image'))
 159                 return
 160             self.dialog.destroy()
 161             self.dialog = None
 162             button = self.xml.get_object('PHOTO_button')
 163             image = button.get_image()
 164             image.set_from_pixbuf(pixbuf)
 165             button.show()
 166             text_button = self.xml.get_object('NOPHOTO_button')
 167             text_button.hide()
 168             self.avatar_encoded = base64.encodestring(data)
 169             # returns None if unknown type
 170             self.avatar_mime_type = mimetypes.guess_type(path_to_file)[0]
 171             if must_delete:
 172                 try:
 173                     os.remove(path_to_file)
 174                 except OSError:
 175                     gajim.log.debug('Cannot remove %s' % path_to_file)
 176
 177         def on_clear(widget):
 178             self.dialog.destroy()
 179             self.dialog = None
 180             self.on_clear_button_clicked(widget)
 181
 182         def on_cancel(widget):
 183             self.dialog.destroy()
 184             self.dialog = None
 185
 186         if self.dialog:
 187             self.dialog.present()
 188         else:
 189             self.dialog = dialogs.AvatarChooserDialog(on_response_ok = on_ok,
 190                     on_response_cancel = on_cancel, on_response_clear = on_clear)
 191
 192     def on_PHOTO_button_press_event(self, widget, event):
 193         """
 194         If right-clicked, show popup
 195         """
 196         if event.button == 3 and self.avatar_encoded: # right click
 197             menu = gtk.Menu()
 198
 199             # Try to get pixbuf
 200             pixbuf = gtkgui_helpers.get_avatar_pixbuf_from_cache(self.jid,
 201                     use_local=False)
 202
 203             if pixbuf not in (None, 'ask'):
 204                 nick = gajim.config.get_per('accounts', self.account, 'name')
 205                 menuitem = gtk.ImageMenuItem(gtk.STOCK_SAVE_AS)
 206                 menuitem.connect('activate',
 207                     gtkgui_helpers.on_avatar_save_as_menuitem_activate,
 208                     self.jid, nick)
 209                 menu.append(menuitem)
 210             # show clear
 211             menuitem = gtk.ImageMenuItem(gtk.STOCK_CLEAR)
 212             menuitem.connect('activate', self.on_clear_button_clicked)
 213             menu.append(menuitem)
 214             menu.connect('selection-done', lambda w:w.destroy())
 215             # show the menu
 216             menu.show_all()
 217             menu.popup(None, None, None, event.button, event.time)
 218         elif event.button == 1: # left click
 219             self.on_set_avatar_button_clicked(widget)
 220
 221     def on_BDAY_entry_focus_out_event(self, widget, event):
 222         txt = widget.get_text()
 223         if not txt:
 224             return
 225         try:
 226             time.strptime(txt, '%Y-%m-%d')
 227         except ValueError:
 228             if not widget.is_focus():
 229                 pritext = _('Wrong date format')
 230                 dialogs.ErrorDialog(pritext, _('Format of the date must be '
 231                     'YYYY-MM-DD'))
 232                 gobject.idle_add(lambda: widget.grab_focus())
 233             return True
 234
 235     def set_value(self, entry_name, value):
 236         try:
 237             self.xml.get_object(entry_name).set_text(value)
 238         except AttributeError:
 239             pass
 240
 241     def set_values(self, vcard_):
 242         button = self.xml.get_object('PHOTO_button')
 243         image = button.get_image()
 244         text_button = self.xml.get_object('NOPHOTO_button')
 245         if not 'PHOTO' in vcard_:
 246             # set default image
 247             image.set_from_pixbuf(None)
 248             button.hide()
 249             text_button.show()
 250         for i in vcard_.keys():
 251             if i == 'PHOTO':
 252                 pixbuf, self.avatar_encoded, self.avatar_mime_type = \
 253                         vcard.get_avatar_pixbuf_encoded_mime(vcard_[i])
 254                 if not pixbuf:
 255                     image.set_from_pixbuf(None)
 256                     button.hide()
 257                     text_button.show()
 258                     continue
 259                 pixbuf = gtkgui_helpers.get_scaled_pixbuf(pixbuf, 'vcard')
 260                 image.set_from_pixbuf(pixbuf)
 261                 button.show()
 262                 text_button.hide()
 263                 continue
 264             if i == 'ADR' or i == 'TEL' or i == 'EMAIL':
 265                 for entry in vcard_[i]:
 266                     add_on = '_HOME'
 267                     if 'WORK' in entry:
 268                         add_on = '_WORK'
 269                     for j in entry.keys():
 270                         self.set_value(i + add_on + '_' + j + '_entry', entry[j])
 271             if isinstance(vcard_[i], dict):
 272                 for j in vcard_[i].keys():
 273                     self.set_value(i + '_' + j + '_entry', vcard_[i][j])
 274             else:
 275                 if i == 'DESC':
 276                     self.xml.get_object('DESC_textview').get_buffer().set_text(
 277                             vcard_[i], 0)
 278                 else:
 279                     self.set_value(i + '_entry', vcard_[i])
 280         if self.update_progressbar_timeout_id is not None:
 281             if self.message_id:
 282                 self.statusbar.remove_message(self.context_id, self.message_id)
 283             self.message_id = self.statusbar.push(self.context_id,
 284                     _('Information received'))
 285             self.remove_statusbar_timeout_id = gobject.timeout_add_seconds(3,
 286                     self.remove_statusbar, self.message_id)
 287             gobject.source_remove(self.update_progressbar_timeout_id)
 288             self.progressbar.hide()
 289             self.progressbar.set_fraction(0)
 290             self.update_progressbar_timeout_id = None
 291
 292     def _nec_vcard_received(self, obj):
 293         if obj.conn.name != self.account:
 294             return
 295         if obj.jid != self.jid:
 296             return
 297         self.set_values(obj.vcard_dict)
 298
 299     def add_to_vcard(self, vcard_, entry, txt):
 300         """
 301         Add an information to the vCard dictionary
 302         """
 303         entries = entry.split('_')
 304         loc = vcard_
 305         if len(entries) == 3: # We need to use lists
 306             if entries[0] not in loc:
 307                 loc[entries[0]] = []
 308             found = False
 309             for e in loc[entries[0]]:
 310                 if entries[1] in e:
 311                     e[entries[2]] = txt
 312                     break
 313             else:
 314                 loc[entries[0]].append({entries[1]: '', entries[2]: txt})
 315             return vcard_
 316         while len(entries) > 1:
 317             if entries[0] not in loc:
 318                 loc[entries[0]] = {}
 319             loc = loc[entries[0]]
 320             del entries[0]
 321         loc[entries[0]] = txt
 322         return vcard_
 323
 324     def make_vcard(self):
 325         """
 326         Make the vCard dictionary
 327         """
 328         entries = ['FN', 'NICKNAME', 'BDAY', 'EMAIL_HOME_USERID', 'URL',
 329                 'TEL_HOME_NUMBER', 'N_FAMILY', 'N_GIVEN', 'N_MIDDLE', 'N_PREFIX',
 330                 'N_SUFFIX', 'ADR_HOME_STREET', 'ADR_HOME_EXTADR', 'ADR_HOME_LOCALITY',
 331                 'ADR_HOME_REGION', 'ADR_HOME_PCODE', 'ADR_HOME_CTRY', 'ORG_ORGNAME',
 332                 'ORG_ORGUNIT', 'TITLE', 'ROLE', 'TEL_WORK_NUMBER', 'EMAIL_WORK_USERID',
 333                 'ADR_WORK_STREET', 'ADR_WORK_EXTADR', 'ADR_WORK_LOCALITY',
 334                 'ADR_WORK_REGION', 'ADR_WORK_PCODE', 'ADR_WORK_CTRY']
 335         vcard_ = {}
 336         for e in entries:
 337             txt = self.xml.get_object(e + '_entry').get_text().decode('utf-8')
 338             if txt != '':
 339                 vcard_ = self.add_to_vcard(vcard_, e, txt)
 340
 341         # DESC textview
 342         buff = self.xml.get_object('DESC_textview').get_buffer()
 343         start_iter = buff.get_start_iter()
 344         end_iter = buff.get_end_iter()
 345         txt = buff.get_text(start_iter, end_iter, 0)
 346         if txt != '':
 347             vcard_['DESC'] = txt.decode('utf-8')
 348
 349         # Avatar
 350         if self.avatar_encoded:
 351             vcard_['PHOTO'] = {'BINVAL': self.avatar_encoded}
 352             if self.avatar_mime_type:
 353                 vcard_['PHOTO']['TYPE'] = self.avatar_mime_type
 354         return vcard_
 355
 356     def on_ok_button_clicked(self, widget):
 357         if self.update_progressbar_timeout_id:
 358             # Operation in progress
 359             return
 360         if gajim.connections[self.account].connected < 2:
 361             dialogs.ErrorDialog(_('You are not connected to the server'),
 362                     _('Without a connection you can not publish your contact '
 363                     'information.'))
 364             return
 365         vcard_ = self.make_vcard()
 366         nick = ''
 367         if 'NICKNAME' in vcard_:
 368             nick = vcard_['NICKNAME']
 369             gajim.connections[self.account].send_nickname(nick)
 370         if nick == '':
 371             nick = gajim.config.get_per('accounts', self.account, 'name')
 372         gajim.nicks[self.account] = nick
 373         gajim.connections[self.account].send_vcard(vcard_)
 374         self.message_id = self.statusbar.push(self.context_id,
 375                 _('Sending profile...'))
 376         self.progressbar.show()
 377         self.update_progressbar_timeout_id = gobject.timeout_add(100,
 378                 self.update_progressbar)
 379
 380     def _nec_vcard_published(self, obj):
 381         if obj.conn.name != self.account:
 382             return
 383         if self.update_progressbar_timeout_id is not None:
 384             gobject.source_remove(self.update_progressbar_timeout_id)
 385             self.update_progressbar_timeout_id = None
 386         self.window.destroy()
 387
 388     def _nec_vcard_not_published(self, obj):
 389         if obj.conn.name != self.account:
 390             return
 391         if self.message_id:
 392             self.statusbar.remove_message(self.context_id, self.message_id)
 393         self.message_id = self.statusbar.push(self.context_id,
 394             _('Information NOT published'))
 395         self.remove_statusbar_timeout_id = gobject.timeout_add_seconds(3,
 396             self.remove_statusbar, self.message_id)
 397         if self.update_progressbar_timeout_id is not None:
 398             gobject.source_remove(self.update_progressbar_timeout_id)
 399             self.progressbar.set_fraction(0)
 400             self.update_progressbar_timeout_id = None
 401         dialogs.InformationDialog(_('vCard publication failed'),
 402             _('There was an error while publishing your personal information, '
 403             'try again later.'))
 404
 405     def on_cancel_button_clicked(self, widget):
 406         self.window.destroy()