libervia-backend: sat_frontends/jp/cmd

annotate sat_frontends/jp/cmd_file.py @ 3433:63d20854df3a

core (constants): added EXIT_CONFLICT

author	Goffi <goffi@goffi.org>
date	Fri, 04 Dec 2020 12:30:46 +0100
parents	f30b238d9c45
children	be6d91572633

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

Mercurial > libervia-backend

annotate sat_frontends/jp/cmd_file.py @ 3433:63d20854df3a