# HG changeset patch # User Goffi # Date 1490809364 -7200 # Node ID 5831e8fcc5f822da931db64304551127c3891dc3 # Parent 4e06cd44e6672c88779a4073a7148428c231d141 jp (invitation): invitation command first draft diff -r 4e06cd44e667 -r 5831e8fcc5f8 frontends/src/jp/cmd_invitation.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/frontends/src/jp/cmd_invitation.py Wed Mar 29 19:42:44 2017 +0200 @@ -0,0 +1,137 @@ +#!/usr/bin/env python2 +# -*- coding: utf-8 -*- + +# jp: a SàT 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 +from sat.core.i18n import _ +from sat_frontends.jp.constants import Const as C +from functools import partial + +__commands__ = ["Invitation"] + + +class Create(base.CommandBase): + + def __init__(self, host): + base.CommandBase.__init__(self, host, 'create', use_profile=False, use_output=C.OUTPUT_DICT, help=_(u'create and send an invitation')) + self.need_loop=True + + def add_parser_options(self): + self.parser.add_argument("-j", "--jid", type=base.unicode_decoder, default='', help='jid of the invitee (default: generate one)') + self.parser.add_argument("-P", "--password", type=base.unicode_decoder, default='', help='password of the invitee profile/XMPP account (default: generate one)') + self.parser.add_argument("-n", "--name", type=base.unicode_decoder, default='', help='name of the invitee') + self.parser.add_argument("-N", "--host-name", type=base.unicode_decoder, default='', help='name of the host') + self.parser.add_argument("-e", "--email", type=base.unicode_decoder, default='', help='email to send the invitation to (if --no-email is set, email will just be saved)') + self.parser.add_argument("--no-email", action="store_true", help='do NOT send invitation email') + self.parser.add_argument("-l", "--lang", type=base.unicode_decoder, default='', help='main language spoken by the invitee') + self.parser.add_argument("-u", "--url", type=base.unicode_decoder, default='', help='template to construt the URL') + self.parser.add_argument("-s", "--subject", type=base.unicode_decoder, default='', help='subject of the invitation email (default: generic subject)') + self.parser.add_argument("-b", "--body", type=base.unicode_decoder, default='', help='body of the invitation email (default: generic body)') + self.parser.add_argument("-x", "--extra", metavar=('KEY', 'VALUE'), type=base.unicode_decoder, action='append', nargs=2, default=[], help='extra data to associate with invitation/invitee') + self.parser.add_argument("-p", "--profile", type=base.unicode_decoder, default='', help="profile doing the invitation (default: don't associate profile") + + def invitationCreateCb(self, invitation_data): + self.output(invitation_data) + self.host.quit(C.EXIT_OK) + + def invitationCreateEb(self, failure_): + self.disp(u"can't create invitation: {reason}".format( + reason=failure_), error=True) + self.host.quit(C.EXIT_BRIDGE_ERRBACK) + + def start(self): + extra = dict(self.args.extra) + email = self.args.email + if self.args.no_email: + if email: + extra['email'] = email + else: + if not email: + self.parser.error(_(u'you need to specify an email address to send email invitation')) + + self.host.bridge.invitationCreate( + self.args.jid, + self.args.password, + self.args.name, + self.args.host_name, + email, + self.args.lang, + self.args.url, + self.args.subject, + self.args.body, + extra, + self.args.profile, + callback=self.invitationCreateCb, + errback=self.invitationCreateEb) + + +class Get(base.CommandBase): + + def __init__(self, host): + base.CommandBase.__init__(self, host, 'get', use_profile=False, use_output=C.OUTPUT_DICT, help=_(u'get invitation data')) + self.need_loop=True + + def add_parser_options(self): + self.parser.add_argument("id", type=base.unicode_decoder, + help=_(u"invitation UUID")) + self.parser.add_argument("-j", "--with-jid", action="store_true", help=_(u"start profile session and retrieve jid")) + + def output_data(self, data, jid_=None): + if jid_ is not None: + data['jid'] = jid_ + self.output(data) + self.host.quit() + + def invitationGetCb(self, invitation_data): + if self.args.with_jid: + profile = invitation_data[u'guest_profile'] + def session_started(dummy): + self.host.bridge.asyncGetParamA( + u'JabberID', + u'Connection', + profile_key=profile, + callback=lambda jid_: self.output_data(invitation_data, jid_), + errback=partial(self.errback, + msg=_(u"can't retrieve jid: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK)) + + self.host.bridge.profileStartSession( + invitation_data[u'password'], + profile, + callback=session_started, + errback=partial(self.errback, + msg=_(u"can't start session: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK)) + else: + self.output_data(invitation_data) + + def start(self): + self.host.bridge.invitationGet( + self.args.id, + callback=self.invitationGetCb, + errback=partial(self.errback, + msg=_(u"can't get invitation data: {}"), + exit_code=C.EXIT_BRIDGE_ERRBACK)) + + +class Invitation(base.CommandBase): + subcommands = (Create, Get) + + def __init__(self, host): + super(Invitation, self).__init__(host, 'invitation', use_profile=False, help=_(u'invitation of user(s) without XMPP account'))