blob: de28ac00076aa22016f628a1d8e181b3e5bc1aac (
plain)
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
|
# the generic time format is based on python time.strftime
# https://docs.python.org/3/library/time.html#time.strftime
from sqlglot.trie import in_trie, new_trie
def format_time(string, mapping, trie=None):
"""
Converts a time string given a mapping.
Examples:
>>> format_time("%Y", {"%Y": "YYYY"})
'YYYY'
mapping: Dictionary of time format to target time format
trie: Optional trie, can be passed in for performance
"""
if not string:
return None
start = 0
end = 1
size = len(string)
trie = trie or new_trie(mapping)
current = trie
chunks = []
sym = None
while end <= size:
chars = string[start:end]
result, current = in_trie(current, chars[-1])
if result == 0:
if sym:
end -= 1
chars = sym
sym = None
start += len(chars)
chunks.append(chars)
current = trie
elif result == 2:
sym = chars
end += 1
if result and end > size:
chunks.append(chars)
return "".join(mapping.get(chars, chars) for chars in chunks)
|