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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
from __future__ import annotations
from collections.abc import Sequence
import functools
import re
DECODE_DEFAULT_CHARS = ";/?:@&=+$,#"
DECODE_COMPONENT_CHARS = ""
decode_cache: dict[str, list[str]] = {}
def get_decode_cache(exclude: str) -> Sequence[str]:
if exclude in decode_cache:
return decode_cache[exclude]
cache: list[str] = []
decode_cache[exclude] = cache
for i in range(128):
ch = chr(i)
cache.append(ch)
for i in range(len(exclude)):
ch_code = ord(exclude[i])
cache[ch_code] = "%" + ("0" + hex(ch_code)[2:].upper())[-2:]
return cache
# Decode percent-encoded string.
#
def decode(string: str, exclude: str = DECODE_DEFAULT_CHARS) -> str:
cache = get_decode_cache(exclude)
repl_func = functools.partial(repl_func_with_cache, cache=cache)
return re.sub(r"(%[a-f0-9]{2})+", repl_func, string, flags=re.IGNORECASE)
def repl_func_with_cache(match: re.Match, cache: Sequence[str]) -> str:
seq = match.group()
result = ""
i = 0
l = len(seq) # noqa: E741
while i < l:
b1 = int(seq[i + 1 : i + 3], 16)
if b1 < 0x80:
result += cache[b1]
i += 3 # emulate JS for loop statement3
continue
if (b1 & 0xE0) == 0xC0 and (i + 3 < l):
# 110xxxxx 10xxxxxx
b2 = int(seq[i + 4 : i + 6], 16)
if (b2 & 0xC0) == 0x80:
all_bytes = bytes((b1, b2))
try:
result += all_bytes.decode()
except UnicodeDecodeError:
result += "\ufffd" * 2
i += 3
i += 3 # emulate JS for loop statement3
continue
if (b1 & 0xF0) == 0xE0 and (i + 6 < l):
# 1110xxxx 10xxxxxx 10xxxxxx
b2 = int(seq[i + 4 : i + 6], 16)
b3 = int(seq[i + 7 : i + 9], 16)
if (b2 & 0xC0) == 0x80 and (b3 & 0xC0) == 0x80:
all_bytes = bytes((b1, b2, b3))
try:
result += all_bytes.decode()
except UnicodeDecodeError:
result += "\ufffd" * 3
i += 6
i += 3 # emulate JS for loop statement3
continue
if (b1 & 0xF8) == 0xF0 and (i + 9 < l):
# 111110xx 10xxxxxx 10xxxxxx 10xxxxxx
b2 = int(seq[i + 4 : i + 6], 16)
b3 = int(seq[i + 7 : i + 9], 16)
b4 = int(seq[i + 10 : i + 12], 16)
if (b2 & 0xC0) == 0x80 and (b3 & 0xC0) == 0x80 and (b4 & 0xC0) == 0x80:
all_bytes = bytes((b1, b2, b3, b4))
try:
result += all_bytes.decode()
except UnicodeDecodeError:
result += "\ufffd" * 4
i += 9
i += 3 # emulate JS for loop statement3
continue
result += "\ufffd"
i += 3 # emulate JS for loop statement3
return result
|