У меня есть вопрос об извлечении имен таблиц из Postgresql - PullRequest
Новая
       20

У меня есть вопрос об извлечении имен таблиц из Postgresql

0 голосов
/ 29 ноября 2018

Я хочу спросить об извлечении имен таблиц из запросов Postgresql с использованием Python 3.X.

Вот мои условия.

1.Python 3.X пакеты sqlparse 2.Postgresql

Это мой SQL-код. 

$postgresql

With 
a as (
  select x,y
  from g
)

select x, y,z
from      b
left join a  on b.id = a.id

Iхочу получить результат из вышеприведенного запроса с использованием sqlparse от Python3.X. 

#table name : g, b

Как мне улучшить мой следующий код? 

$Python3

import sqlparse
from sqlparse.sql import Where, Comparison, Parenthesis, Identifier


class RecursiveTokenParser(object):
 def __init__(self, query):
    self.query = query
    self.names = []

 def get_table_names(self):
     elements = sqlparse.parse(self.query)

     for token in elements[0].tokens:

         if isinstance(token, Identifier):
             self.identifier(token)
         elif isinstance(token, Parenthesis):
             self.parenthesis(token)

         elif isinstance(token, Where):
             self.where(token)

     return [str(name) for name in self.names]

 def where(self, token):

    for subtoken in token.tokens:
        if isinstance(subtoken, Comparison):
           self.comparison(subtoken)

 def comparison(self, token):
    for subtoken in token.tokens:
        if isinstance(subtoken, Parenthesis):
           self.parenthesis(subtoken)

 def parenthesis(self, token):

    for subtoken in token.tokens:
        if isinstance(subtoken, Identifier):
           self.identifier(subtoken)
        elif isinstance(subtoken, Parenthesis):
           self.parenthesis(subtoken)

 def identifier(self, token):
     self.names.append(token)

 def get_query(self):  
    return self.query


sql = """

     WITH a as
     (
      SELECT 
        id, x, y
      FROM d
     )

     SELECT 
         x,y, z
     FROM e 
     left join a c
     ON a.id = e.id 
     """
t = RecursiveTokenParser(sql)

print(t.get_query())
print(t.get_table_names())

result: ['a as \ n (\n SELECT \ n id, x, y \ n ОТ d \ n) ',' e ',' a c ']

...