在python中只需要regex表达式来获取表名和主键

Question

也许这个问题很愚蠢，但我无法生成regex表达式来获取表名和主键。你知道吗

表格：

CREATE TABLE 'dhcpr_dhcprelayinterface'  (
        'vrId' integer default 0,
        'ifName' string ,
        PRIMARY KEY(ifName,vrId),
        FOREIGN KEY (vrId) REFERENCES 'vr_vr'(vrId) ON DELETE CASCADE ON UPDATE CASCADE);

我正在使用： begin = re.compile(r"CREATE TABLE[ \"]*([^ \"]+)[ \"]*[(]([^/;]+)[/;]",re.IGNORECASE)获取所有表名和数据。你知道吗

但我只需要带有表名和主键的数据。你知道吗

预期输出：

dhcpr_dhcprelayinterface
PRIMARY KEY(ifName,vrId)

Answer 1

我相信你可以用正则表达式或者^{}来解决这个问题，但是这里有一个“有趣”的方法来解决这个问题仅仅是为了教育目的-在内存数据库中使用^{}-实际创建表并从^{} internal table中获取table_name，从^{}中获取主键列：

import sqlite3

query = """
CREATE TABLE 'dhcpr_dhcprelayinterface'  (
        'vrId' integer default 0,
        'ifName' string ,
        PRIMARY KEY(ifName,vrId),
        FOREIGN KEY (vrId) REFERENCES 'vr_vr'(vrId) ON DELETE CASCADE ON UPDATE CASCADE);
"""

db = sqlite3.connect(":memory:")
cursor = db.cursor()

cursor.execute(query)
db.commit()

# get table name
cursor.execute("select name from sqlite_master where type = 'table'")
table_name = cursor.fetchone()[0]
print(table_name)

# get primary key columns
cursor.execute("PRAGMA table_info(%s);" % table_name)
pk_columns = [row[1] for row in cursor.fetchall()[::-1]]
print(pk_columns)

印刷品：

dhcpr_dhcprelayinterface
['ifName', 'vrId']