libervia-backend: sat_frontends/jp/cmd

annotate sat_frontends/jp/cmd_file.py @ 3410:55f9a38864af

jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: when a file is proposed by an entity not in roster, the dialog asking for confirmation is displayed. If the bare jid of the sender is explicitely waited (i.e. it's in arguments), then the dialog is automatically confirmed.

author	Goffi <goffi@goffi.org>
date	Thu, 12 Nov 2020 14:53:16 +0100
parents	19bc03743aeb
children	f30b238d9c45

rev	line source
3137 559a625a236b fixed shebangs Goffi <goffi@goffi.org> parents: 3136 diff changeset	1 #!/usr/bin/env python3
559a625a236b fixed shebangs Goffi <goffi@goffi.org> parents: 3136 diff changeset	2
815 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
3136 9d0df638c8b4 dates update Goffi <goffi@goffi.org> parents: 3121 diff changeset	5 # Copyright (C) 2009-2020 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
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	21 from . import base
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	22 from . import xmlui_manager
814 59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	23 import sys
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	24 import os
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	25 import os.path
59c7bc51c323 jp: refactoring using ArgParse Dal <kedals0@gmail.com> parents: 811 diff changeset	26 import tarfile
771 bfabeedbf32e core: i18n refactoring: Goffi <goffi@goffi.org> parents: 721 diff changeset	27 from sat.core.i18n import _
2932 472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	28 from sat.tools.common import data_format
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.jp.constants import Const as C
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	30 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	31 from sat_frontends.tools import jid
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	32 from sat.tools.common.ansi import ANSI as A
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	33 from urllib.parse import urlparse
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	34 from pathlib import Path
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	35 import tempfile
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	36 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	37 import json
402 f03688bdb858 jp: use with statement to open fifo Goffi <goffi@goffi.org> parents: 401 diff changeset	38
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	39 __commands__ = ["File"]
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	40
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	41
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	42 class Send(base.CommandBase):
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	43 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	44 super(Send, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	45 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	46 "send",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	47 use_progress=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	48 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	49 help=_("send a file to a contact"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	50 )
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	51
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	52 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	53 self.parser.add_argument(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	54 "files", type=str, nargs="+", metavar="file", help=_("a list of file")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	55 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	56 self.parser.add_argument(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	57 "jid", help=_("the destination jid")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	58 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	59 self.parser.add_argument(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	60 "-b", "--bz2", action="store_true", help=_("make a bzip2 tarball")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	61 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	62 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	63 "-d",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	64 "--path",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	65 help=("path to the directory where the file must be stored"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	66 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	67 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	68 "-N",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	69 "--namespace",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	70 help=("namespace of the file"),
2624 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",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	75 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	76 help=("name to use (DEFAULT: use source file name)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	77 )
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	78
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	79 async def onProgressStarted(self, metadata):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	80 self.disp(_("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	81
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	82 async def onProgressFinished(self, metadata):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	83 self.disp(_("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	84
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	85 async def onProgressError(self, error_msg):
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	86 if error_msg == C.PROGRESS_ERROR_DECLINED:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	87 self.disp(_("The file has been refused by your contact"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	88 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	89 self.disp(_("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	90
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	91 async def gotId(self, data, file_):
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	92 """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	93
de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	94 @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	95 @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	96 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	97 # FIXME: this show progress only for last progress_id
3408 19bc03743aeb jp (file/send): don't add main resource to bare jid anymore: Goffi <goffi@goffi.org> parents: 3325 diff changeset	98 self.disp(_("File request sent to {jid}".format(jid=self.args.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	99 try:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	100 await self.set_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	101 except KeyError:
a17a91531fbe jp (file): print a message and quit if progress_id is not received Goffi <goffi@goffi.org> parents: 1606 diff changeset	102 # 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	103 self.disp(
3408 19bc03743aeb jp (file/send): don't add main resource to bare jid anymore: Goffi <goffi@goffi.org> parents: 3325 diff changeset	104 _("Can't send file to {jid}".format(jid=self.args.jid)), error=True
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	105 )
1621 a17a91531fbe jp (file): print a message and quit if progress_id is not received Goffi <goffi@goffi.org> parents: 1606 diff changeset	106 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	107
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	108 async def start(self):
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	109 for file_ in self.args.files:
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	110 if not os.path.exists(file_):
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	111 self.disp(_(f"file {file_!r} doesn't exist!"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	112 self.host.quit(C.EXIT_BAD_ARG)
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	113 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	114 self.disp(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	115 _(f"{file_!r} is a dir! Please send files inside or use compression")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	116 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	117 self.host.quit(C.EXIT_BAD_ARG)
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	118
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	119 extra = {}
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	120 if self.args.path:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	121 extra["path"] = self.args.path
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	122 if self.args.namespace:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	123 extra["namespace"] = self.args.namespace
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	124
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	125 if self.args.bz2:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	126 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	127 self.host.addOnQuitCallback(os.unlink, buf.name)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	128 self.disp(_("bz2 is an experimental option, use with caution"))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	129 # FIXME: check free space
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	130 self.disp(_("Starting compression, please wait..."))
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	131 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	132 bz2 = tarfile.open(mode="w:bz2", fileobj=buf)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	133 archive_name = "{}.tar.bz2".format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	134 os.path.basename(self.args.files[0]) or "compressed_files"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	135 )
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	136 for file_ in self.args.files:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	137 self.disp(_("Adding {}").format(file_), 1)
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	138 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	139 bz2.close()
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	140 self.disp(_("Done !"), 1)
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	141
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	142 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	143 send_data = await self.host.bridge.fileSend(
3408 19bc03743aeb jp (file/send): don't add main resource to bare jid anymore: Goffi <goffi@goffi.org> parents: 3325 diff changeset	144 self.args.jid,
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	145 buf.name,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	146 self.args.name or archive_name,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	147 "",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	148 extra,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	149 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	150 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	151 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	152 self.disp(f"can't send file: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	153 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	154 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	155 await self.gotId(send_data, file_)
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	156 else:
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	157 for file_ in self.args.files:
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	158 path = os.path.abspath(file_)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	159 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	160 send_data = await self.host.bridge.fileSend(
3408 19bc03743aeb jp (file/send): don't add main resource to bare jid anymore: Goffi <goffi@goffi.org> parents: 3325 diff changeset	161 self.args.jid,
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	162 path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	163 self.args.name,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	164 "",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	165 extra,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	166 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	167 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	168 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	169 self.disp(f"can't send file {file_!r}: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	170 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	171 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	172 await self.gotId(send_data, file_)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	173
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	174
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	175 class Request(base.CommandBase):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	176 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	177 super(Request, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	178 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	179 "request",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	180 use_progress=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	181 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	182 help=_("request a file from a contact"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	183 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	184
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	185 @property
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	186 def filename(self):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	187 return self.args.name or self.args.hash or "output"
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	188
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	189 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	190 self.parser.add_argument(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	191 "jid", help=_("the destination jid")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	192 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	193 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	194 "-D",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	195 "--dest",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	196 help=_(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	197 "destination path where the file will be saved (default: "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	198 "[current_dir]/[name\|hash])"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	199 ),
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 "-n",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	203 "--name",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	204 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	205 help=_("name of the file"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	206 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	207 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	208 "-H",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	209 "--hash",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	210 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	211 help=_("hash of the file"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	212 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	213 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	214 "-a",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	215 "--hash-algo",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	216 default="sha-256",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	217 help=_("hash algorithm use for --hash (default: sha-256)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	218 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	219 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	220 "-d",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	221 "--path",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	222 help=("path to the directory containing the file"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	223 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	224 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	225 "-N",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	226 "--namespace",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	227 help=("namespace of the file"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	228 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	229 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	230 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	231 "--force",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	232 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	233 help=_("overwrite existing file without confirmation"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	234 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	235
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	236 async def onProgressStarted(self, metadata):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	237 self.disp(_("File copy started"), 2)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	238
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	239 async def onProgressFinished(self, metadata):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	240 self.disp(_("File received successfully"), 2)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	241
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	242 async def onProgressError(self, error_msg):
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	243 if error_msg == C.PROGRESS_ERROR_DECLINED:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	244 self.disp(_("The file request has been refused"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	245 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	246 self.disp(_("Error while requesting file: {}").format(error_msg), error=True)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	247
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	248 async def start(self):
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	249 if not self.args.name and not self.args.hash:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	250 self.parser.error(_("at least one of --name or --hash must be provided"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	251 if self.args.dest:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	252 path = os.path.abspath(os.path.expanduser(self.args.dest))
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	253 if os.path.isdir(path):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	254 path = os.path.join(path, self.filename)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	255 else:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	256 path = os.path.abspath(self.filename)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	257
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	258 if os.path.exists(path) and not self.args.force:
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	259 message = _("File {path} already exists! Do you want to overwrite?").format(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	260 path = path)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	261 await self.host.confirmOrQuit(message, _("file request cancelled"))
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	262
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	263 self.full_dest_jid = await self.host.get_full_jid(self.args.jid)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	264 extra = {}
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	265 if self.args.path:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	266 extra["path"] = self.args.path
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	267 if self.args.namespace:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	268 extra["namespace"] = self.args.namespace
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	269 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	270 progress_id = await self.host.bridge.fileJingleRequest(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	271 self.full_dest_jid,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	272 path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	273 self.args.name,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	274 self.args.hash,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	275 self.args.hash_algo if self.args.hash else "",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	276 extra,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	277 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	278 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	279 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	280 self.disp(msg=_(f"can't request file: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	281 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	282 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	283 await self.set_progress_id(progress_id)
587 952322b1d490 Remove trailing whitespaces. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 572 diff changeset	284
952322b1d490 Remove trailing whitespaces. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 572 diff changeset	285
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	286 class Receive(base.CommandAnswering):
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	287 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	288 super(Receive, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	289 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	290 "receive",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	291 use_progress=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	292 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	293 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	294 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	295 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	296 self.action_callbacks = {
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	297 C.META_TYPE_FILE: self.onFileAction,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	298 C.META_TYPE_OVERWRITE: self.onOverwriteAction,
3410 55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	299 C.META_TYPE_NOT_IN_ROSTER_LEAK: self.onNotInRosterAction,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	300 }
1606 de785fcf9a7b jp (base, file): file command and progress fixes and adaptation to new API: Goffi <goffi@goffi.org> parents: 1396 diff changeset	301
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	302 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	303 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	304 "jids",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	305 nargs="*",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	306 help=_("jids accepted (accept everything if none is specified)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	307 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	308 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	309 "-m",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	310 "--multiple",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	311 action="store_true",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	312 help=_("accept multiple files (you'll have to stop manually)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	313 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	314 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	315 "-f",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	316 "--force",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	317 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	318 help=_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	319 "force overwritting of existing files (/!\\ name is choosed by sender)"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	320 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	321 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	322 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	323 "--path",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	324 default=".",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	325 metavar="DIR",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	326 help=_("destination path (default: working directory)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	327 )
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	328
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	329 async def onProgressStarted(self, metadata):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	330 self.disp(_("File copy started"), 2)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	331
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	332 async def onProgressFinished(self, metadata):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	333 self.disp(_("File received successfully"), 2)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	334 if metadata.get("hash_verified", False):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	335 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	336 self.disp(_(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	337 f"hash checked: {metadata['hash_algo']}:{metadata['hash']}"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	338 except KeyError:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	339 self.disp(_("hash is checked but hash value is missing", 1), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	340 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	341 self.disp(_("hash can't be verified"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	342
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	343 async def onProgressError(self, e):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	344 self.disp(_(f"Error while receiving file: {e}"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	345
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	346 def getXmluiId(self, action_data):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	347 # FIXME: we temporarily use ElementTree, but a real XMLUI managing module
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	348 # should be available in the futur
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	349 # TODO: XMLUI module
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	350 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	351 xml_ui = action_data["xmlui"]
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	352 except KeyError:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	353 self.disp(_("Action has no XMLUI"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	354 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	355 ui = ET.fromstring(xml_ui.encode("utf-8"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	356 xmlui_id = ui.get("submit")
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	357 if not xmlui_id:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	358 self.disp(_("Invalid XMLUI received"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	359 return xmlui_id
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	360
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	361 async def onFileAction(self, action_data, action_id, security_limit, profile):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	362 xmlui_id = self.getXmluiId(action_data)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	363 if xmlui_id is None:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	364 return self.host.quitFromSignal(1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	365 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	366 from_jid = jid.JID(action_data["meta_from_jid"])
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	367 except KeyError:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	368 self.disp(_("Ignoring action without from_jid data"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	369 return
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	370 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	371 progress_id = action_data["meta_progress_id"]
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	372 except KeyError:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	373 self.disp(_("ignoring action without progress id"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	374 return
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	375
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	376 if not self.bare_jids or from_jid.bare in self.bare_jids:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	377 if self._overwrite_refused:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	378 self.disp(_("File refused because overwrite is needed"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	379 await self.host.bridge.launchAction(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	380 xmlui_id, {"cancelled": C.BOOL_TRUE}, profile_key=profile
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	381 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	382 return self.host.quitFromSignal(2)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	383 await self.set_progress_id(progress_id)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	384 xmlui_data = {"path": self.path}
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	385 await self.host.bridge.launchAction(xmlui_id, xmlui_data, profile_key=profile)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	386
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	387 async def onOverwriteAction(self, action_data, action_id, security_limit, profile):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	388 xmlui_id = self.getXmluiId(action_data)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	389 if xmlui_id is None:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	390 return self.host.quitFromSignal(1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	391 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	392 progress_id = action_data["meta_progress_id"]
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	393 except KeyError:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	394 self.disp(_("ignoring action without progress id"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	395 return
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	396 self.disp(_("Overwriting needed"), 1)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	397
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	398 if progress_id == self.progress_id:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	399 if self.args.force:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	400 self.disp(_("Overwrite accepted"), 2)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	401 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	402 self.disp(_("Refused to overwrite"), 2)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	403 self._overwrite_refused = True
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	404
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	405 xmlui_data = {"answer": C.boolConst(self.args.force)}
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	406 await self.host.bridge.launchAction(xmlui_id, xmlui_data, profile_key=profile)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	407
3410 55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	408 async def onNotInRosterAction(self, action_data, action_id, security_limit, profile):
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	409 xmlui_id = self.getXmluiId(action_data)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	410 if xmlui_id is None:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	411 return self.host.quitFromSignal(1)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	412 try:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	413 from_jid = jid.JID(action_data["meta_from_jid"])
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	414 except ValueError:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	415 self.disp(_('invalid "from_jid" value received, ignoring: {value}').format(
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	416 value=from_jid), error=True)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	417 return
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	418 except KeyError:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	419 self.disp(_('ignoring action without "from_jid" value'), error=True)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	420 return
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	421 self.disp(_("Confirmation needed for request from an entity not in roster"), 1)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	422
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	423 if from_jid.bare in self.bare_jids:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	424 # if the sender is expected, we can confirm the session
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	425 confirmed = True
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	426 self.disp(_("Sender confirmed because she or he is explicitly expected"), 1)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	427 else:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	428 xmlui = xmlui_manager.create(self.host, action_data["xmlui"])
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	429 confirmed = await self.host.confirm(xmlui.dlg.message)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	430
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	431 xmlui_data = {"answer": C.boolConst(confirmed)}
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	432 await self.host.bridge.launchAction(xmlui_id, xmlui_data, profile_key=profile)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	433 if not confirmed and not self.args.multiple:
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	434 self.disp(_("Session refused for {from_jid}").format(from_jid=from_jid))
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	435 self.host.quitFromSignal(0)
55f9a38864af jp (file/receive): display or answer to C.META_TYPE_NOT_IN_ROSTER_LEAK dialog: Goffi <goffi@goffi.org> parents: 3408 diff changeset	436
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	437 async 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	438 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	439 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	440 if not os.path.isdir(self.path):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	441 self.disp(_("Given path is not a directory !", error=True))
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	442 self.host.quit(C.EXIT_BAD_ARG)
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	443 if self.args.multiple:
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	444 self.host.quit_on_progress_end = False
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	445 self.disp(_("waiting for incoming file request"), 2)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	446 await self.start_answering()
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	447
c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	448
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	449 class Get(base.CommandBase):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	450
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	451 def __init__(self, host):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	452 super(Get, self).__init__(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	453 host, "get", use_progress=True, use_verbose=True,
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	454 help=_("download a file from URI")
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	455 )
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	456
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	457 def add_parser_options(self):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	458 self.parser.add_argument(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	459 '-o', '--dest_file', type=str, default='',
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	460 help=_("destination file (DEFAULT: filename from URL)")
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	461 )
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	462 self.parser.add_argument(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	463 "-f",
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	464 "--force",
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	465 action="store_true",
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	466 help=_("overwrite existing file without confirmation"),
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	467 )
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	468 self.parser.add_argument("uri", type=str, help=_("URI of the file to retrieve"))
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	469
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	470 async def onProgressStarted(self, metadata):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	471 self.disp(_("File download started"), 2)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	472
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	473 async def onProgressFinished(self, metadata):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	474 self.disp(_("File downloaded successfully"), 2)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	475
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	476 async def onProgressError(self, error_msg):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	477 self.disp(_("Error while downloading file: {}").format(error_msg), error=True)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	478
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	479 async def gotId(self, data):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	480 """Called when a progress id has been received"""
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	481 try:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	482 await self.set_progress_id(data["progress"])
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	483 except KeyError:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	484 if 'xmlui' in data:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	485 ui = xmlui_manager.create(self.host, data['xmlui'])
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	486 await ui.show()
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	487 else:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	488 self.disp(_("Can't download file"), error=True)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	489 self.host.quit(C.EXIT_ERROR)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	490
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	491 async def start(self):
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	492 uri = self.args.uri
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	493 dest_file = self.args.dest_file
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	494 if not dest_file:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	495 parsed_uri = urlparse(uri)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	496 dest_file = Path(parsed_uri.path).name.strip() or "downloaded_file"
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	497
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	498 dest_file = Path(dest_file).expanduser().resolve()
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	499 if dest_file.exists() and not self.args.force:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	500 message = _("File {path} already exists! Do you want to overwrite?").format(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	501 path = dest_file)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	502 await self.host.confirmOrQuit(message, _("file download cancelled"))
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	503
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	504 options = {}
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	505
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	506 try:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	507 download_data = await self.host.bridge.fileDownload(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	508 uri,
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	509 str(dest_file),
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	510 data_format.serialise(options),
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	511 self.profile,
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	512 )
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	513 except Exception as e:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	514 self.disp(f"error while trying to download a file: {e}", error=True)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	515 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	516 else:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	517 await self.gotId(download_data)
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	518
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	519
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	520 class Upload(base.CommandBase):
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	521 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	522 super(Upload, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	523 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	524 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	525
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	526 def add_parser_options(self):
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	527 self.parser.add_argument(
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	528 "-e",
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	529 "--encrypt",
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	530 action="store_true",
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	531 help=_("encrypt file using AES-GCM"),
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	532 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	533 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	534 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	535 "jid",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	536 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	537 help=_("jid of upload component (nothing to autodetect)"),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	538 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	539 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	540 "--ignore-tls-errors",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	541 action="store_true",
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	542 help=_(r"ignore invalide TLS certificate (/!\ Dangerous /!\)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	543 )
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	544
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	545 async def onProgressStarted(self, metadata):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	546 self.disp(_("File upload started"), 2)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	547
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	548 async def onProgressFinished(self, metadata):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	549 self.disp(_("File uploaded successfully"), 2)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	550 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	551 url = metadata["url"]
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	552 except KeyError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	553 self.disp("download URL not found in metadata")
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	554 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	555 self.disp(_("URL to retrieve the file:"), 1)
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	556 # XXX: url is displayed alone on a line to make parsing easier
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	557 self.disp(url)
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	558
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	559 async def onProgressError(self, error_msg):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	560 self.disp(_("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	561
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	562 async def gotId(self, data, file_):
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	563 """Called when a progress id has been received
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	564
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	565 @param pid(unicode): progress id
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	566 @param file_(str): file path
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	567 """
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	568 try:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	569 await self.set_progress_id(data["progress"])
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	570 except KeyError:
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	571 if 'xmlui' in data:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	572 ui = xmlui_manager.create(self.host, data['xmlui'])
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	573 await ui.show()
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	574 else:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	575 self.disp(_("Can't upload file"), error=True)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	576 self.host.quit(C.EXIT_ERROR)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	577
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	578 async def start(self):
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	579 file_ = self.args.file
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	580 if not os.path.exists(file_):
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	581 self.disp(_(f"file {file_!r} doesn't exist !"), error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	582 self.host.quit(C.EXIT_BAD_ARG)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	583 if os.path.isdir(file_):
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	584 self.disp(_(f"{file_!r} is a dir! Can't upload a dir"))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	585 self.host.quit(C.EXIT_BAD_ARG)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	586
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	587 if self.args.jid is None:
3068 e189ceca7e8b jp (file/upload): fixed exception when no jid is specified Goffi <goffi@goffi.org> parents: 3040 diff changeset	588 self.full_dest_jid = ""
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	589 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	590 self.full_dest_jid = await self.host.get_full_jid(self.args.jid)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	591
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	592 options = {}
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	593 if self.args.ignore_tls_errors:
3089 e75024e41f81 plugin upload, XEP-0363: code modernisation + preparation for extension: Goffi <goffi@goffi.org> parents: 3068 diff changeset	594 options["ignore_tls_errors"] = True
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	595 if self.args.encrypt:
c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	596 options["encryption"] = C.ENC_AES_GCM
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	597
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	598 path = os.path.abspath(file_)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	599 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	600 upload_data = await self.host.bridge.fileUpload(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	601 path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	602 "",
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	603 self.full_dest_jid,
3089 e75024e41f81 plugin upload, XEP-0363: code modernisation + preparation for extension: Goffi <goffi@goffi.org> parents: 3068 diff changeset	604 data_format.serialise(options),
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	605 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	606 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	607 except Exception as e:
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	608 self.disp(f"error while trying to upload a file: {e}", error=True)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	609 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	610 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	611 await self.gotId(upload_data, file_)
1643 17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	612
17f9b911899a jp (file): new file/upload command Goffi <goffi@goffi.org> parents: 1627 diff changeset	613
3325 7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	614 class ShareAffiliationsSet(base.CommandBase):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	615
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	616 def __init__(self, host):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	617 super(ShareAffiliationsSet, self).__init__(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	618 host,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	619 "set",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	620 use_output=C.OUTPUT_DICT,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	621 help=_("set affiliations for a shared file/directory"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	622 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	623
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	624 def add_parser_options(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	625 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	626 "-N",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	627 "--namespace",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	628 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	629 help=_("namespace of the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	630 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	631 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	632 "-P",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	633 "--path",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	634 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	635 help=_("path to the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	636 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	637 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	638 "-a",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	639 "--affiliation",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	640 dest="affiliations",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	641 metavar=("JID", "AFFILIATION"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	642 required=True,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	643 action="append",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	644 nargs=2,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	645 help=_("entity/affiliation couple(s)"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	646 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	647 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	648 "jid",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	649 help=_("jid of file sharing entity"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	650 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	651
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	652 async def start(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	653 affiliations = dict(self.args.affiliations)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	654 try:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	655 affiliations = await self.host.bridge.FISAffiliationsSet(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	656 self.args.jid,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	657 self.args.namespace,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	658 self.args.path,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	659 affiliations,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	660 self.profile,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	661 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	662 except Exception as e:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	663 self.disp(f"can't set affiliations: {e}", error=True)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	664 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	665 else:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	666 self.host.quit()
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	667
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	668
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	669 class ShareAffiliationsGet(base.CommandBase):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	670 def __init__(self, host):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	671 super(ShareAffiliationsGet, self).__init__(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	672 host,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	673 "get",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	674 use_output=C.OUTPUT_DICT,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	675 help=_("retrieve affiliations of a shared file/directory"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	676 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	677
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	678 def add_parser_options(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	679 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	680 "-N",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	681 "--namespace",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	682 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	683 help=_("namespace of the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	684 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	685 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	686 "-P",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	687 "--path",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	688 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	689 help=_("path to the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	690 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	691 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	692 "jid",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	693 help=_("jid of sharing entity"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	694 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	695
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	696 async def start(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	697 try:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	698 affiliations = await self.host.bridge.FISAffiliationsGet(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	699 self.args.jid,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	700 self.args.namespace,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	701 self.args.path,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	702 self.profile,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	703 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	704 except Exception as e:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	705 self.disp(f"can't get affiliations: {e}", error=True)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	706 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	707 else:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	708 await self.output(affiliations)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	709 self.host.quit()
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	710
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	711
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	712 class ShareAffiliations(base.CommandBase):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	713 subcommands = (ShareAffiliationsGet, ShareAffiliationsSet)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	714
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	715 def __init__(self, host):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	716 super(ShareAffiliations, self).__init__(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	717 host, "affiliations", use_profile=False, help=_("affiliations management")
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	718 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	719
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	720
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	721 class ShareConfigurationSet(base.CommandBase):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	722
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	723 def __init__(self, host):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	724 super(ShareConfigurationSet, self).__init__(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	725 host,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	726 "set",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	727 use_output=C.OUTPUT_DICT,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	728 help=_("set configuration for a shared file/directory"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	729 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	730
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	731 def add_parser_options(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	732 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	733 "-N",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	734 "--namespace",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	735 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	736 help=_("namespace of the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	737 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	738 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	739 "-P",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	740 "--path",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	741 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	742 help=_("path to the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	743 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	744 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	745 "-f",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	746 "--field",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	747 action="append",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	748 nargs=2,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	749 dest="fields",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	750 required=True,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	751 metavar=("KEY", "VALUE"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	752 help=_("configuration field to set (required)"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	753 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	754 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	755 "jid",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	756 help=_("jid of file sharing entity"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	757 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	758
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	759 async def start(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	760 configuration = dict(self.args.fields)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	761 try:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	762 configuration = await self.host.bridge.FISConfigurationSet(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	763 self.args.jid,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	764 self.args.namespace,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	765 self.args.path,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	766 configuration,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	767 self.profile,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	768 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	769 except Exception as e:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	770 self.disp(f"can't set configuration: {e}", error=True)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	771 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	772 else:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	773 self.host.quit()
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	774
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	775
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	776 class ShareConfigurationGet(base.CommandBase):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	777 def __init__(self, host):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	778 super(ShareConfigurationGet, self).__init__(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	779 host,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	780 "get",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	781 use_output=C.OUTPUT_DICT,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	782 help=_("retrieve configuration of a shared file/directory"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	783 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	784
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	785 def add_parser_options(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	786 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	787 "-N",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	788 "--namespace",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	789 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	790 help=_("namespace of the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	791 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	792 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	793 "-P",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	794 "--path",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	795 default="",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	796 help=_("path to the repository"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	797 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	798 self.parser.add_argument(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	799 "jid",
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	800 help=_("jid of sharing entity"),
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	801 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	802
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	803 async def start(self):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	804 try:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	805 configuration = await self.host.bridge.FISConfigurationGet(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	806 self.args.jid,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	807 self.args.namespace,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	808 self.args.path,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	809 self.profile,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	810 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	811 except Exception as e:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	812 self.disp(f"can't get configuration: {e}", error=True)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	813 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	814 else:
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	815 await self.output(configuration)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	816 self.host.quit()
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	817
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	818
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	819 class ShareConfiguration(base.CommandBase):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	820 subcommands = (ShareConfigurationGet, ShareConfigurationSet)
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	821
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	822 def __init__(self, host):
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	823 super(ShareConfiguration, self).__init__(
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	824 host, "configuration", use_profile=False,
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	825 help=_("file sharing node configuration")
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	826 )
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	827
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	828
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	829 class ShareList(base.CommandBase):
817 c39117d00f35 jp: refactoring: Goffi <goffi@goffi.org> parents: 815 diff changeset	830 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	831 extra_outputs = {"default": self.default_output}
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	832 super(ShareList, self).__init__(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	833 host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	834 "list",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	835 use_output=C.OUTPUT_LIST_DICT,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	836 extra_outputs=extra_outputs,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	837 help=_("retrieve files shared by an entity"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	838 use_verbose=True,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	839 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	840
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	841 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	842 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	843 "-d",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	844 "--path",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	845 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	846 help=_("path to the directory containing the files"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	847 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	848 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	849 "jid",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	850 nargs="?",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	851 default="",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	852 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	853 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	854
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	855 def _name_filter(self, name, row):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	856 if row.type == C.FILE_TYPE_DIRECTORY:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	857 return A.color(C.A_DIRECTORY, name)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	858 elif row.type == C.FILE_TYPE_FILE:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	859 return A.color(C.A_FILE, name)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	860 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	861 self.disp(_("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	862 return name
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	863
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	864 def _size_filter(self, size, row):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	865 if not size:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	866 return ""
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	867 size = int(size)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	868 # cf. https://stackoverflow.com/a/1094933 (thanks)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	869 suffix = "o"
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	870 for unit in ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi"]:
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	871 if abs(size) < 1024.0:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	872 return A.color(A.BOLD, "{:.2f}".format(size), unit, suffix)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	873 size /= 1024.0
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	874
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	875 return A.color(A.BOLD, "{:.2f}".format(size), "Yi", suffix)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	876
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	877 def default_output(self, files_data):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	878 """display files a way similar to ls"""
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	879 files_data.sort(key=lambda d: d["name"].lower())
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	880 show_header = False
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	881 if self.verbosity == 0:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	882 keys = headers = ("name", "type")
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	883 elif self.verbosity == 1:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	884 keys = headers = ("name", "type", "size")
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	885 elif self.verbosity > 1:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	886 show_header = True
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	887 keys = ("name", "type", "size", "file_hash")
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	888 headers = ("name", "type", "size", "hash")
3121 040ca99e25fe jp (common): various Table fixes: Goffi <goffi@goffi.org> parents: 3094 diff changeset	889 table = common.Table.fromListDict(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	890 self.host,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	891 files_data,
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	892 keys=keys,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	893 headers=headers,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	894 filters={"name": self._name_filter, "size": self._size_filter},
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	895 defaults={"size": "", "file_hash": ""},
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	896 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	897 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	898
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	899 async def start(self):
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	900 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	901 files_data = await self.host.bridge.FISList(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	902 self.args.jid,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	903 self.args.path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	904 {},
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	905 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	906 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	907 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	908 self.disp(f"can't retrieve shared files: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	909 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	910
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	911 await self.output(files_data)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	912 self.host.quit()
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	913
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	914
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	915 class SharePath(base.CommandBase):
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	916 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	917 super(SharePath, self).__init__(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	918 host, "path", help=_("share a file or directory"), use_verbose=True
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	919 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	920
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	921 def add_parser_options(self):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	922 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	923 "-n",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	924 "--name",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	925 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	926 help=_("virtual name to use (default: use directory/file name)"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	927 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	928 perm_group = self.parser.add_mutually_exclusive_group()
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	929 perm_group.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	930 "-j",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	931 "--jid",
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	932 metavar="JID",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	933 action="append",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	934 dest="jids",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	935 default=[],
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	936 help=_("jid of contacts allowed to retrieve the files"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	937 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	938 perm_group.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	939 "--public",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	940 action="store_true",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	941 help=_(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	942 r"share publicly the file(s) (/!\ everybody will be able to access "
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	943 r"them)"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	944 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	945 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	946 self.parser.add_argument(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	947 "path",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	948 help=_("path to a file or directory to share"),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	949 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	950
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	951 async def start(self):
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	952 self.path = os.path.abspath(self.args.path)
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	953 if self.args.public:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	954 access = {"read": {"type": "public"}}
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	955 else:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	956 jids = self.args.jids
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	957 if jids:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	958 access = {"read": {"type": "whitelist", "jids": jids}}
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	959 else:
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	960 access = {}
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	961 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	962 name = await self.host.bridge.FISSharePath(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	963 self.args.name,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	964 self.path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	965 json.dumps(access, ensure_ascii=False),
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	966 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	967 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	968 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	969 self.disp(f"can't share path: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	970 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	971 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	972 self.disp(_(f'{self.path} shared under the name "{name}"'))
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	973 self.host.quit()
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	974
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	975
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	976 class ShareInvite(base.CommandBase):
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	977 def __init__(self, host):
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	978 super(ShareInvite, self).__init__(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	979 host, "invite", help=_("send invitation for a shared repository")
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	980 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	981
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	982 def add_parser_options(self):
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	983 self.parser.add_argument(
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	984 "-n",
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	985 "--name",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	986 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	987 help=_("name of the repository"),
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	988 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	989 self.parser.add_argument(
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	990 "-N",
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	991 "--namespace",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	992 default="",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	993 help=_("namespace of the repository"),
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	994 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	995 self.parser.add_argument(
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	996 "-P",
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	997 "--path",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	998 help=_("path to the repository"),
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	999 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1000 self.parser.add_argument(
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1001 "-t",
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1002 "--type",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1003 choices=["files", "photos"],
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1004 default="files",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1005 help=_("type of the repository"),
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1006 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1007 self.parser.add_argument(
2932 472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1008 "-T",
472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1009 "--thumbnail",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1010 help=_("https URL of a image to use as thumbnail"),
2932 472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1011 )
472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1012 self.parser.add_argument(
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1013 "service",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1014 help=_("jid of the file sharing service hosting the repository"),
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1015 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1016 self.parser.add_argument(
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1017 "jid",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1018 help=_("jid of the person to invite"),
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1019 )
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1020
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1021 async def start(self):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1022 self.path = os.path.normpath(self.args.path) if self.args.path else ""
2932 472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1023 extra = {}
472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1024 if self.args.thumbnail is not None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1025 if not self.args.thumbnail.startswith('http'):
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1026 self.parser.error(_("only http(s) links are allowed with --thumbnail"))
2932 472fadadefe6 jp (file/invite): added a --thumbnail argument Goffi <goffi@goffi.org> parents: 2916 diff changeset	1027 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1028 extra['thumb_url'] = self.args.thumbnail
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1029 try:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1030 await self.host.bridge.FISInvite(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1031 self.args.jid,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1032 self.args.service,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1033 self.args.type,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1034 self.args.namespace,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1035 self.path,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1036 self.args.name,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1037 data_format.serialise(extra),
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1038 self.profile,
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1039 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1040 except Exception as e:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1041 self.disp(f"can't send invitation: {e}", error=True)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1042 self.host.quit(C.EXIT_BRIDGE_ERRBACK)
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1043 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1044 self.disp(
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1045 _(f'invitation sent to {self.args.jid}')
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1046 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3028 diff changeset	1047 self.host.quit()
2916 0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1048
0b9faea5cb58 jp (file/share): added invite command to invite an entity to a file sharing repository Goffi <goffi@goffi.org> parents: 2771 diff changeset	1049
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1050 class Share(base.CommandBase):
3325 7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	1051 subcommands = (
7ebda4b54170 jp (file/share): added commands to manage affiliations and configuration + documentation Goffi <goffi@goffi.org> parents: 3137 diff changeset	1052 ShareList, SharePath, ShareInvite, ShareAffiliations, ShareConfiguration)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1053
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1054 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	1055 super(Share, self).__init__(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1056 host, "share", use_profile=False, help=_("files sharing management")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	1057 )
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1058
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1059
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1060 class File(base.CommandBase):
3094 c3cb18236bdf jp (file): new `get` command + encryption with upload: Goffi <goffi@goffi.org> parents: 3089 diff changeset	1061 subcommands = (Send, Request, Receive, Get, Upload, Share)
2505 8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1062
8e770ac05b0c jp (file): file sharing + improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	1063 def __init__(self, host):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	1064 super(File, self).__init__(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2939 diff changeset	1065 host, "file", use_profile=False, help=_("files sending/receiving/management")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2592 diff changeset	1066 )

Mercurial > libervia-backend

annotate sat_frontends/jp/cmd_file.py @ 3410:55f9a38864af