mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2025-01-03 19:54:09 +00:00
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
"""
|
|
Ported to Python 3.
|
|
"""
|
|
from __future__ import unicode_literals
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
from future.utils import PY2
|
|
if PY2:
|
|
from future.builtins import filter, map, zip, ascii, chr, hex, input, next, oct, open, pow, round, super, bytes, dict, list, object, range, str, max, min # noqa: F401
|
|
|
|
import re
|
|
from urllib.parse import quote as url_quote
|
|
import json
|
|
from allmydata.scripts.common import get_alias, DEFAULT_ALIAS, escape_path, \
|
|
UnknownAliasError
|
|
from allmydata.scripts.common_http import do_http, format_http_error
|
|
from allmydata.util.encodingutil import to_bytes
|
|
|
|
# this script is used for both 'mv' and 'ln'
|
|
|
|
def mv(options, mode="move"):
|
|
nodeurl = options['node-url']
|
|
aliases = options.aliases
|
|
from_file = options.from_file
|
|
to_file = options.to_file
|
|
stdout = options.stdout
|
|
stderr = options.stderr
|
|
|
|
if nodeurl[-1] != "/":
|
|
nodeurl += "/"
|
|
try:
|
|
rootcap, from_path = get_alias(aliases, from_file, DEFAULT_ALIAS)
|
|
except UnknownAliasError as e:
|
|
e.display(stderr)
|
|
return 1
|
|
from_path = str(from_path, "utf-8")
|
|
from_url = nodeurl + "uri/%s" % url_quote(rootcap)
|
|
if from_path:
|
|
from_url += "/" + escape_path(from_path)
|
|
# figure out the source cap
|
|
resp = do_http("GET", from_url + "?t=json")
|
|
if not re.search(r'^2\d\d$', str(resp.status)):
|
|
print(format_http_error("Error", resp), file=stderr)
|
|
return 1
|
|
data = resp.read()
|
|
nodetype, attrs = json.loads(data)
|
|
cap = to_bytes(attrs.get("rw_uri") or attrs["ro_uri"])
|
|
|
|
# now get the target
|
|
try:
|
|
rootcap, path = get_alias(aliases, to_file, DEFAULT_ALIAS)
|
|
except UnknownAliasError as e:
|
|
e.display(stderr)
|
|
return 1
|
|
to_url = nodeurl + "uri/%s" % url_quote(rootcap)
|
|
path = str(path, "utf-8")
|
|
if path:
|
|
to_url += "/" + escape_path(path)
|
|
|
|
if to_url.endswith("/"):
|
|
# "mv foo.txt bar/" == "mv foo.txt bar/foo.txt"
|
|
to_url += escape_path(from_path[from_path.rfind("/")+1:])
|
|
|
|
to_url += "?t=uri&replace=only-files"
|
|
|
|
resp = do_http("PUT", to_url, cap)
|
|
status = resp.status
|
|
if not re.search(r'^2\d\d$', str(status)):
|
|
if status == 409:
|
|
print("Error: You can't overwrite a directory with a file", file=stderr)
|
|
else:
|
|
print(format_http_error("Error", resp), file=stderr)
|
|
if mode == "move":
|
|
print("NOT removing the original", file=stderr)
|
|
return 1
|
|
|
|
if mode == "move":
|
|
# now remove the original
|
|
resp = do_http("DELETE", from_url)
|
|
if not re.search(r'^2\d\d$', str(resp.status)):
|
|
print(format_http_error("Error deleting original after move", resp), file=stderr)
|
|
return 2
|
|
|
|
print("OK", file=stdout)
|
|
return 0
|