python数据库如何连接数据库连接

Python数据库如何连接数据库

Python连接数据库的方法有多种，主要包括：使用Python标准库sqlite3、使用第三方库如SQLAlchemy、PyMySQL、psycopg2等。其中，使用第三方库SQLAlchemy是最推荐的，因为它提供了ORM（对象关系映射）功能，能让开发者更方便地进行数据库操作。接下来，我们将详细介绍SQLAlchemy的使用方法。

一、使用SQLAlchemy连接数据库

SQLAlchemy是一个Python SQL工具包和对象关系映射（ORM）库。它提供了完整的SQL功能，并且支持多种数据库后端，如SQLite、MySQL、PostgreSQL等。

1、安装SQLAlchemy

在开始使用SQLAlchemy之前，你需要确保已安装SQLAlchemy库。可以通过以下命令进行安装：

pip install sqlalchemy

2、创建数据库连接

使用SQLAlchemy连接数据库的第一步是创建一个引擎（Engine）。引擎是与数据库进行通信的核心对象。下面是一个连接MySQL数据库的例子：

from sqlalchemy import create_engine
创建数据库连接
engine = create_engine('mysql+pymysql://username:password@hostname/database')

在上述代码中，mysql+pymysql表示使用PyMySQL驱动程序连接MySQL数据库，username、password、hostname和database分别是数据库的用户名、密码、主机名和数据库名。

3、创建会话（Session）

会话是与数据库进行交互的核心对象。它负责管理对象的持久化状态，并将对象的变化同步到数据库中。

from sqlalchemy.orm import sessionmaker
创建会话类
Session = sessionmaker(bind=engine)
创建会话对象
session = Session()

4、定义模型（Model）

模型是数据库表的抽象表示。在SQLAlchemy中，模型类需要继承declarative_base类，并定义表的字段。

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String
创建基础类
Base = declarative_base()
定义模型类
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    age = Column(Integer)

5、创建表

使用模型类创建数据库表。

# 创建所有表
Base.metadata.create_all(engine)

6、增删改查操作

插入数据

# 创建新用户对象
new_user = User(name='John', age=30)
添加到会话
session.add(new_user)
提交会话
session.commit()

查询数据

# 查询所有用户
users = session.query(User).all()
for user in users:
    print(user.name, user.age)

更新数据

# 查询要更新的用户
user = session.query(User).filter_by(name='John').first()
更新用户年龄
user.age = 31
提交会话
session.commit()

删除数据

# 查询要删除的用户
user = session.query(User).filter_by(name='John').first()
删除用户
session.delete(user)
提交会话
session.commit()

二、使用PyMySQL连接MySQL数据库

PyMySQL是一个纯Python实现的MySQL客户端库。它支持Python 3.x，并且可以直接使用MySQL的API进行操作。

1、安装PyMySQL

首先，安装PyMySQL库：

pip install pymysql

2、连接数据库

使用PyMySQL连接MySQL数据库：

import pymysql
创建数据库连接
connection = pymysql.connect(
    host='hostname',
    user='username',
    password='password',
    database='database'
)

3、创建游标（Cursor）

游标用于执行SQL语句和获取查询结果。

# 创建游标对象
cursor = connection.cursor()

4、执行SQL语句

插入数据

# SQL语句
sql = "INSERT INTO users (name, age) VALUES (%s, %s)"
执行SQL语句
cursor.execute(sql, ('John', 30))
提交事务
connection.commit()

查询数据

# SQL语句
sql = "SELECT * FROM users"
执行SQL语句
cursor.execute(sql)
获取查询结果
result = cursor.fetchall()
for row in result:
    print(row)

更新数据

# SQL语句
sql = "UPDATE users SET age = %s WHERE name = %s"
执行SQL语句
cursor.execute(sql, (31, 'John'))
提交事务
connection.commit()

删除数据

# SQL语句
sql = "DELETE FROM users WHERE name = %s"
执行SQL语句
cursor.execute(sql, ('John',))
提交事务
connection.commit()

5、关闭连接

操作完成后，记得关闭游标和数据库连接。

# 关闭游标
cursor.close()
关闭数据库连接
connection.close()

三、使用psycopg2连接PostgreSQL数据库

psycopg2是一个用于连接PostgreSQL数据库的Python库。它是目前最流行的PostgreSQL数据库适配器。

1、安装psycopg2

首先，安装psycopg2库：

pip install psycopg2

2、连接数据库

使用psycopg2连接PostgreSQL数据库：

import psycopg2
创建数据库连接
connection = psycopg2.connect(
    host='hostname',
    user='username',
    password='password',
    database='database'
)

3、创建游标（Cursor）

游标用于执行SQL语句和获取查询结果。

# 创建游标对象
cursor = connection.cursor()

4、执行SQL语句

插入数据

# SQL语句
sql = "INSERT INTO users (name, age) VALUES (%s, %s)"
执行SQL语句
cursor.execute(sql, ('John', 30))
提交事务
connection.commit()

查询数据

# SQL语句
sql = "SELECT * FROM users"
执行SQL语句
cursor.execute(sql)
获取查询结果
result = cursor.fetchall()
for row in result:
    print(row)

更新数据

# SQL语句
sql = "UPDATE users SET age = %s WHERE name = %s"
执行SQL语句
cursor.execute(sql, (31, 'John'))
提交事务
connection.commit()

删除数据

# SQL语句
sql = "DELETE FROM users WHERE name = %s"
执行SQL语句
cursor.execute(sql, ('John',))
提交事务
connection.commit()

5、关闭连接

操作完成后，记得关闭游标和数据库连接。

# 关闭游标
cursor.close()
关闭数据库连接
connection.close()

四、使用SQLite3连接SQLite数据库

SQLite3是Python标准库中内置的数据库模块，无需额外安装，适用于轻量级、嵌入式应用。

1、连接数据库

使用SQLite3连接SQLite数据库：

import sqlite3
创建数据库连接
connection = sqlite3.connect('database.db')

2、创建游标（Cursor）

游标用于执行SQL语句和获取查询结果。

# 创建游标对象
cursor = connection.cursor()

3、执行SQL语句

插入数据

# SQL语句
sql = "INSERT INTO users (name, age) VALUES (?, ?)"
执行SQL语句
cursor.execute(sql, ('John', 30))
提交事务
connection.commit()

查询数据

# SQL语句
sql = "SELECT * FROM users"
执行SQL语句
cursor.execute(sql)
获取查询结果
result = cursor.fetchall()
for row in result:
    print(row)

更新数据

# SQL语句
sql = "UPDATE users SET age = ? WHERE name = ?"
执行SQL语句
cursor.execute(sql, (31, 'John'))
提交事务
connection.commit()

删除数据

# SQL语句
sql = "DELETE FROM users WHERE name = ?"
执行SQL语句
cursor.execute(sql, ('John',))
提交事务
connection.commit()

5、关闭连接

操作完成后，记得关闭游标和数据库连接。

# 关闭游标
cursor.close()
关闭数据库连接
connection.close()

五、总结

通过以上介绍，我们详细了解了Python如何使用不同库连接不同类型的数据库，并进行基本的增删改查操作。SQLAlchemy提供了强大的ORM功能，使得数据库操作更加方便和高效；PyMySQL和psycopg2提供了直接操作MySQL和PostgreSQL数据库的接口；SQLite3作为Python标准库，适用于轻量级应用。在实际开发中，选择合适的数据库和连接方式，可以大大提高开发效率和代码的可维护性。