# HG changeset patch # User Goffi # Date 1473797987 -7200 # Node ID db5cda61740f1cfb9f2b574060d306801e940b2c # Parent 3d633458d96279b63153b16a0e3d5fab0729a5c5 jp (avatar): avatar get/set implementations, first draft diff -r 3d633458d962 -r db5cda61740f frontends/src/jp/cmd_avatar.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/frontends/src/jp/cmd_avatar.py Tue Sep 13 22:19:47 2016 +0200 @@ -0,0 +1,111 @@ +#!/usr/bin/env python2 +# -*- coding: utf-8 -*- + +# jp: a SAT command line tool +# Copyright (C) 2009-2016 Jérôme Poisson (goffi@goffi.org) + +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 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 Affero General Public License for more details. + +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . + + +import base +import os +import os.path +from sat.core.i18n import _ +from sat_frontends.jp.constants import Const as C +from sat.tools import config +import subprocess + + +__commands__ = ["Avatar"] +DISPLAY_CMD = ['xv', 'display', 'gwenview', 'showtell'] + + +class Set(base.CommandBase): + def __init__(self, host): + super(Set, self).__init__(host, 'set', use_verbose=True, help=_('set avatar of the profile')) + self.need_loop=True + + def add_parser_options(self): + self.parser.add_argument("image_path", type=str, help=_("path to the image to upload")) + + def start(self): + """Send files to jabber contact""" + path = self.args.image_path + if not os.path.exists(path): + self.disp(_(u"file [{}] doesn't exist !").format(path), error=True) + self.host.quit(1) + path = os.path.abspath(path) + self.host.bridge.setAvatar(path, self.profile, callback=self._avatarCb, errback=self._avatarEb) + + def _avatarCb(self): + self.disp(_("avatar has been set"), 1) + self.host.quit() + + def _avatarEb(self, failure_): + self.disp(_("error while uploading avatar: {msg}").format(msg=failure_), error=True) + self.host.quit(C.EXIT_ERROR) + + +class Get(base.CommandBase): + + def __init__(self, host): + super(Get, self).__init__(host, 'get', use_verbose=True, help=_('retrieve avatar of an entity')) + self.need_loop=True + + def add_parser_options(self): + self.parser.add_argument("jid", type=base.unicode_decoder, help=_("entity")) + self.parser.add_argument("-s", "--show", action="store_true", help=_(u"show avatar")) + + def showImage(self, path): + sat_conf = config.parseMainConf() + cmd = config.getConfig(sat_conf, 'jp', 'image_cmd') + cmds = [cmd] + DISPLAY_CMD if cmd else DISPLAY_CMD + for cmd in cmds: + try: + ret = subprocess.call([cmd] + [path]) + except OSError: + pass + else: + if ret == 0: + break + else: + # didn't worked with commands, we try our luck with webbrowser + # in some cases, webbrowser can actually open the associated display program + import webbrowser + webbrowser.open(path) + + def _getAvatarCb(self, avatar_path): + if not avatar_path: + self.disp(_(u"No avatar found."), 1) + self.host.quit(C.EXIT_NOT_FOUND) + + self.disp(avatar_path) + if self.args.show: + self.showImage(avatar_path) + + self.host.quit() + + def _getAvatarEb(self, failure_): + self.disp(_("error while getting avatar: {msg}").format(msg=failure_), error=True) + self.host.quit(C.EXIT_ERROR) + + def start(self): + self.host.bridge.getAvatar(self.args.jid, self.profile, callback=self._getAvatarCb, errback=self._getAvatarEb) + + +class Avatar(base.CommandBase): + subcommands = (Set, Get) + + def __init__(self, host): + super(Avatar, self).__init__(host, 'avatar', use_profile=False, help=_('avatar uploading/retrieving'))