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
|
def get_inner_spec(column_name, spec):
brackets = 0
offset = len(column_name)
i = offset
for i, ch in enumerate(spec[offset:], offset):
if ch == '(':
brackets += 1
elif ch == ')':
brackets -= 1
if brackets == 0:
break
return spec[offset + 1:i]
def get_inner_columns(spec):
brackets = 0
column_begin = 0
columns = []
for i, x in enumerate(spec + ','):
if x == ',':
if brackets == 0:
columns.append(spec[column_begin:i])
column_begin = i + 1
elif x == '(':
brackets += 1
elif x == ')':
brackets -= 1
elif x == ' ':
if brackets == 0:
column_begin = i + 1
return columns
def get_inner_columns_with_types(spec):
brackets = 0
prev_comma = 0
prev_space = 0
columns = []
for i, x in enumerate(spec + ','):
if x == ',':
if brackets == 0:
columns.append((
spec[prev_comma:prev_space].strip(),
spec[prev_space:i]
))
prev_comma = i + 1
elif x == '(':
brackets += 1
elif x == ')':
brackets -= 1
elif x == ' ':
if brackets == 0:
prev_space = i + 1
return columns
|
