• 技术文章 >常见问题 >Python常见问题

    Python怎么从sql提取表名

    月亮邮递员月亮邮递员2020-03-25 16:40:06原创3652
    本篇文章介绍了Python使用ply和re模块提取sql语句中表名的方法,具有一定的参考价值,希望对各位学习Python的朋友有帮助!

    Python怎么从sql提取表名

    ply:词法语法解析模块;

    re:正则表达式模块。

    全部代码如下:

    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

    import ply.lex as lex, re

     

     

    def extract_table_name_from_sql(sql_str):

     

        # remove the /* */ comments

        q = re.sub(r"/\*[^*]*\*+(?:[^*/][^*]*\*+)*/", "", sql_str)

     

        # remove whole line -- and # comments

        lines = [line for line in q.splitlines() if not re.match("^\s*(--|#)", line)]

     

        # remove trailing -- and # comments

        q = " ".join([re.split("--|#", line)[0] for line in lines])

     

        # split on blanks, parens and semicolons

        tokens = re.split(r"[\s)(;]+", q)

     

        # scan the tokens. if we see a FROM or JOIN, we set the get_next

        # flag, and grab the next one (unless it's SELECT).

     

        result = []

        get_next = False

        for token in tokens:

            if get_next:

                if token.lower() not in ["", "select"]:

                    result.append(token)

                get_next = False

            get_next = token.lower() in ["from", "join"]

     

        return result

     

    sql2="SELECT a.time_updated_server/1000,content,nick,name FROM      " \

         "table1 a JOIN   " \

         "table2 b ON a.sender_id = b.user_id JOIN table3 c ON a.channel_id = c.channel_id JOIN table4 d ON c.store_id = d.store_id WHERE sender_id NOT IN(SELECT user_id FROM table5 WHERE store_id IN ('agent_store:1', 'ask:1')) AND to_timestamp(a.time_updated_server/1000)::date >= '2014-05-01' GROUP BY 1,2,3,4 HAVING sum(1) > 500 ORDER BY 1 ASC"

    print(extract_table_name_from_sql(sql2))

    更多技术请关注Python视频教程

    专题推荐:python sql 提取表名
    上一篇:Python怎么连接oracle数据库 下一篇:Python怎么查看函数库

    相关文章推荐

    • VB怎么连接PostgreSQL• SQL文件怎么导入MySQL里• Python怎么运行shell脚本• Python怎么连接oracle数据库

    全部评论我要评论

    © 2021 Python学习网 苏ICP备2021003149号-1

  • 取消发布评论
  • 

    Python学习网