brieftaube/BTvenv/lib/python3.6/site-packages/yarl/__init__.py

950 lines
30 KiB
Python
Raw Normal View History

2020-08-27 21:39:46 +02:00
import warnings
from collections.abc import Mapping, Sequence
from ipaddress import ip_address
from urllib.parse import (SplitResult, parse_qsl,
urljoin, urlsplit, urlunsplit)
from multidict import MultiDict, MultiDictProxy
import idna
from .quoting import _Quoter, _Unquoter
__version__ = '1.2.6'
__all__ = ('URL',)
# is_leaf()
DEFAULT_PORTS = {
'http': 80,
'https': 443,
'ws': 80,
'wss': 443,
}
sentinel = object()
class cached_property:
"""Use as a class method decorator. It operates almost exactly like
the Python `@property` decorator, but it puts the result of the
method it decorates into the instance dict after the first call,
effectively replacing the function it decorates with an instance
variable. It is, in Python parlance, a data descriptor.
"""
def __init__(self, wrapped):
self.wrapped = wrapped
try:
self.__doc__ = wrapped.__doc__
except AttributeError: # pragma: no cover
self.__doc__ = ""
self.name = wrapped.__name__
def __get__(self, inst, owner, _sentinel=sentinel):
if inst is None:
return self
val = inst._cache.get(self.name, _sentinel)
if val is not _sentinel:
return val
val = self.wrapped(inst)
inst._cache[self.name] = val
return val
def __set__(self, inst, value):
raise AttributeError("cached property is read-only")
class URL:
# Don't derive from str
# follow pathlib.Path design
# probably URL will not suffer from pathlib problems:
# it's intended for libraries like aiohttp,
# not to be passed into standard library functions like os.open etc.
# URL grammar (RFC 3986)
# pct-encoded = "%" HEXDIG HEXDIG
# reserved = gen-delims / sub-delims
# gen-delims = ":" / "/" / "?" / "#" / "[" / "]" / "@"
# sub-delims = "!" / "$" / "&" / "'" / "(" / ")"
# / "*" / "+" / "," / ";" / "="
# unreserved = ALPHA / DIGIT / "-" / "." / "_" / "~"
# URI = scheme ":" hier-part [ "?" query ] [ "#" fragment ]
# hier-part = "//" authority path-abempty
# / path-absolute
# / path-rootless
# / path-empty
# scheme = ALPHA *( ALPHA / DIGIT / "+" / "-" / "." )
# authority = [ userinfo "@" ] host [ ":" port ]
# userinfo = *( unreserved / pct-encoded / sub-delims / ":" )
# host = IP-literal / IPv4address / reg-name
# IP-literal = "[" ( IPv6address / IPvFuture ) "]"
# IPvFuture = "v" 1*HEXDIG "." 1*( unreserved / sub-delims / ":" )
# IPv6address = 6( h16 ":" ) ls32
# / "::" 5( h16 ":" ) ls32
# / [ h16 ] "::" 4( h16 ":" ) ls32
# / [ *1( h16 ":" ) h16 ] "::" 3( h16 ":" ) ls32
# / [ *2( h16 ":" ) h16 ] "::" 2( h16 ":" ) ls32
# / [ *3( h16 ":" ) h16 ] "::" h16 ":" ls32
# / [ *4( h16 ":" ) h16 ] "::" ls32
# / [ *5( h16 ":" ) h16 ] "::" h16
# / [ *6( h16 ":" ) h16 ] "::"
# ls32 = ( h16 ":" h16 ) / IPv4address
# ; least-significant 32 bits of address
# h16 = 1*4HEXDIG
# ; 16 bits of address represented in hexadecimal
# IPv4address = dec-octet "." dec-octet "." dec-octet "." dec-octet
# dec-octet = DIGIT ; 0-9
# / %x31-39 DIGIT ; 10-99
# / "1" 2DIGIT ; 100-199
# / "2" %x30-34 DIGIT ; 200-249
# / "25" %x30-35 ; 250-255
# reg-name = *( unreserved / pct-encoded / sub-delims )
# port = *DIGIT
# path = path-abempty ; begins with "/" or is empty
# / path-absolute ; begins with "/" but not "//"
# / path-noscheme ; begins with a non-colon segment
# / path-rootless ; begins with a segment
# / path-empty ; zero characters
# path-abempty = *( "/" segment )
# path-absolute = "/" [ segment-nz *( "/" segment ) ]
# path-noscheme = segment-nz-nc *( "/" segment )
# path-rootless = segment-nz *( "/" segment )
# path-empty = 0<pchar>
# segment = *pchar
# segment-nz = 1*pchar
# segment-nz-nc = 1*( unreserved / pct-encoded / sub-delims / "@" )
# ; non-zero-length segment without any colon ":"
# pchar = unreserved / pct-encoded / sub-delims / ":" / "@"
# query = *( pchar / "/" / "?" )
# fragment = *( pchar / "/" / "?" )
# URI-reference = URI / relative-ref
# relative-ref = relative-part [ "?" query ] [ "#" fragment ]
# relative-part = "//" authority path-abempty
# / path-absolute
# / path-noscheme
# / path-empty
# absolute-URI = scheme ":" hier-part [ "?" query ]
__slots__ = ('_cache', '_val')
_QUOTER = _Quoter()
_PATH_QUOTER = _Quoter(safe='@:', protected='/+')
_QUERY_QUOTER = _Quoter(safe='?/:@', protected='=+&;', qs=True)
_QUERY_PART_QUOTER = _Quoter(safe='?/:@', qs=True)
_FRAGMENT_QUOTER = _Quoter(safe='?/:@')
_UNQUOTER = _Unquoter()
_PATH_UNQUOTER = _Unquoter(unsafe='+')
_QS_UNQUOTER = _Unquoter(qs=True)
def __new__(cls, val='', *, encoded=False, strict=None):
if strict is not None: # pragma: no cover
warnings.warn("strict parameter is ignored")
if type(val) is cls:
return val
if type(val) is str:
val = urlsplit(val)
elif type(val) is SplitResult:
if not encoded:
raise ValueError("Cannot apply decoding to SplitResult")
elif isinstance(val, str):
val = urlsplit(str(val))
else:
raise TypeError("Constructor parameter should be str")
if not encoded:
if not val[1]: # netloc
netloc = ''
else:
host = val.hostname
if host is None:
raise ValueError(
"Invalid URL: host is required for abolute urls.")
netloc = cls._make_netloc(val.username,
val.password,
host,
val.port,
encode=True)
path = cls._PATH_QUOTER(val[2])
if netloc:
path = cls._normalize_path(path)
query = cls._QUERY_QUOTER(val[3])
fragment = cls._FRAGMENT_QUOTER(val[4])
val = SplitResult(val[0], netloc, path, query, fragment)
self = object.__new__(cls)
self._val = val
self._cache = {}
return self
@classmethod
def build(cls, *, scheme='', user='', password='', host='', port=None,
path='', query=None, query_string='', fragment='',
encoded=False):
"""Creates and returns a new URL"""
if host and not scheme:
raise ValueError(
'Can\'t build URL with "host" but without "scheme".')
if not host and scheme:
raise ValueError(
'Can\'t build URL with "scheme" but without "host".')
if query and query_string:
raise ValueError(
"Only one of \"query\" or \"query_string\" should be passed")
if not user and not password and not host and not port:
netloc = ''
else:
netloc = cls._make_netloc(user, password, host, port,
encode=not encoded)
if not encoded:
path = cls._PATH_QUOTER(path)
if netloc:
path = cls._normalize_path(path)
query_string = cls._QUERY_QUOTER(query_string)
fragment = cls._FRAGMENT_QUOTER(fragment)
url = cls(
SplitResult(
scheme,
netloc,
path,
query_string,
fragment
),
encoded=True
)
if query:
return url.with_query(query)
else:
return url
def __init_subclass__(cls):
raise TypeError("Inheritance a class {!r} from URL "
"is forbidden".format(cls))
def __str__(self):
val = self._val
if not val.path and self.is_absolute() and (val.query or val.fragment):
val = val._replace(path='/')
return urlunsplit(val)
def __repr__(self):
return "{}('{}')".format(self.__class__.__name__, str(self))
def __eq__(self, other):
if not type(other) is URL:
return NotImplemented
val1 = self._val
if not val1.path and self.is_absolute():
val1 = val1._replace(path='/')
val2 = other._val
if not val2.path and other.is_absolute():
val2 = val2._replace(path='/')
return val1 == val2
def __hash__(self):
ret = self._cache.get('hash')
if ret is None:
val = self._val
if not val.path and self.is_absolute():
val = val._replace(path='/')
ret = self._cache['hash'] = hash(val)
return ret
def __le__(self, other):
if not type(other) is URL:
return NotImplemented
return self._val <= other._val
def __lt__(self, other):
if not type(other) is URL:
return NotImplemented
return self._val < other._val
def __ge__(self, other):
if not type(other) is URL:
return NotImplemented
return self._val >= other._val
def __gt__(self, other):
if not type(other) is URL:
return NotImplemented
return self._val > other._val
def __truediv__(self, name):
name = self._PATH_QUOTER(name)
if name.startswith('/'):
raise ValueError("Appending path "
"starting from slash is forbidden")
path = self._val.path
if path == '/':
new_path = '/' + name
elif not path and not self.is_absolute():
new_path = name
else:
parts = path.rstrip('/').split('/')
parts.append(name)
new_path = '/'.join(parts)
if self.is_absolute():
new_path = self._normalize_path(new_path)
return URL(self._val._replace(path=new_path, query='', fragment=''),
encoded=True)
def __getstate__(self):
return self._val,
def __setstate__(self, state):
if state[0] is None and isinstance(state[1], dict):
# default style pickle
self._val = state[1]['_val']
else:
self._val, *unused = state
self._cache = {}
def is_absolute(self):
"""A check for absolute URLs.
Return True for absolute ones (having scheme or starting
with //), False otherwise.
"""
return self.raw_host is not None
def is_default_port(self):
"""A check for default port.
Return True if port is default for specified scheme,
e.g. 'http://python.org' or 'http://python.org:80', False
otherwise.
"""
if self.port is None:
return False
default = DEFAULT_PORTS.get(self.scheme)
if default is None:
return False
return self.port == default
def origin(self):
"""Return an URL with scheme, host and port parts only.
user, password, path, query and fragment are removed.
"""
# TODO: add a keyword-only option for keeping user/pass maybe?
if not self.is_absolute():
raise ValueError("URL should be absolute")
if not self._val.scheme:
raise ValueError("URL should have scheme")
v = self._val
netloc = self._make_netloc(None, None, v.hostname, v.port,
encode=False)
val = v._replace(netloc=netloc, path='', query='', fragment='')
return URL(val, encoded=True)
def relative(self):
"""Return a relative part of the URL.
scheme, user, password, host and port are removed.
"""
if not self.is_absolute():
raise ValueError("URL should be absolute")
val = self._val._replace(scheme='', netloc='')
return URL(val, encoded=True)
@property
def scheme(self):
"""Scheme for absolute URLs.
Empty string for relative URLs or URLs starting with //
"""
return self._val.scheme
@property
def raw_user(self):
"""Encoded user part of URL.
None if user is missing.
"""
# not .username
ret = self._val.username
if not ret:
return None
return ret
@cached_property
def user(self):
"""Decoded user part of URL.
None if user is missing.
"""
return self._UNQUOTER(self.raw_user)
@property
def raw_password(self):
"""Encoded password part of URL.
None if password is missing.
"""
return self._val.password
@cached_property
def password(self):
"""Decoded password part of URL.
None if password is missing.
"""
return self._UNQUOTER(self.raw_password)
@property
def raw_host(self):
"""Encoded host part of URL.
None for relative URLs.
"""
# Use host instead of hostname for sake of shortness
# May add .hostname prop later
return self._val.hostname
@cached_property
def host(self):
"""Decoded host part of URL.
None for relative URLs.
"""
raw = self.raw_host
if raw is None:
return None
if '%' in raw:
# Hack for scoped IPv6 addresses like
# fe80::2%Проверка
# presence of '%' sign means only IPv6 address, so idna is useless.
return raw
try:
return idna.decode(raw.encode('ascii'))
except UnicodeError: # e.g. '::1'
return raw.encode('ascii').decode('idna')
@property
def port(self):
"""Port part of URL.
None for relative URLs or URLs without explicit port and
scheme without default port substitution.
"""
return self._val.port or DEFAULT_PORTS.get(self._val.scheme)
@property
def raw_path(self):
"""Encoded path of URL.
/ for absolute URLs without path part.
"""
ret = self._val.path
if not ret and self.is_absolute():
ret = '/'
return ret
@cached_property
def path(self):
"""Decoded path of URL.
/ for absolute URLs without path part.
"""
return self._PATH_UNQUOTER(self.raw_path)
@cached_property
def query(self):
"""A MultiDictProxy representing parsed query parameters in decoded
representation.
Empty value if URL has no query part.
"""
ret = MultiDict(parse_qsl(self.raw_query_string,
keep_blank_values=True))
return MultiDictProxy(ret)
@property
def raw_query_string(self):
"""Encoded query part of URL.
Empty string if query is missing.
"""
return self._val.query
@cached_property
def query_string(self):
"""Decoded query part of URL.
Empty string if query is missing.
"""
return self._QS_UNQUOTER(self.raw_query_string)
@cached_property
def path_qs(self):
"""Decoded path of URL with query."""
if not self.query_string:
return self.path
return '{}?{}'.format(self.path, self.query_string)
@cached_property
def raw_path_qs(self):
"""Encoded path of URL with query."""
if not self.raw_query_string:
return self.raw_path
return '{}?{}'.format(self.raw_path, self.raw_query_string)
@property
def raw_fragment(self):
"""Encoded fragment part of URL.
Empty string if fragment is missing.
"""
return self._val.fragment
@cached_property
def fragment(self):
"""Decoded fragment part of URL.
Empty string if fragment is missing.
"""
return self._UNQUOTER(self.raw_fragment)
@cached_property
def raw_parts(self):
"""A tuple containing encoded *path* parts.
('/',) for absolute URLs if *path* is missing.
"""
path = self._val.path
if self.is_absolute():
if not path:
parts = ['/']
else:
parts = ['/'] + path[1:].split('/')
else:
if path.startswith('/'):
parts = ['/'] + path[1:].split('/')
else:
parts = path.split('/')
return tuple(parts)
@cached_property
def parts(self):
"""A tuple containing decoded *path* parts.
('/',) for absolute URLs if *path* is missing.
"""
return tuple(self._UNQUOTER(part) for part in self.raw_parts)
@cached_property
def parent(self):
"""A new URL with last part of path removed and cleaned up query and
fragment.
"""
path = self.raw_path
if not path or path == '/':
if self.raw_fragment or self.raw_query_string:
return URL(self._val._replace(query='', fragment=''),
encoded=True)
return self
parts = path.split('/')
val = self._val._replace(path='/'.join(parts[:-1]),
query='', fragment='')
return URL(val, encoded=True)
@cached_property
def raw_name(self):
"""The last part of raw_parts."""
parts = self.raw_parts
if self.is_absolute():
parts = parts[1:]
if not parts:
return ''
else:
return parts[-1]
else:
return parts[-1]
@cached_property
def name(self):
"""The last part of parts."""
return self._UNQUOTER(self.raw_name)
@classmethod
def _normalize_path(cls, path):
# Drop '.' and '..' from path
segments = path.split('/')
resolved_path = []
for seg in segments:
if seg == '..':
try:
resolved_path.pop()
except IndexError:
# ignore any .. segments that would otherwise cause an
# IndexError when popped from resolved_path if
# resolving for rfc3986
pass
elif seg == '.':
continue
else:
resolved_path.append(seg)
if segments[-1] in ('.', '..'):
# do some post-processing here.
# if the last segment was a relative dir,
# then we need to append the trailing '/'
resolved_path.append('')
return '/'.join(resolved_path)
@classmethod
def _encode_host(cls, host):
try:
ip, sep, zone = host.partition('%')
ip = ip_address(ip)
except ValueError:
try:
host = idna.encode(host, uts46=True).decode('ascii')
except UnicodeError:
host = host.encode('idna').decode('ascii')
else:
host = ip.compressed
if sep:
host += '%' + zone
if ip.version == 6:
host = '[' + host + ']'
return host
@classmethod
def _make_netloc(cls, user, password, host, port, encode):
if encode:
ret = cls._encode_host(host)
else:
ret = host
if port:
ret = ret + ':' + str(port)
if password:
if not user:
user = ''
else:
if encode:
user = cls._QUOTER(user)
if encode:
password = cls._QUOTER(password)
user = user + ':' + password
elif user and encode:
user = cls._QUOTER(user)
if user:
ret = user + '@' + ret
return ret
def with_scheme(self, scheme):
"""Return a new URL with scheme replaced."""
# N.B. doesn't cleanup query/fragment
if not isinstance(scheme, str):
raise TypeError("Invalid scheme type")
if not self.is_absolute():
raise ValueError("scheme replacement is not allowed "
"for relative URLs")
return URL(self._val._replace(scheme=scheme.lower()), encoded=True)
def with_user(self, user):
"""Return a new URL with user replaced.
Autoencode user if needed.
Clear user/password if user is None.
"""
# N.B. doesn't cleanup query/fragment
val = self._val
if user is None:
password = None
elif isinstance(user, str):
user = self._QUOTER(user)
password = val.password
else:
raise TypeError("Invalid user type")
if not self.is_absolute():
raise ValueError("user replacement is not allowed "
"for relative URLs")
return URL(self._val._replace(netloc=self._make_netloc(user,
password,
val.hostname,
val.port,
encode=False)),
encoded=True)
def with_password(self, password):
"""Return a new URL with password replaced.
Autoencode password if needed.
Clear password if argument is None.
"""
# N.B. doesn't cleanup query/fragment
if password is None:
pass
elif isinstance(password, str):
password = self._QUOTER(password)
else:
raise TypeError("Invalid password type")
if not self.is_absolute():
raise ValueError("password replacement is not allowed "
"for relative URLs")
val = self._val
return URL(
self._val._replace(
netloc=self._make_netloc(val.username,
password,
val.hostname,
val.port,
encode=False)),
encoded=True)
def with_host(self, host):
"""Return a new URL with host replaced.
Autoencode host if needed.
Changing host for relative URLs is not allowed, use .join()
instead.
"""
# N.B. doesn't cleanup query/fragment
if not isinstance(host, str):
raise TypeError("Invalid host type")
if not self.is_absolute():
raise ValueError("host replacement is not allowed "
"for relative URLs")
if not host:
raise ValueError("host removing is not allowed")
host = self._encode_host(host)
val = self._val
return URL(
self._val._replace(netloc=self._make_netloc(val.username,
val.password,
host,
val.port,
encode=False)),
encoded=True)
def with_port(self, port):
"""Return a new URL with port replaced.
Clear port to default if None is passed.
"""
# N.B. doesn't cleanup query/fragment
if port is not None and not isinstance(port, int):
raise TypeError(
"port should be int or None, got {}".format(type(port)))
if not self.is_absolute():
raise ValueError("port replacement is not allowed "
"for relative URLs")
val = self._val
return URL(
self._val._replace(netloc=self._make_netloc(val.username,
val.password,
val.hostname,
port,
encode=False)),
encoded=True)
def with_path(self, path, *, encoded=False):
"""Return a new URL with path replaced."""
if not encoded:
path = self._PATH_QUOTER(path)
if self.is_absolute():
path = self._normalize_path(path)
if len(path) > 0 and path[0] != '/':
path = '/' + path
return URL(self._val._replace(path=path, query='', fragment=''),
encoded=True)
def _get_str_query(self, *args, **kwargs):
if kwargs:
if len(args) > 0:
raise ValueError("Either kwargs or single query parameter "
"must be present")
query = kwargs
elif len(args) == 1:
query = args[0]
else:
raise ValueError("Either kwargs or single query parameter "
"must be present")
if query is None:
query = ''
elif isinstance(query, Mapping):
quoter = self._QUERY_PART_QUOTER
lst = []
for k, v in query.items():
if isinstance(v, str):
pass
elif type(v) == int: # no subclasses like bool
v = str(v)
else:
raise TypeError("Invalid variable type: mapping value "
"should be str or int, got {!r}".format(v))
lst.append(
quoter(k) + '=' + quoter(v))
query = '&'.join(lst)
elif isinstance(query, str):
query = self._QUERY_QUOTER(query)
elif isinstance(query, (bytes, bytearray, memoryview)):
raise TypeError("Invalid query type: bytes, bytearray and "
"memoryview are forbidden")
elif isinstance(query, Sequence):
quoter = self._QUERY_PART_QUOTER
query = '&'.join(quoter(k) + '=' + quoter(v)
for k, v in query)
else:
raise TypeError("Invalid query type: only str, mapping or "
"sequence of (str, str) pairs is allowed")
return query
def with_query(self, *args, **kwargs):
"""Return a new URL with query part replaced.
Accepts any Mapping (e.g. dict, multidict.MultiDict instances)
or str, autoencode the argument if needed.
It also can take an arbitrary number of keyword arguments.
Clear query if None is passed.
"""
# N.B. doesn't cleanup query/fragment
new_query = self._get_str_query(*args, **kwargs)
return URL(
self._val._replace(path=self._val.path, query=new_query),
encoded=True)
def update_query(self, *args, **kwargs):
"""Return a new URL with query part updated."""
s = self._get_str_query(*args, **kwargs)
new_query = MultiDict(parse_qsl(s, keep_blank_values=True))
query = MultiDict(self.query)
query.update(new_query)
return URL(self._val._replace(query=self._get_str_query(query)),
encoded=True)
def with_fragment(self, fragment):
"""Return a new URL with fragment replaced.
Autoencode fragment if needed.
Clear fragment to default if None is passed.
"""
# N.B. doesn't cleanup query/fragment
if fragment is None:
fragment = ''
elif not isinstance(fragment, str):
raise TypeError("Invalid fragment type")
return URL(
self._val._replace(
fragment=self._FRAGMENT_QUOTER(fragment)),
encoded=True)
def with_name(self, name):
"""Return a new URL with name (last part of path) replaced.
Query and fragment parts are cleaned up.
Name is encoded if needed.
"""
# N.B. DOES cleanup query/fragment
if not isinstance(name, str):
raise TypeError("Invalid name type")
if '/' in name:
raise ValueError("Slash in name is not allowed")
name = self._PATH_QUOTER(name)
if name in ('.', '..'):
raise ValueError(". and .. values are forbidden")
parts = list(self.raw_parts)
if self.is_absolute():
if len(parts) == 1:
parts.append(name)
else:
parts[-1] = name
parts[0] = '' # replace leading '/'
else:
parts[-1] = name
if parts[0] == '/':
parts[0] = '' # replace leading '/'
return URL(
self._val._replace(path='/'.join(parts),
query='', fragment=''), encoded=True)
def join(self, url):
"""Join URLs
Construct a full (absolute) URL by combining a base URL
(self) with another URL (url).
Informally, this uses components of the base URL, in
particular the addressing scheme, the network location and
(part of) the path, to provide missing components in the
relative URL.
"""
# See docs for urllib.parse.urljoin
if not isinstance(url, URL):
raise TypeError("url should be URL")
return URL(urljoin(str(self), str(url)), encoded=True)
def human_repr(self):
"""Return decoded human readable string for URL representation."""
return urlunsplit(SplitResult(self.scheme,
self._make_netloc(self.user,
self.password,
self.host,
self._val.port,
encode=False),
self.path,
self.query_string,
self.fragment))