Python连接MySQL之Python库pymysql

向右看齐 2023-10-09 11:05 68阅读 0赞

#  连接数据库 #

pymysql连接数据库的方式和使用sqlite的方式基本相同：

*  使用connect创建连接对象
 *  connect.cursor创建游标对象，SQL语句的执行基本都在游标上进行
 *  cursor.executeXXX方法执行SQL语句，cursor.fetchXXX获取查询结果等
 *  调用close方法关闭游标cursor和数据库连接

<table style="width:602px;"> 
 <tbody> 
  <tr> 
   <td> <pre><code><span>import pymssql <span># server 数据库服务器名称或IP <span># user 用户名 <span># password 密码 <span># database 数据库名称 conn = pymssql.connect(server, user, password, database) cursor = conn.cursor()</span></span></span></span></span></code></pre> </td> 
  </tr> 
 </tbody> 
</table>

新建、插入操作

cursor.execute("""
    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)
    )
    """)
    cursor.executemany(
        "INSERT INFO persons VALUES (%d,%s,%s)",
        [
            (1, 'John Smith', 'John Doe'),
            (2, 'Jane Doe', 'Joe Dog'),
            (3, 'Mike T.', 'Sarah H.'),
        ]
    )
    # 如果没有指定autocommit属性为True的话就需要调用commit()方法
    conn.commit()

查询操作

cursor.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')
    row = cursor.fetchone()
    while row:
        print('ID=%d,Name=%s' % (row[0], row[1]))
        row = cursor.fetchone()
    # 也可以使用for循环来迭代查询结果
    # for row in cursor:
    #     print("ID=%d,Name=%s" % (row[0], row[1]))
    # 关闭连接
    conn.close()

# 游标使用注意事项 #

一个连接一次只能有一个游标的查询处于活跃状态，如下：

<table> 
 <tbody> 
  <tr> 
   <td> <pre>c1 = conn.cursor()<br>c1.execute('SELECT * FROM persons')<br>c2 = conn.cursor()<br>c2.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')<br>print("all person")<br>print(c1.fetchall())  # 显示出的是c2游标查询出来的结果<br>print("John Doe")<br>print(c2.fetchall())  # 不会有任何显示</pre> </td> 
  </tr> 
 </tbody> 
</table>

为了避免上述的问题可以使用以下两种方式：

*  创建多个连接来保证多个查询可以并行执行在不同连接的游标上
 *  使用fetchall方法获取到游标查询结果之后再执行下一个查询，如下：

c1.execute('SELECT ...')
    c1_list = c1.fetchall()
    c2.execute('SELECT ...')
    c2_list = c2.fetchall()

# 游标返回行为字典变量 #

上述例子中游标获取的查询结果每一行为元祖类型，可以通过在创建游标时指定as\_dict参数来使游标返回字典变量。

字典中的键为数据表的列名

import pymysql
    
    conn = pymysql.connect(server, user, password, database)
    cursor = conn.cursor(as_dict=True)
    cursor.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')
    for row in cursor:
        print("ID=%d,Name=%s" % (row['id'], row['name']))
    conn.close()

# 使用with语句（上下文管理器） #

可以通过使用with语句来省去显示的调用close方法关闭连接和游标

<table> 
 <tbody> 
  <tr> 
   <td> <pre>with pymysql.connect(server,user,password,database) as conn:<br>    with conn.cursor(as_dict=True) as cursor:<br>        cursor.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')<br>        for row in cursor:<br>            print("ID=%d,Name=%s" % (row['id'], row['name']))</pre> </td> 
  </tr> 
 </tbody> 
</table>

# 调用存储过程 #

pymysql 2.0.0以上的版本可以通过cursor.callproc方法来调用存储过程

with pymssql.connect(server, user, password, database) as conn:
        with conn.cursor(as_dict=True) as cursor:
            # 创建存储过程
            cursor.execute("""
            CREATE PROCEDURE FindPerson
                @name VARCHAR(100)
            AS BEGIN
                SELECT * FROM persons WHERE name = @name
            END
            """)
    
            # 调用存储过程
            cursor.callproc('FindPerson', ('Jane Doe',))
            for row in cursor:
                print("ID=%d, Name=%s" % (row['id'], row['name']))

转载于:https://www.cnblogs.com/Black-rainbow/p/9597991.html