1. PingCode智库首页
  2. 百科

python中如何调用数据库

Edit2 • • 百科

python中如何调用数据库

在Python中调用数据库的常见方法有：使用特定数据库的连接库、ORM框架、SQLAlchemy等。 其中，使用特定数据库的连接库是最基础的方法，ORM框架则提供了一种更高级的操作方式，而SQLAlchemy则结合了这两者的优点。下面将详细介绍每种方法的使用步骤和注意事项。

一、使用特定数据库的连接库

1、SQLite

SQLite是一个轻量级的嵌入式数据库，Python内置了对SQLite的支持，可以直接使用sqlite3模块进行数据库操作。

示例代码：

import sqlite3


连接到数据库（如果数据库不存在，则会自动创建）


conn = sqlite3.connect('example.db')


创建一个游标对象


cursor = conn.cursor()


创建表


cursor.execute('''CREATE TABLE IF NOT EXISTS users


                  (id INTEGER PRIMARY KEY AUTOINCREMENT,


                   name TEXT,


                   age INTEGER)''')


插入数据


cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))


cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Bob', 30))


提交事务


conn.commit()


查询数据


cursor.execute("SELECT * FROM users")


rows = cursor.fetchall()


for row in rows:


    print(row)


关闭连接


conn.close()

2、MySQL

MySQL是一个流行的关系型数据库管理系统。可以使用mysql-connector-python或PyMySQL等库来进行数据库操作。

示例代码：

import mysql.connector


连接到数据库


conn = mysql.connector.connect(


    host='localhost',


    user='yourusername',


    password='yourpassword',


    database='yourdatabase'


)


创建一个游标对象


cursor = conn.cursor()


创建表


cursor.execute('''CREATE TABLE IF NOT EXISTS users


                  (id INT AUTO_INCREMENT PRIMARY KEY,


                   name VARCHAR(255),


                   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()


查询数据


cursor.execute("SELECT * FROM users")


rows = cursor.fetchall()


for row in rows:


    print(row)


关闭连接


conn.close()

3、PostgreSQL

PostgreSQL是一个功能强大的开源对象关系型数据库系统。可以使用psycopg2库来进行数据库操作。

示例代码：

import psycopg2


连接到数据库


conn = psycopg2.connect(


    dbname='yourdatabase',


    user='yourusername',


    password='yourpassword',


    host='localhost'


)


创建一个游标对象


cursor = conn.cursor()


创建表


cursor.execute('''CREATE TABLE IF NOT EXISTS users


                  (id SERIAL PRIMARY KEY,


                   name VARCHAR(255),


                   age INTEGER)''')


插入数据


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()


查询数据


cursor.execute("SELECT * FROM users")


rows = cursor.fetchall()


for row in rows:


    print(row)


关闭连接


conn.close()

二、使用ORM框架

ORM（Object-Relational Mapping）框架通过映射数据库表和类，使得开发者可以通过面向对象的方式进行数据库操作。常见的ORM框架有Django ORM和SQLAlchemy。

1、Django ORM

Django是一个高级的Python Web框架，内置了ORM功能。以下是一个简单的示例：

安装Django：

pip install django

定义模型：

# models.py


from django.db import models


class User(models.Model):


    name = models.CharField(max_length=255)


    age = models.IntegerField()

操作数据库：

# 使用Django Shell


python manage.py shell


from yourapp.models import User


创建数据


user1 = User(name='Alice', age=25)


user1.save()


查询数据


users = User.objects.all()


for user in users:


    print(user.name, user.age)

2、SQLAlchemy

SQLAlchemy是一个强大的Python SQL工具包和对象关系映射器。它既支持ORM，又提供了对原生SQL的良好支持。

安装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, autoincrement=True)


    name = Column(String)


    age = Column(Integer)


创建表


Base.metadata.create_all(engine)


创建会话


Session = sessionmaker(bind=engine)


session = Session()


插入数据


user1 = User(name='Alice', age=25)


user2 = User(name='Bob', age=30)


session.add(user1)


session.add(user2)


session.commit()


查询数据


users = session.query(User).all()


for user in users:


    print(user.name, user.age)


关闭会话


session.close()

三、连接和事务管理

无论使用何种方式连接数据库，管理好连接和事务是至关重要的。确保在操作完成后关闭连接，使用事务来保证数据一致性。

1、自动提交和手动提交

大多数数据库连接库默认开启自动提交模式，但在复杂操作中，手动管理事务更为安全。

示例：

conn = sqlite3.connect('example.db')


conn.isolation_level = None  # 关闭自动提交


cursor = conn.cursor()


try:


    cursor.execute("BEGIN")


    cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))


    cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Bob', 30))


    cursor.execute("COMMIT")


except Exception as e:


    cursor.execute("ROLLBACK")


    print(f"Transaction failed: {e}")


finally:


    conn.close()

2、连接池

对于高并发的应用，使用连接池可以显著提高性能。SQLAlchemy和其他连接库通常都支持连接池。

示例：

from sqlalchemy.pool import QueuePool


engine = create_engine('sqlite:///example.db', poolclass=QueuePool, pool_size=10)

四、错误处理

在数据库操作中，错误处理是不可忽视的部分。捕获异常并妥善处理，可以提高程序的健壮性。

1、常见异常

常见的数据库操作异常包括连接失败、SQL语法错误、数据类型不匹配等。

示例：

try:


    conn = sqlite3.connect('example.db')


    cursor = conn.cursor()


    cursor.execute("SELECT * FROM non_existent_table")


except sqlite3.DatabaseError as e:


    print(f"Database error: {e}")


except sqlite3.OperationalError as e:


    print(f"Operational error: {e}")


finally:


    conn.close()

2、日志记录

使用日志记录库，可以将错误信息记录到日志文件中，方便排查和调试。

示例：

import logging


logging.basicConfig(filename='db_errors.log', level=logging.ERROR)


try:


    conn = sqlite3.connect('example.db')


    cursor = conn.cursor()


    cursor.execute("SELECT * FROM non_existent_table")


except sqlite3.DatabaseError as e:


    logging.error(f"Database error: {e}")


finally:


    conn.close()

五、性能优化

性能优化是数据库操作中一个重要的方面。合理的索引、查询优化和缓存策略都能显著提升性能。

1、索引

索引可以加快查询速度，但过多的索引会影响插入和更新操作的性能。

示例：

cursor.execute("CREATE INDEX idx_users_name ON users (name)")

2、查询优化

尽量避免使用复杂的子查询和多表连接，合理拆分查询任务。

3、缓存

缓存常用数据，减少对数据库的频繁访问。可以使用Redis等内存数据库进行缓存。

示例：

import redis


cache = redis.StrictRedis(host='localhost', port=6379, db=0)


缓存数据


cache.set('user_1', 'Alice')


获取缓存数据


user = cache.get('user_1')


print(user)

六、安全性

数据库操作的安全性同样重要，特别是在处理用户输入时，防止SQL注入攻击。

1、参数化查询

使用参数化查询可以有效防止SQL注入。

示例：

cursor.execute("SELECT * FROM users WHERE name = ?", (user_input,))

2、权限控制

合理设置数据库用户权限，避免使用高权限用户进行操作。

3、加密

对敏感数据进行加密存储，确保数据安全。

示例：

from cryptography.fernet import Fernet


生成密钥


key = Fernet.generate_key()


cipher_suite = Fernet(key)


加密数据


encrypted_text = cipher_suite.encrypt(b"Sensitive Data")


解密数据


decrypted_text = cipher_suite.decrypt(encrypted_text)


print(decrypted_text)

七、项目管理系统

在进行大型项目开发时，使用项目管理系统可以提高团队协作效率。推荐使用以下两个系统：

1、研发项目管理系统PingCode

PingCode是一款专注于研发项目管理的工具，支持需求管理、任务跟踪、缺陷管理等功能，帮助团队高效协作。

2、通用项目协作软件Worktile

Worktile是一款通用的项目协作软件，支持任务管理、时间管理、文件管理等功能，适用于各种类型的项目。

总结

在Python中调用数据库涉及到选择合适的数据库连接库或ORM框架、管理好连接和事务、处理错误、优化性能以及确保安全性。在大型项目中，使用项目管理系统可以进一步提升团队协作效率。通过上述方法和技巧，相信你可以更加高效、安全地进行数据库操作。

相关问答FAQs：

Q: 如何在Python中连接数据库并进行操作？
A: 在Python中，可以使用各种库和模块来连接和操作数据库。其中最常用的是sqlite3、MySQLdb和psycopg2等。可以根据需要选择适合的库，并按照其文档进行安装和配置。

Q: 如何在Python中执行SQL查询语句？
A: 在Python中，可以使用数据库连接对象的execute()方法执行SQL查询语句。首先，需要建立数据库连接，然后使用连接对象的cursor()方法创建游标对象。接下来，可以使用游标对象的execute()方法执行SQL查询语句。

Q: 如何在Python中获取数据库查询结果？
A: 在Python中，可以使用游标对象的fetchone()和fetchall()方法获取数据库查询结果。fetchone()方法用于获取查询结果的下一行，fetchall()方法用于获取所有的查询结果。可以使用循环遍历的方式逐行处理查询结果，或者直接使用fetchall()方法获取所有结果并进行处理。

文章包含AI辅助创作，作者：Edit2，如若转载，请注明出处：https://docs.pingcode.com/baike/1784870

赞 (0)
Edit2Edit2
免费注册
电话联系

4008001024

微信咨询
微信咨询
返回顶部