如何从 Python 中解析 MySQL 内置代码？

Question

I've exported my MySQL database to a .txt/.sql file (link) 并想用它在其他计算机上重建数据库。虽然在 HeidiSQL 中运行脚本工作得非常好，但我想在启动我的应用程序之前确保所有表都存在并且正确。

我可以将构建的代码复制粘贴到我的 Python 代码中，但导出是我的数据库的直接表示，不应该包含任何错误，而复制粘贴...所以我尝试解析文件，但我的解析技巧还有很多不足之处。

这是代码的sn-p：

CREATE TABLE IF NOT EXISTS `data` (
  `dataid` int(10) unsigned NOT NULL AUTO_INCREMENT,
  `measurementid` int(10) unsigned NOT NULL DEFAULT '0',
  `frame` int(10) unsigned NOT NULL,
  `sensor_row` int(10) unsigned NOT NULL,
  `sensor_col` int(10) unsigned NOT NULL,
  `value` float unsigned NOT NULL,
  PRIMARY KEY (`dataid`),
  UNIQUE KEY `measurementid_frame_sensor_row_sensor_col` (`measurementid`,`frame`,`sensor_row`,`sensor_col`)
) ENGINE=MyISAM DEFAULT CHARSET=latin1;

所以假设数据库data 已经创建，我正在寻找一种方法来解析构建的代码并创建所有必需的表（如果它们不存在或更改） 如果他们改变了。

关于如何解决这个问题的任何建议？

Answer 1

您可以直接通过 python 数据库适配器从文件中完全执行 sql：

import MySQLdb
sql_dump = open('builtcode.sql', 'r').read()

conn = MySQLdb.connect(db='soans', passwd='*****')
cur = conn.cursor()

cur.execute(sql_dump)

cur.connection.close()

给予：

mysql> show tables;
+-----------------+
| Tables_in_soans |
+-----------------+
| anamnesis_main  |
| averagecontact  |

Answer 2

只需读取文件并执行 sql 字符串

我对您还需要做什么感到困惑。如果表已经存在，则创建表语句将不会运行。任何 alter 语句都会运行。

如果您需要将当前架构与您尝试将其更新到的架构进行比较，这比“解析”文件所暗示的更难，您可能希望使用某种迁移框架（南for django 是我的最爱之一）或模式比较工具。

Answer 3

您可以尝试sqlparse module 将文件拆分为单独的语句，然后执行每个语句。

您还可以通过高级方式操作 SQL 令牌，例如 stripping out all comments (except those inside functions)。您可以使用此技术从 CREATE TABLE 语句中解析表名并检查表是否存在。

以下是拆分语句的示例：

import sqlparse

queries = '''
CREATE TABLE foo (id INTEGER);
CREATE TABLE bar (id INTEGER, name VARCHAR);
SELECT foo FROM bar WHERE id IN (1,2,3);
DELETE FROM bar WHERE id IN (3, 4);
-- a comment
DELETE FROM bar WHERE name IN ('chaos','atlas');
'''

for i, stmt, in enumerate(sqlparse.split(queries)):
    sql = stmt.strip()
    if not sql:
        continue
    print i, sql

输出：

0 CREATE TABLE foo (id INTEGER);
1 CREATE TABLE bar (id INTEGER, name VARCHAR);
2 SELECT foo FROM bar WHERE id IN (1,2,3);
3 DELETE FROM bar WHERE id IN (3, 4);
4 -- a comment
DELETE FROM bar WHERE name IN ('chaos','atlas');