7

438849f1 · 王晓庆 · 436b6994 · 438849f1 · 438849f1 · 438849f1
--- a/createdb.sql
+++ b/createdb.sql
-CREATE DATABASE wxq2
-
 USE wxq2

-SELECT Name from sys.Databases
+-- 创建persons表
+IF OBJECT_ID('persons', 'U') IS NOT NULL
+  DROP TABLE persons
+CREATE TABLE persons (
+  id INT NOT NULL,
+  name varchar(100),
+  salesrep varchar(100),
+  PRIMARY KEY(id)
+)
--- a/query.sql
+++ b/query.sql
+USE wxq2
+
+SELECT * FROM persons WHERE salesrep='Joe Dog'
--- a/seed.sql
+++ b/seed.sql
+USE wxq2
+
+-- # 插入三条测试数据
+INSERT INTO persons VALUES ('1', 'John Smith', 'John Doe')
+INSERT INTO persons VALUES ('2', 'Jane Doe', 'Joe Dog')
+INSERT INTO persons VALUES ('3', 'Mike T.', 'Sarah H.')
--- a/testdb.py
+++ b/testdb.py
@@ -4,32 +4,46 @@ import pymssql
 # 查询时需要用到
 import pandas as pd

-#建立连接并获取cursor
-conn = pymssql.connect(host='localhost', user='SA', password='<MyStrong@Passw0rd>')
-cursor = conn.cursor()
-
-# 创建数据库不一致
-#conn.autocommit(True)
-#cursor.execute("CREATE DATABASE wxq2")
-#conn.autocommit(False)
-
-# 读取 sql 文件文本内容
-sql = open('createdb.sql', 'r', encoding = 'utf8')
-sqltxt = sql.readlines()
-# 此时 sqltxt 为 list 类型
+# 将sql文件的内容转换成字符串
+def parse_sql(filename):
+    # 读取 sql 文件文本内容，sqltxt 为 list 类型
+    sql = open(filename, 'r', encoding = 'utf8')
+    sqltxt = sql.readlines()
 
-# 读取之后关闭文件
-sql.close()
+    # 读取之后关闭文件
+    sql.close()

-# list 转 str
-sql = "".join(sqltxt)
+    # list 转 str
+    sql = "".join(sqltxt)

-# 查询
-conn.autocommit(True)
-df = pd.read_sql(sql, conn)
-conn.autocommit(False)
+    return sql

-print(df)
+# 主函数
+if __name__ == '__main__':

-# 连接用完后记得关闭以释放资源
-conn.close()
+    #建立连接并获取cursor
+    conn = pymssql.connect(host='localhost', user='SA', password='<MyStrong@Passw0rd>')
+    cursor = conn.cursor()
+
+    # 创建测试数据库
+    conn.autocommit(True)
+    cursor.execute("CREATE DATABASE wxq2")
+    conn.autocommit(False)
+
+    # 创建数据库表
+    createdb_sql = parse_sql('createdb.sql')
+    pd.read_sql(createdb_sql, conn)
+
+    # 填充数据
+    seed_sql = parse_sql('seed.sql')
+    pd.read_sql(createdb_sql, conn)
+
+    # 运行学生代码
+    query_sql = parse_sql('query.sql')
+    df = pd.read_sql(query_sql, conn)
+
+    # 输出查询结果
+    print(df)
+
+    # 连接用完后记得关闭以释放资源
+    conn.close()