PyDBML/pydbml/definitions/table.py at a7295224ac09bf400c8d79ead3c8d2d301bee5ae · Vanderhoof/PyDBML · GitHub

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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import pyparsing as pp

from pydbml.parser.blueprints import TableBlueprint
from .column import table_column, table_column_with_properties
from .common import _, _c, end, hex_color, note, note_object
from .generic import name, string_literal
from .index import indexes

pp.ParserElement.set_default_whitespace_chars(' \t\r')

alias = pp.WordStart() + pp.Literal('as').suppress() - pp.WordEnd() - name


header_color = (
    pp.CaselessLiteral('headercolor:').suppress() + _
    - pp.Combine(hex_color)('header_color')
)
table_setting = _ + (note('note') | header_color) + _
table_settings = '[' + table_setting + (',' + table_setting)[...] + ']'


def parse_table_settings(s, loc, tok):
    '''
    [headercolor: #cccccc, note: 'note']
    '''
    result = {}
    if 'note' in tok:
        result['note'] = tok['note']
    if 'header_color' in tok:
        result['header_color'] = tok['header_color']
    return result


table_settings.set_parse_action(parse_table_settings)


note_element = note | note_object

prop = name + pp.Suppress(":") + string_literal

table_element = _ + (
    table_column.set_results_name('columns', list_all_matches=True) |
    note_element('note') |
    indexes.set_results_name('indexes', list_all_matches=True)
) + _
table_element_with_property = _ + (
    table_column_with_properties.set_results_name('columns', list_all_matches=True) |
    note_element('note') |
    indexes.set_results_name('indexes', list_all_matches=True) |
    prop.set_results_name('property', list_all_matches=True)
) + _

table_body = table_element[...]
table_body_with_properties = table_element_with_property[...]

table_name = (name('schema') + '.' + name('name')) | (name('name'))

table = _c + (
    pp.CaselessLiteral("table").suppress()
    + table_name
    + alias('alias')[0, 1]
    + table_settings('settings')[0, 1] + _
    + '{' - table_body + _ + '}'
) + end

table_with_properties = _c + (
    pp.CaselessLiteral("table").suppress()
    + table_name
    + alias('alias')[0, 1]
    + table_settings('settings')[0, 1] + _
    + '{' - table_body_with_properties + _ + '}'
) + end


def parse_table(s, loc, tok):
    '''
    Table bookings as bb [headercolor: #cccccc] {
      id integer
      country varchar [NOT NULL, ref: > countries.country_name]
      booking_date date unique pk
      created_at timestamp

      indexes {
          (id, country) [pk] // composite primary key
      }
    }
    '''
    init_dict = {
        'name': tok['name'],
    }
    if 'schema' in tok:
        init_dict['schema'] = tok['schema']
    if 'settings' in tok:
        init_dict.update(tok['settings'])
    if 'alias' in tok:
        init_dict['alias'] = tok['alias'][0]
    if 'note' in tok:
        # will override one from settings
        init_dict['note'] = tok['note'][0]
    if 'indexes' in tok:
        init_dict['indexes'] = tok['indexes'][0]
    if 'columns' in tok:
        init_dict['columns'] = tok['columns']
    if 'comment_before' in tok:
        comment = '\n'.join(c[0] for c in tok['comment_before'])
        init_dict['comment'] = comment
    if 'property' in tok:
        init_dict['properties'] = {k: v for k, v in tok['property']}

    if not init_dict.get('columns'):
        raise SyntaxError(f'Table {init_dict["name"]} at position {loc} has no columns!')

    result = TableBlueprint(**init_dict)

    return result


table.set_parse_action(parse_table)
table_with_properties.set_parse_action(parse_table)