forked from astropy/astropy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathunescaper.py
More file actions
60 lines (45 loc) · 1.49 KB
/
Copy pathunescaper.py
File metadata and controls
60 lines (45 loc) · 1.49 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# Licensed under a 3-clause BSD style license - see LICENSE.rst
"""URL unescaper functions."""
from __future__ import (absolute_import, division, print_function,
unicode_literals)
# STDLIB
from xml.sax import saxutils
__all__ = ['unescape_all']
# This is DIY
_bytes_entities = {b'&': b'&', b'<': b'<', b'>': b'>',
b'&&': b'&', b'&&': b'&', b'%2F': b'/'}
_bytes_keys = [b'&&', b'&&', b'&', b'<', b'>', b'%2F']
# This is used by saxutils
_str_entities = {'&&': '&', '&&': '&', '%2F': '/'}
_str_keys = ['&&', '&&', '&', '<', '>', '%2F']
def unescape_all(url):
"""Recursively unescape a given URL.
.. note:: '&&' becomes a single '&'.
Parameters
----------
url : str or bytes
URL to unescape.
Returns
-------
clean_url : str or bytes
Unescaped URL.
"""
if isinstance(url, bytes):
func2use = _unescape_bytes
keys2use = _bytes_keys
else:
func2use = _unescape_str
keys2use = _str_keys
clean_url = func2use(url)
not_done = [clean_url.count(key) > 0 for key in keys2use]
if True in not_done:
return unescape_all(clean_url)
else:
return clean_url
def _unescape_str(url):
return saxutils.unescape(url, _str_entities)
def _unescape_bytes(url):
clean_url = url
for key in _bytes_keys:
clean_url = clean_url.replace(key, _bytes_entities[key])
return clean_url