SQL2Cypher/tokens.py at master · UNSW-database/SQL2Cypher · 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
import sqlparse
from sqlparse import sql

query = "UPDATE Customers c SET c.ContactName = 'Alfred Schmidt', c.City= 'Frankfurt' WHERE c.CustomerID = 1;"

query_tokens = sqlparse.parse(query)[0].tokens


def count_tables(tokens):
    """
    count how many tables
    :param tokens:
    :return:
    """
    count = 0
    for token in tokens:
        if token.ttype is None and type(token) is sql.Identifier:
            if token.get_name() == token.get_real_name():
                raise Exception("Please use a different alias name")
            count += 1
    return count


def value_condition(tokens):
    """
    extract value condition in the tokens
    :param tokens: sql queries token
    :return:
    """
    values = " SET "
    match = "MATCH "
    where = " "
    for token in tokens:
        # print('token[%s] type[%s]' % (token, token.ttype) )
        if token.ttype is None and isinstance(token, sql.IdentifierList):
            for id in token.get_identifiers():
                if values != " SET ":
                    values += ", " + str(id)
                else:
                    values += str(id)
        elif token.ttype is None and type(token) is sql.Identifier:
            # tables
            match += "({}:{}) ".format(str(token.get_name()), str(token.get_real_name()))
        elif token.ttype is None and type(token) is sql.Where:
            # where condition
            where += str(token).replace(";", "")
    return match + where + values


def parse(tokens):
    if count_tables(tokens) == 1:
        query = value_condition(tokens)
        print(query)
    else:
        raise Exception("Can not parse relationships now")


if __name__ == '__main__':
    parse(query_tokens)

# for token in query_tokens:
#     print('token[%s] type[%s] id: [%s]' % (token, token.ttype, type(token)))
#
# # print(sql.IdentifierList.value)
# for token in query_tokens:
#     # print('token[%s] type[%s]' % (token, token.ttype) )
#     if token.ttype is None and isinstance(token, sql.IdentifierList):
#         print('Identifierlist:')
#         for id in token.get_identifiers():
#             print(id)
#
#
# for token in query_tokens:
#     # print('token[%s] type[%s]' % (token, token.ttype) )
#     if token.ttype is None and type(token) is sql.Identifier:
#         print('Table:')
#         print(token.get_name(), token.get_real_name())
#
#
# for token in query_tokens:
#     # print('token[%s] type[%s]' % (token, token.ttype) )
#
#     if token.ttype is None and type(token) is sql.Where:
#         print('Where:')
#         print(token)