一日一技:python连接mysql数据库（如何用python连接数据库）

off999 2024-10-12 06:15 50 浏览 0 评论

安装PyMySQL模块

$ sudo pip3 install PyMySQL  #Windows用户直接pip3 install PyMySQL，本节用的是UBUNTU系统。

连接举例如下，新建一个version.py文件。写入：

#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'mydb')

with con:
    
    cur = con.cursor()
    cur.execute("SELECT VERSION()")

    version = cur.fetchone()
    
    print("Database version: {}".format(version[0]))

在MySQL中，我们可以使用SELECT VERSION（）来获取MySQL的版本。

输出结果为：

Database version: 5.7.23-0ubuntu0.16.04.1 #输出显示为mysql5.7.23版本

PyMySQL模块中的 fetchAll方法

fetchAll（）方法检索查询结果的所有（剩余）行，并将它们作为序列序列返回，新建一个retrieve_all.py文件举例说明：

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'testdb')

with con: 

    cur = con.cursor()
    cur.execute("SELECT * FROM cities")

    rows = cur.fetchall()

    for row in rows:
        print("{0} {1} {2}".format(row[0], row[1], row[2]))

输出结果为：

1 Bratislava 432000
2 Budapest 1759000
3 Prague 1280000
4 Warsaw 1748000
5 Los Angeles 3971000
6 New York 8550000
7 Edinburgh 464000
8 Berlin 3671000

PyMySQL模块中的字典光标

默认游标以元组的元组返回数据。当我们使用字典游标时，数据以Python字典的形式发送。这样，我们可以通过列名称来引用数据。

新建一个dictionary_cursor.py的文件：

import pymysql
import pymysql.cursors

con = pymysql.connect(host='localhost',
        user='user17',
        password='s$cret',
        db='mydb',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor)

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cities")

    rows = cur.fetchall()

    for row in rows:
        print(row["id"], row["name"])

PyMySQL列的标题

接下来，我们将展示如何使用数据库表中的数据打印列标题。

import  pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'testdb')

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cities")

    rows = cur.fetchall()

    desc = cur.description

    print("{0:>3} {1:>10}".format(desc[0][0], desc[1][0]))

    for row in rows:    
        print("{0:3} {1:>10}".format(row[0], row[2]))

输出结果如下：

 id       name
  1     432000
  2    1759000
  3    1280000
  4    1748000
  5    3971000
  6    8550000
  7     464000
  8    3671000

PyMySQL准备的声明

在编写预备语句时，我们使用占位符，而不是直接将值写入语句中。准备好的语句可提高安全性和性能。

新建一个prepared_statement.py文件，写入：

import pymysql

con = pymysql.connect('localhost', 'user17', 
    's$cret', 'testdb')

# user input
myid = 4

with con:    

    cur = con.cursor()
        
    cur.execute("SELECT * FROM cities WHERE id=%s", myid) 
    
    cid, name, population  = cur.fetchone()
    print(cid, name, population)

运行结果为：

4 Warsaw 1748000

PyMySQL受影响的行

rowcount是只读游标属性，它指定最后一个SELECT，UPDATE或INSERT语句产生的行数。

import pymysql

con = pymysql.connect('localhost', 'user17',
   's$cret', 'mydb')

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cities WHERE id IN (1, 2, 3)")

    # rows = cur.fetchall()

    # for row in rows:
    #     print("{0} {1} {2}".format(row[0], row[1], row[2]))

    print("The query affected {} rows".format(cur.rowcount))

运行结果如下：