libervia-backend: sat_frontends/jp/cmd

annotate sat_frontends/jp/cmd_file.py @ 3528:849374e59178

component file sharing: quotas implementation: quotas can now be specified using the `quotas_json` option of `component file_sharing` section in settings. This must be a dict where: - `users` key contains default quotas for all users - `admins` key contains quotas for administrators (not implemented yet) - `jids` contain bare JID to quota mapping, to have user-specific quota The value can be either a int for quota in bytes, or a case insensitive string with an optional multiplier symbol (e.g. "500 Mio"). `None` can be used for explicit unlimited quota (which is the default is `users` is not set). When a file size is too big for quota, upload is refused with an error message indicating allowed quota, used space, and the size of the file that user wants to upload.

author	Goffi <goffi@goffi.org>
date	Wed, 05 May 2021 15:37:33 +0200
parents	584379473925
children	04283582966f

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

Mercurial > libervia-backend

annotate sat_frontends/jp/cmd_file.py @ 3528:849374e59178