Есть ли функция для анализа полного запроса SQL в Python? - PullRequest
Новая
       42

Есть ли функция для анализа полного запроса SQL в Python?

1 голос
/ 02 ноября 2019

Я использую запросы postgreSQL. Я хочу извлечь всю информацию из запроса SQL, например, 

sql = " select d_year, s_nation, p_category, sum(lo_revenue - lo_supplycost) as profit from DATES, CUSTOMER, SUPPLIER, PART, LINEORDER where lo_custkey  =  c_custkey and lo_suppkey  =  s_suppkey and lo_partkey  =  p_partkey and lo_orderdate  =  d_datekey and c_region  =  'AFRICA' and s_region  =  'AFRICA' and (d_year  =  1996 or d_year  =  1997) and (p_mfgr  =  'MFGR#2' or p_mfgr  =  'MFGR#4') group by d_year, s_nation, p_category order by d_year, s_nation, p_category "

Я хочу получить все соответствующие таблицы, все предикаты выбора и все предикаты соединения, группировать по частям и упорядочивать по частям.

Я использовал sqlparse и нашел способ получить только соответствующие таблицы. Есть ли примеры, как извлечь эту информацию?

1 Ответ

0 голосов
/ 04 ноября 2019

Этот алгоритм дает точный элемент между каждым ключевым словом. Я использовал sqlparse

       parsed = sqlparse.parse(sql)
       stmt = parsed[0]
       from_seen = False
       select_seen = False
       where_seen = False
       groupby_seen = False
       orderby_seen = False

       for token in stmt.tokens:
           if select_seen:
               if isinstance(token, IdentifierList):
                   for identifier in token.get_identifiers():
                       print("{} {}\n".format("Attr = ", identifier) )
               elif isinstance(token, Identifier):
                   print("{} {}\n".format("Attr = ", token))
           if from_seen:
               if isinstance(token, IdentifierList):
                   for identifier in token.get_identifiers():
                       print("{} {}\n".format("TAB = ", identifier) )
               elif isinstance(token, Identifier):
                   print("{} {}\n".format("TAB = ", token))
           if orderby_seen:
               if isinstance(token, IdentifierList):
                   for identifier in token.get_identifiers():
                       print("{} {}\n".format("ORDERBY att = ", identifier) )
               elif isinstance(token, Identifier):
                   print("{} {}\n".format("ORDERBY att = ", token))
           if groupby_seen:
               if isinstance(token, IdentifierList):
                   for identifier in token.get_identifiers():
                       print("{} {}\n".format("GROUPBY att = ", identifier) )
               elif isinstance(token, Identifier):
                   print("{} {}\n".format("GROUPBY att = ", token))

           if isinstance(token, Where):
               select_seen = False
               from_seen = False
               where_seen = True
               groupby_seen = False
               orderby_seen = False
               for where_tokens in token:
                   if isinstance(where_tokens, Comparison):
                       print("{} {}\n".format("Comparaison = ", where_tokens))
                   elif isinstance(where_tokens, Parenthesis):
                       print("{} {}\n".format("Parenthesis = ", where_tokens))
                       # tables.append(token)
           if token.ttype is Keyword and token.value.upper() == "GROUP BY":
               select_seen = False
               from_seen = False
               where_seen = False
               groupby_seen = True
               orderby_seen = False
           if token.ttype is Keyword and token.value.upper() == "ORDER BY":
               select_seen = False
               from_seen = False
               where_seen = False
               groupby_seen = False
               orderby_seen = True
           if token.ttype is Keyword and token.value.upper() == "FROM":
               select_seen = False
               from_seen = True
               where_seen = False
               groupby_seen = False
               orderby_seen = False
           if token.ttype is DML and token.value.upper() == "SELECT":
               select_seen = True
               from_seen = False
               where_seen = False
               groupby_seen = False
               orderby_seen = False
...