【python】python连接mysql数据库（pymysql）

旧城等待， 2022-11-28 13:30 8阅读 0赞

#### 1.安装pymysql ####

pip3 install mysql

#### 2.使用 ####

##### 2.1 创建连接 #####

我们导入pymysql包之后，使用connect()方法获取一个连接（Connection），该方法可以传入30多个参数，我们只要传入主要的几个就行

<table> 
 <thead> 
  <tr> 
   <th>参数</th> 
   <th>说明</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>host</td> 
   <td>主机IP地址</td> 
  </tr> 
  <tr> 
   <td>port</td> 
   <td>连接的端口</td> 
  </tr> 
  <tr> 
   <td>user</td> 
   <td>登录用户</td> 
  </tr> 
  <tr> 
   <td>password</td> 
   <td>登录密码</td> 
  </tr> 
  <tr> 
   <td>database</td> 
   <td>数据库名</td> 
  </tr> 
  <tr> 
   <td>charset</td> 
   <td>编码(防止乱码)</td> 
  </tr> 
 </tbody> 
</table>

# 导入pymysql
    import pymysql
    
    # 连接数据库
    conn = pymysql.connect(host='localhost', port=3306, user='root', password='mysql',database='python', charset='utf8')

##### 2.2 执行MySQL语句 #####

我们在连接数据库成功之后，可以获取到一个游标cursor，我们通过cursor的execute()方法执行SQL语句

# 获取游标对象
    cursor = conn.cursor()
    
    # 查询 SQL 语句
    sql = "select * from students;"
    
    # 执行 SQL 语句，返回执行后受影响的行数
    row_nums = cursor.execute(sql)

##### 2.3 查询数据（DQL） #####

如果执行的是查询类的操作，那我们可以取出查询到的数据

cursor对象有fetchone()和fetchall()两种方法取出数据，fetchone()是取出一个，fetchall()是取出全部，两个方法得到的都是元组

for data in cursor.fetchall():
        print(data)

##### 2.4 操作数据（DML） #####

需要注意的是，pymysql是默认开启事务的，所以我们在操作数据的时候，可以选择提交(commit)或回滚(rollback)

try:
        sql = "update students set name = '冰冰' where id = 1;"
        row_nums = cursor.execute(sql)
        id row_nums > 0 :
        	print('操作成功')
        # 提交数据到数据库
        conn.commit()
    except Exception as e:
        # 回滚数据， 即撤销刚刚的SQL语句操作
        conn.rollback()

##### 2.5 关闭游标和链接 #####

业务结束之后，我们需要关闭游标和连接，释放资源

# 关闭游标
    cursor.close()
    
    # 关闭连接
    conn.close()

#### 3.防注入问题 ####

如果在需要执行查询语句传参的时候，我们尽量不要自己拼接字符串，而是交给cursor.execute()，这样可以避免SQL注入的问题，防止信息泄露  
cursor.execute()可以传递两个参数，一个数SQL语句，另一个是参数列表

find_str = input("请输入要查找的商品：")
     
     # 构建一个列表，把参数丢进列表
     params = [find_str]
     count = cs1.execute("select * from goods where name=%s", params)

#### 4.实例（插入数据） ####

from datetime import datetime
    
    import pymysql
    
    
    def start():
        conn = pymysql.connect(host='localhost', port=3306, user='root', password='888888', database='test_db', charset='utf8')
        now_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        cursor = conn.cursor()
        team_list = [
            ("默认销售团队", 101, now_str, now_str, "T", "T"),
            ("默认服务团队", 102, now_str, now_str, "T", "T"),
            ......
        ]
        sql_insert = "insert into team(team_name,team_leader_id,create_time,update_time,is_default,is_active) values(%s,%s,%s,%s,%s,%s)"  # 省略主键id
        try:
            cursor.execute("truncate team;")  # 删除原有数据
            cursor.executemany(sql_insert, team_list)
            conn.commit()
        except Exception as e:
            conn.rollback()
            print(e)
        finally:
            cursor.close()
            conn.close()
    
    
    if __name__ == '__main__':
        start()