libervia-backend: sat_frontends/jp/cmd

annotate sat_frontends/jp/cmd_file.py @ 2787:298408833ec2

memory (sqlite): optimizations indexes were missing on foreign keys for "message", "subject" and "thread" tables, seriously impacting performances. In addition to those indexes, two indexes have been added to speed ordering by timestamp on "history", and one on "files" table. history.rowid is not used anymore as there is an index on (profile_id, received_timestamp) which will speed up the query. Primary keys order has been changed to use automatic index in most common cases (filtering by profile_id then namespace).

author	Goffi <goffi@goffi.org>
date	Sat, 19 Jan 2019 22:49:32 +0100
parents	003b8b4b56a7
children	0b9faea5cb58

rev	line source
1960 3e168cde7a7d jp: fixed shebang python call Goffi <goffi@goffi.org> parents: 1880 diff changeset	1 #!/usr/bin/env python2
815 f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	2 # -- coding: utf-8 --
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	3
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	4 # jp: a SAT command line tool
2771 003b8b4b56a7 date update Goffi <goffi@goffi.org> parents: 2624 diff changeset	5 # Copyright (C) 2009-2019 Jérôme Poisson (goffi@goffi.org)
815 f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	6
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	7 # This program is free software: you can redistribute it and/or modify
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	8 # it under the terms of the GNU Affero General Public License as published by
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	9 # the Free Software Foundation, either version 3 of the License, or
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	10 # (at your option) any later version.
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	11
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	12 # This program is distributed in the hope that it will be useful,
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	15 # GNU Affero General Public License for more details.
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	16
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	17 # You should have received a copy of the GNU Affero General Public License
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
f8d534ed1d1e jp: added missing license headers Goffi <goffi@goffi.org> parents: 814 diff changeset	19
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	20
814 59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	21 import base
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	22 import sys
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	23 import os
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	24 import os.path
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	25 import tarfile
771 bfabeedbf32e core: i18n refactoring: Goffi <goffi@goffi.org> parents: 721 diff changeset	26 from sat.core.i18n import _
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	27 from sat_frontends.jp.constants import Const as C
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	28 from sat_frontends.jp import common
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	29 from sat_frontends.tools import jid
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	30 from sat.tools.common.ansi import ANSI as A
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	31 import tempfile
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	32 import xml.etree.ElementTree as ET # FIXME: used temporarily to manage XMLUI
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	33 from functools import partial
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	34 import json
402 f03688bdb858 jp: use with statement to open fifo Goffi <goffi@goffi.org> parents: 401 diff changeset	35
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	36 __commands__ = ["File"]
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	37
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	38
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	39 class Send(base.CommandBase):
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	40 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	41 super(Send, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	42 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	43 "send",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	44 use_progress=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	45 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	46 help=_("send a file to a contact"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	47 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	48 self.need_loop = True
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	49
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	50 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	51 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	52 "files", type=str, nargs="+", metavar="file", help=_(u"a list of file")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	53 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	54 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	55 "jid", type=base.unicode_decoder, help=_(u"the destination jid")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	56 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	57 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	58 "-b", "--bz2", action="store_true", help=_(u"make a bzip2 tarball")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	59 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	60 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	61 "-d",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	62 "--path",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	63 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	64 help=(u"path to the directory where the file must be stored"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	65 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	66 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	67 "-N",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	68 "--namespace",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	69 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	70 help=(u"namespace of the file"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	71 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	72 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	73 "-n",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	74 "--name",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	75 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	76 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	77 help=(u"name to use (DEFAULT: use source file name)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	78 )
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	79
1864 96ba685162f6 jp: all commands now use the new start method and set need_loop in __init__ when needed Goffi <goffi@goffi.org> parents: 1824 diff changeset	80 def start(self):
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	81 """Send files to jabber contact"""
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	82 self.send_files()
401 b2caa2615c4c jp roster name manegement + Pipe transfer Goffi <goffi@goffi.org> parents: 393 diff changeset	83
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	84 def onProgressStarted(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	85 self.disp(_(u"File copy started"), 2)
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	86
5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	87 def onProgressFinished(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	88 self.disp(_(u"File sent successfully"), 2)
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	89
5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	90 def onProgressError(self, error_msg):
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	91 if error_msg == C.PROGRESS_ERROR_DECLINED:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	92 self.disp(_(u"The file has been refused by your contact"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	93 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	94 self.disp(_(u"Error while sending file: {}").format(error_msg), error=True)
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	95
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	96 def gotId(self, data, file_):
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	97 """Called when a progress id has been received
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	98
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	99 @param pid(unicode): progress id
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	100 @param file_(str): file path
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	101 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	102 # FIXME: this show progress only for last progress_id
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	103 self.disp(_(u"File request sent to {jid}".format(jid=self.full_dest_jid)), 1)
1621 a17a91531fbe jp (file): print a message and quit if progress_id is not received Goffi <goffi@goffi.org> parents: 1606 diff changeset	104 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	105 self.progress_id = data["progress"]
1621 a17a91531fbe jp (file): print a message and quit if progress_id is not received Goffi <goffi@goffi.org> parents: 1606 diff changeset	106 except KeyError:
a17a91531fbe jp (file): print a message and quit if progress_id is not received Goffi <goffi@goffi.org> parents: 1606 diff changeset	107 # TODO: if 'xmlui' key is present, manage xmlui message display
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	108 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	109 _(u"Can't send file to {jid}".format(jid=self.full_dest_jid)), error=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	110 )
1621 a17a91531fbe jp (file): print a message and quit if progress_id is not received Goffi <goffi@goffi.org> parents: 1606 diff changeset	111 self.host.quit(2)
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	112
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	113 def error(self, failure):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	114 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	115 _("Error while trying to send a file: {reason}").format(reason=failure),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	116 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	117 )
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	118 self.host.quit(1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	119
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	120 def send_files(self):
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	121 for file_ in self.args.files:
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	122 if not os.path.exists(file_):
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	123 self.disp(_(u"file [{}] doesn't exist !").format(file_), error=True)
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	124 self.host.quit(1)
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	125 if not self.args.bz2 and os.path.isdir(file_):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	126 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	127 _(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	128 u"[{}] is a dir ! Please send files inside or use compression"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	129 ).format(file_)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	130 )
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	131 self.host.quit(1)
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	132
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	133 self.full_dest_jid = self.host.get_full_jid(self.args.jid)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	134 extra = {}
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	135 if self.args.path:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	136 extra[u"path"] = self.args.path
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	137 if self.args.namespace:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	138 extra[u"namespace"] = self.args.namespace
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	139
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	140 if self.args.bz2:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	141 with tempfile.NamedTemporaryFile("wb", delete=False) as buf:
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	142 self.host.addOnQuitCallback(os.unlink, buf.name)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	143 self.disp(_(u"bz2 is an experimental option, use with caution"))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	144 # FIXME: check free space
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	145 self.disp(_(u"Starting compression, please wait..."))
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	146 sys.stdout.flush()
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	147 bz2 = tarfile.open(mode="w:bz2", fileobj=buf)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	148 archive_name = u"{}.tar.bz2".format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	149 os.path.basename(self.args.files[0]) or u"compressed_files"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	150 )
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	151 for file_ in self.args.files:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	152 self.disp(_(u"Adding {}").format(file_), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	153 bz2.add(file_)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	154 bz2.close()
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	155 self.disp(_(u"Done !"), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	156
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	157 self.host.bridge.fileSend(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	158 self.full_dest_jid,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	159 buf.name,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	160 self.args.name or archive_name,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	161 "",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	162 extra,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	163 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	164 callback=lambda pid, file_=buf.name: self.gotId(pid, file_),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	165 errback=self.error,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	166 )
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	167 else:
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	168 for file_ in self.args.files:
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	169 path = os.path.abspath(file_)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	170 self.host.bridge.fileSend(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	171 self.full_dest_jid,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	172 path,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	173 self.args.name,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	174 "",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	175 extra,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	176 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	177 callback=lambda pid, file_=file_: self.gotId(pid, file_),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	178 errback=self.error,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	179 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	180
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	181
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	182 class Request(base.CommandBase):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	183 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	184 super(Request, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	185 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	186 "request",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	187 use_progress=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	188 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	189 help=_("request a file from a contact"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	190 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	191 self.need_loop = True
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	192
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	193 @property
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	194 def filename(self):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	195 return self.args.name or self.args.hash or u"output"
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	196
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	197 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	198 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	199 "jid", type=base.unicode_decoder, help=_(u"the destination jid")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	200 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	201 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	202 "-D",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	203 "--dest",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	204 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	205 help=_(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	206 u"destination path where the file will be saved (default: [current_dir]/[name\|hash])"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	207 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	208 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	209 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	210 "-n",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	211 "--name",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	212 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	213 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	214 help=_(u"name of the file"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	215 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	216 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	217 "-H",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	218 "--hash",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	219 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	220 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	221 help=_(u"hash of the file"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	222 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	223 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	224 "-a",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	225 "--hash-algo",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	226 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	227 default=u"sha-256",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	228 help=_(u"hash algorithm use for --hash (default: sha-256)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	229 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	230 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	231 "-d",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	232 "--path",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	233 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	234 help=(u"path to the directory containing the file"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	235 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	236 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	237 "-N",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	238 "--namespace",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	239 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	240 help=(u"namespace of the file"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	241 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	242 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	243 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	244 "--force",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	245 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	246 help=_(u"overwrite existing file without confirmation"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	247 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	248
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	249 def onProgressStarted(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	250 self.disp(_(u"File copy started"), 2)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	251
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	252 def onProgressFinished(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	253 self.disp(_(u"File received successfully"), 2)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	254
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	255 def onProgressError(self, error_msg):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	256 if error_msg == C.PROGRESS_ERROR_DECLINED:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	257 self.disp(_(u"The file request has been refused"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	258 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	259 self.disp(_(u"Error while requesting file: {}").format(error_msg), error=True)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	260
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	261 def gotId(self, progress_id):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	262 """Called when a progress id has been received
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	263
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	264 @param progress_id(unicode): progress id
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	265 """
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	266 self.progress_id = progress_id
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	267
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	268 def error(self, failure):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	269 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	270 _("Error while trying to send a file: {reason}").format(reason=failure),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	271 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	272 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	273 self.host.quit(1)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	274
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	275 def start(self):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	276 if not self.args.name and not self.args.hash:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	277 self.parser.error(_(u"at least one of --name or --hash must be provided"))
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	278 # extra = dict(self.args.extra)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	279 if self.args.dest:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	280 path = os.path.abspath(os.path.expanduser(self.args.dest))
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	281 if os.path.isdir(path):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	282 path = os.path.join(path, self.filename)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	283 else:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	284 path = os.path.abspath(self.filename)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	285
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	286 if os.path.exists(path) and not self.args.force:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	287 message = _(u"File {path} already exists! Do you want to overwrite?").format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	288 path=path
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	289 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	290 confirm = raw_input(u"{} (y/N) ".format(message).encode("utf-8"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	291 if confirm not in (u"y", u"Y"):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	292 self.disp(_(u"file request cancelled"))
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	293 self.host.quit(2)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	294
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	295 self.full_dest_jid = self.host.get_full_jid(self.args.jid)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	296 extra = {}
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	297 if self.args.path:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	298 extra[u"path"] = self.args.path
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	299 if self.args.namespace:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	300 extra[u"namespace"] = self.args.namespace
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	301 self.host.bridge.fileJingleRequest(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	302 self.full_dest_jid,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	303 path,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	304 self.args.name,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	305 self.args.hash,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	306 self.args.hash_algo if self.args.hash else u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	307 extra,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	308 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	309 callback=self.gotId,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	310 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	311 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	312 msg=_(u"can't request file: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	313 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	314 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	315 )
587 952322b1d490 Remove trailing whitespaces. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 572 diff changeset	316
952322b1d490 Remove trailing whitespaces. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 572 diff changeset	317
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	318 class Receive(base.CommandAnswering):
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	319 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	320 super(Receive, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	321 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	322 "receive",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	323 use_progress=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	324 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	325 help=_("wait for a file to be sent by a contact"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	326 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	327 self._overwrite_refused = False # True when one overwrite as already been refused
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	328 self.action_callbacks = {
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	329 C.META_TYPE_FILE: self.onFileAction,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	330 C.META_TYPE_OVERWRITE: self.onOverwriteAction,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	331 }
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	332
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	333 def onProgressStarted(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	334 self.disp(_(u"File copy started"), 2)
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	335
5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	336 def onProgressFinished(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	337 self.disp(_(u"File received successfully"), 2)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	338 if metadata.get("hash_verified", False):
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	339 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	340 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	341 _(u"hash checked: {algo}:{checksum}").format(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	342 algo=metadata["hash_algo"], checksum=metadata["hash"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	343 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	344 1,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	345 )
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	346 except KeyError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	347 self.disp(_(u"hash is checked but hash value is missing", 1), error=True)
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	348 else:
5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	349 self.disp(_(u"hash can't be verified"), 1)
5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	350
5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	351 def onProgressError(self, error_msg):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	352 self.disp(_(u"Error while receiving file: {}").format(error_msg), error=True)
1627 5a641e7b858a jp (base, file): use of new progress API. Progress callbacks are managed through CommandBase.onProgress* method instead of host attributes. Goffi <goffi@goffi.org> parents: 1621 diff changeset	353
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	354 def getXmluiId(self, action_data):
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	355 # FIXME: we temporarily use ElementTree, but a real XMLUI managing module
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	356 # should be available in the futur
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	357 # TODO: XMLUI module
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	358 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	359 xml_ui = action_data["xmlui"]
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	360 except KeyError:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	361 self.disp(_(u"Action has no XMLUI"), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	362 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	363 ui = ET.fromstring(xml_ui.encode("utf-8"))
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	364 xmlui_id = ui.get("submit")
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	365 if not xmlui_id:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	366 self.disp(_(u"Invalid XMLUI received"), error=True)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	367 return xmlui_id
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	368
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	369 def onFileAction(self, action_data, action_id, security_limit, profile):
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	370 xmlui_id = self.getXmluiId(action_data)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	371 if xmlui_id is None:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	372 return self.host.quitFromSignal(1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	373 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	374 from_jid = jid.JID(action_data["meta_from_jid"])
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	375 except KeyError:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	376 self.disp(_(u"Ignoring action without from_jid data"), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	377 return
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	378 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	379 progress_id = action_data["meta_progress_id"]
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	380 except KeyError:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	381 self.disp(_(u"ignoring action without progress id"), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	382 return
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	383
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	384 if not self.bare_jids or from_jid.bare in self.bare_jids:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	385 if self._overwrite_refused:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	386 self.disp(_(u"File refused because overwrite is needed"), error=True)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	387 self.host.bridge.launchAction(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	388 xmlui_id, {"cancelled": C.BOOL_TRUE}, profile_key=profile
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	389 )
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	390 return self.host.quitFromSignal(2)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	391 self.progress_id = progress_id
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	392 xmlui_data = {"path": self.path}
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	393 self.host.bridge.launchAction(xmlui_id, xmlui_data, profile_key=profile)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	394
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	395 def onOverwriteAction(self, action_data, action_id, security_limit, profile):
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	396 xmlui_id = self.getXmluiId(action_data)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	397 if xmlui_id is None:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	398 return self.host.quitFromSignal(1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	399 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	400 progress_id = action_data["meta_progress_id"]
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	401 except KeyError:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	402 self.disp(_(u"ignoring action without progress id"), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	403 return
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	404 self.disp(_(u"Overwriting needed"), 1)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	405
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	406 if progress_id == self.progress_id:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	407 if self.args.force:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	408 self.disp(_(u"Overwrite accepted"), 2)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	409 else:
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	410 self.disp(_(u"Refused to overwrite"), 2)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	411 self._overwrite_refused = True
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	412
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	413 xmlui_data = {"answer": C.boolConst(self.args.force)}
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	414 self.host.bridge.launchAction(xmlui_id, xmlui_data, profile_key=profile)
393 393b35aa86d2 jp: added --connect option Goffi <goffi@goffi.org> parents: 391 diff changeset	415
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	416 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	417 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	418 "jids",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	419 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	420 nargs="*",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	421 help=_(u"jids accepted (accept everything if none is specified)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	422 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	423 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	424 "-m",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	425 "--multiple",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	426 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	427 help=_(u"accept multiple files (you'll have to stop manually)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	428 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	429 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	430 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	431 "--force",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	432 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	433 help=_(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	434 u"force overwritting of existing files (/!\\ name is choosed by sender)"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	435 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	436 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	437 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	438 "--path",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	439 default=".",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	440 metavar="DIR",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	441 help=_(u"destination path (default: working directory)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	442 )
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	443
1864 96ba685162f6 jp: all commands now use the new start method and set need_loop in __init__ when needed Goffi <goffi@goffi.org> parents: 1824 diff changeset	444 def start(self):
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	445 self.bare_jids = [jid.JID(jid_).bare for jid_ in self.args.jids]
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	446 self.path = os.path.abspath(self.args.path)
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	447 if not os.path.isdir(self.path):
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	448 self.disp(_(u"Given path is not a directory !", error=True))
1880 68944e270c40 jp (file): fixed bad quit call Goffi <goffi@goffi.org> parents: 1864 diff changeset	449 self.host.quit(2)
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	450 if self.args.multiple:
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	451 self.host.quit_on_progress_end = False
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	452 self.disp(_(u"waiting for incoming file request"), 2)
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	453
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	454
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	455 class Upload(base.CommandBase):
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	456 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	457 super(Upload, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	458 host, "upload", use_progress=True, use_verbose=True, help=_("upload a file")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	459 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	460 self.need_loop = True
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	461
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	462 def add_parser_options(self):
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	463 self.parser.add_argument("file", type=str, help=_("file to upload"))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	464 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	465 "jid",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	466 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	467 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	468 help=_("jid of upload component (nothing to autodetect)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	469 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	470 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	471 "--ignore-tls-errors",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	472 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	473 help=_("ignore invalide TLS certificate"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	474 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	475
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	476 def onProgressStarted(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	477 self.disp(_(u"File upload started"), 2)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	478
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	479 def onProgressFinished(self, metadata):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	480 self.disp(_(u"File uploaded successfully"), 2)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	481 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	482 url = metadata["url"]
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	483 except KeyError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	484 self.disp(u"download URL not found in metadata")
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	485 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	486 self.disp(_(u"URL to retrieve the file:"), 1)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	487 # XXX: url is display alone on a line to make parsing easier
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	488 self.disp(url)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	489
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	490 def onProgressError(self, error_msg):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	491 self.disp(_(u"Error while uploading file: {}").format(error_msg), error=True)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	492
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	493 def gotId(self, data, file_):
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	494 """Called when a progress id has been received
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	495
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	496 @param pid(unicode): progress id
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	497 @param file_(str): file path
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	498 """
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	499 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	500 self.progress_id = data["progress"]
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	501 except KeyError:
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	502 # TODO: if 'xmlui' key is present, manage xmlui message display
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	503 self.disp(_(u"Can't upload file"), error=True)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	504 self.host.quit(2)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	505
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	506 def error(self, failure):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	507 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	508 _("Error while trying to upload a file: {reason}").format(reason=failure),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	509 error=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	510 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	511 self.host.quit(1)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	512
1864 96ba685162f6 jp: all commands now use the new start method and set need_loop in __init__ when needed Goffi <goffi@goffi.org> parents: 1824 diff changeset	513 def start(self):
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	514 file_ = self.args.file
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	515 if not os.path.exists(file_):
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	516 self.disp(_(u"file [{}] doesn't exist !").format(file_), error=True)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	517 self.host.quit(1)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	518 if os.path.isdir(file_):
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	519 self.disp(_(u"[{}] is a dir! Can't upload a dir").format(file_))
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	520 self.host.quit(1)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	521
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	522 self.full_dest_jid = (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	523 self.host.get_full_jid(self.args.jid) if self.args.jid is not None else ""
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	524 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	525 options = {}
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	526 if self.args.ignore_tls_errors:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	527 options["ignore_tls_errors"] = C.BOOL_TRUE
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	528
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	529 path = os.path.abspath(file_)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	530 self.host.bridge.fileUpload(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	531 path,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	532 "",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	533 self.full_dest_jid,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	534 options,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	535 self.profile,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	536 callback=lambda pid, file_=file_: self.gotId(pid, file_),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	537 errback=self.error,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	538 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	539
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	540
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	541 class ShareList(base.CommandBase):
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	542 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	543 extra_outputs = {"default": self.default_output}
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	544 super(ShareList, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	545 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	546 "list",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	547 use_output=C.OUTPUT_LIST_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	548 extra_outputs=extra_outputs,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	549 help=_(u"retrieve files shared by an entity"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	550 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	551 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	552 self.need_loop = True
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	553
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	554 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	555 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	556 "-d",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	557 "--path",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	558 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	559 help=_(u"path to the directory containing the files"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	560 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	561 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	562 "jid",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	563 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	564 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	565 default="",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	566 help=_("jid of sharing entity (nothing to check our own jid)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	567 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	568
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	569 def file_gen(self, files_data):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	570 for file_data in files_data:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	571 yield file_data[u"name"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	572 yield file_data.get(u"size", "")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	573 yield file_data.get(u"hash", "")
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	574
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	575 def _name_filter(self, name, row):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	576 if row.type == C.FILE_TYPE_DIRECTORY:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	577 return A.color(C.A_DIRECTORY, name)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	578 elif row.type == C.FILE_TYPE_FILE:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	579 return A.color(C.A_FILE, name)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	580 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	581 self.disp(_(u"unknown file type: {type}").format(type=row.type), error=True)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	582 return name
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	583
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	584 def _size_filter(self, size, row):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	585 if not size:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	586 return u""
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	587 size = int(size)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	588 # cf. https://stackoverflow.com/a/1094933 (thanks)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	589 suffix = u"o"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	590 for unit in [u"", u"Ki", u"Mi", u"Gi", u"Ti", u"Pi", u"Ei", u"Zi"]:
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	591 if abs(size) < 1024.0:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	592 return A.color(A.BOLD, u"{:.2f}".format(size), unit, suffix)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	593 size /= 1024.0
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	594
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	595 return A.color(A.BOLD, u"{:.2f}".format(size), u"Yi", suffix)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	596
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	597 def default_output(self, files_data):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	598 """display files a way similar to ls"""
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	599 files_data.sort(key=lambda d: d["name"].lower())
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	600 show_header = False
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	601 if self.verbosity == 0:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	602 headers = (u"name", u"type")
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	603 elif self.verbosity == 1:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	604 headers = (u"name", u"type", u"size")
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	605 elif self.verbosity > 1:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	606 show_header = True
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	607 headers = (u"name", u"type", u"size", u"hash")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	608 table = common.Table.fromDict(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	609 self.host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	610 files_data,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	611 headers,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	612 filters={u"name": self._name_filter, u"size": self._size_filter},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	613 defaults={u"size": u"", u"hash": u""},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	614 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	615 table.display_blank(show_header=show_header, hide_cols=["type"])
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	616
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	617 def _FISListCb(self, files_data):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	618 self.output(files_data)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	619 self.host.quit()
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	620
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	621 def start(self):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	622 self.host.bridge.FISList(
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	623 self.args.jid,
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	624 self.args.path,
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	625 {},
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	626 self.profile,
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	627 callback=self._FISListCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	628 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	629 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	630 msg=_(u"can't retrieve shared files: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	631 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	632 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	633 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	634
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	635
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	636 class SharePath(base.CommandBase):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	637 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	638 super(SharePath, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	639 host, "path", help=_(u"share a file or directory"), use_verbose=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	640 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	641 self.need_loop = True
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	642
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	643 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	644 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	645 "-n",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	646 "--name",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	647 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	648 default=u"",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	649 help=_(u"virtual name to use (default: use directory/file name)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	650 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	651 perm_group = self.parser.add_mutually_exclusive_group()
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	652 perm_group.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	653 "-j",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	654 "--jid",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	655 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	656 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	657 dest="jids",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	658 default=[],
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	659 help=_(u"jid of contacts allowed to retrieve the files"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	660 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	661 perm_group.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	662 "--public",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	663 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	664 help=_(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	665 u"share publicly the file(s) (/!\\ everybody will be able to access them)"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	666 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	667 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	668 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	669 "path",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	670 type=base.unicode_decoder,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	671 help=_(u"path to a file or directory to share"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	672 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	673
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	674 def _FISSharePathCb(self, name):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	675 self.disp(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	676 _(u'{path} shared under the name "{name}"').format(path=self.path, name=name)
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	677 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	678 self.host.quit()
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	679
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	680 def start(self):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	681 self.path = os.path.abspath(self.args.path)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	682 if self.args.public:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	683 access = {u"read": {u"type": u"public"}}
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	684 else:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	685 jids = self.args.jids
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	686 if jids:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	687 access = {u"read": {u"type": "whitelist", u"jids": jids}}
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	688 else:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	689 access = {}
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	690 self.host.bridge.FISSharePath(
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	691 self.args.name,
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	692 self.path,
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	693 json.dumps(access, ensure_ascii=False),
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	694 self.profile,
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	695 callback=self._FISSharePathCb,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	696 errback=partial(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	697 self.errback,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	698 msg=_(u"can't share path: {}"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	699 exit_code=C.EXIT_BRIDGE_ERRBACK,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	700 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	701 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	702
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	703
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	704 class Share(base.CommandBase):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	705 subcommands = (ShareList, SharePath)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	706
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	707 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	708 super(Share, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	709 host, "share", use_profile=False, help=_(u"files sharing management")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	710 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	711
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	712
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	713 class File(base.CommandBase):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	714 subcommands = (Send, Request, Receive, Upload, Share)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	715
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	716 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	717 super(File, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	718 host, "file", use_profile=False, help=_(u"files sending/receiving/management")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	719 )

Mercurial > libervia-backend

annotate sat_frontends/jp/cmd_file.py @ 2787:298408833ec2