python怎么查询数据库

Python中查询数据库是一项常见的任务，通常用于数据存储、检索和处理，Python提供了多种库来与不同类型的数据库进行交互，如SQLite、MySQL、PostgreSQL等，以下是详细的步骤和示例代码,展示如何使用Python查询数据库。

选择数据库和安装相应的库

你需要选择一个数据库并安装相应的Python库,以下是一些常见的数据库及其对应的Python库：

• SQLite: 内置于Python标准库中,无需额外安装。
• MySQL: 使用mysql-connector-python或PyMySQL库。
• PostgreSQL: 使用psycopg2库。

安装库

pip install mysql-connector-python psycopg2

连接到数据库

连接到数据库是查询的第一步,以下是如何连接到不同类型的数据库的示例。

SQLite

import sqlite3
# 连接到SQLite数据库（如果数据库不存在，则会自动创建）
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

MySQL

import mysql.connector
# 连接到MySQL数据库
conn = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)
cursor = conn.cursor()

PostgreSQL

import psycopg2
# 连接到PostgreSQL数据库
conn = psycopg2.connect(
    host="localhost",
    database="yourdatabase",
    user="yourusername",
    password="yourpassword"
)
cursor = conn.cursor()

创建表和插入数据

在查询之前，通常需要有一个表来存储数据,以下是如何创建表和插入数据的示例。

SQLite

# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
    id INTEGER PRIMARY KEY,
    name TEXT NOT NULL,
    age INTEGER
)
''')
# 插入数据
cursor.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Alice', 25))
cursor.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Bob', 30))
conn.commit()

MySQL

# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    age INT
)
''')
# 插入数据
cursor.execute('INSERT INTO users (name, age) VALUES (%s, %s)', ('Alice', 25))
cursor.execute('INSERT INTO users (name, age) VALUES (%s, %s)', ('Bob', 30))
conn.commit()

PostgreSQL

# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
    id SERIAL PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    age INT
)
''')
# 插入数据
cursor.execute('INSERT INTO users (name, age) VALUES (%s, %s)', ('Alice', 25))
cursor.execute('INSERT INTO users (name, age) VALUES (%s, %s)', ('Bob', 30))
conn.commit()

查询数据

查询数据是数据库操作的核心,以下是如何执行查询并获取结果的示例。

SQLite

# 查询所有用户
cursor.execute('SELECT  FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

MySQL

# 查询所有用户
cursor.execute('SELECT  FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

PostgreSQL

# 查询所有用户
cursor.execute('SELECT  FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

关闭连接

在完成数据库操作后,务必关闭连接以释放资源。

cursor.close()
conn.close()

使用ORM（对象关系映射）

除了直接使用SQL查询，Python还提供了ORM（对象关系映射）工具，如SQLAlchemy,它可以帮助你以更面向对象的方式操作数据库。

安装SQLAlchemy

pip install sqlalchemy

使用SQLAlchemy查询数据

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
# 创建引擎
engine = create_engine('sqlite:///example.db')
Base = declarative_base()
# 定义模型
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)
# 创建会话
Session = sessionmaker(bind=engine)
session = Session()
# 查询所有用户
users = session.query(User).all()
for user in users:
    print(user.id, user.name, user.age)
# 关闭会话
session.close()

常见问题及解决方案

Q1: 如何防止SQL注入？

A1: 使用参数化查询或ORM工具可以有效防止SQL注入，在SQLite中使用作为占位符，在MySQL中使用%s作为占位符。

Q2: 如何处理数据库连接错误？

A2: 可以使用try-except块捕获异常,并在发生错误时进行适当的处理。

try:
    conn = mysql.connector.connect(host="localhost", user="yourusername", password="yourpassword", database="yourdatabase")
except mysql.connector.Error as err:
    print(f"Error: {err}")

Python提供了多种方式来查询数据库，无论是直接使用SQL还是通过ORM工具,都可以方便地进行数据库操作。