50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import urllib.parse
|
||
|
|
import urllib.request
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
|
||
|
|
def get_url_scheme(url):
|
||
|
|
# type: (str) -> Optional[str]
|
||
|
|
if ":" not in url:
|
||
|
|
return None
|
||
|
|
return url.split(":", 1)[0].lower()
|
||
|
|
|
||
|
|
|
||
|
|
def path_to_url(path):
|
||
|
|
# type: (str) -> str
|
||
|
|
"""
|
||
|
|
Convert a path to a file: URL. The path will be made absolute and have
|
||
|
|
quoted path parts.
|
||
|
|
"""
|
||
|
|
path = os.path.normpath(os.path.abspath(path))
|
||
|
|
url = urllib.parse.urljoin("file:", urllib.request.pathname2url(path))
|
||
|
|
return url
|
||
|
|
|
||
|
|
|
||
|
|
def url_to_path(url):
|
||
|
|
# type: (str) -> str
|
||
|
|
"""
|
||
|
|
Convert a file: URL to a path.
|
||
|
|
"""
|
||
|
|
assert url.startswith(
|
||
|
|
"file:"
|
||
|
|
), f"You can only turn file: urls into filenames (not {url!r})"
|
||
|
|
|
||
|
|
_, netloc, path, _, _ = urllib.parse.urlsplit(url)
|
||
|
|
|
||
|
|
if not netloc or netloc == "localhost":
|
||
|
|
# According to RFC 8089, same as empty authority.
|
||
|
|
netloc = ""
|
||
|
|
elif sys.platform == "win32":
|
||
|
|
# If we have a UNC path, prepend UNC share notation.
|
||
|
|
netloc = "\\\\" + netloc
|
||
|
|
else:
|
||
|
|
raise ValueError(
|
||
|
|
f"non-local file URIs are not supported on this platform: {url!r}"
|
||
|
|
)
|
||
|
|
|
||
|
|
path = urllib.request.url2pathname(netloc + path)
|
||
|
|
return path
|