用python如何输入数据库

使用Python输入数据库的方法包括使用合适的数据库驱动程序、建立数据库连接、执行SQL语句等步骤。本文将详细介绍如何使用Python将数据输入到不同类型的数据库中，并提供实用的代码示例。

一、了解数据库类型及对应驱动

在使用Python输入数据到数据库之前，首先需要了解常见的数据库类型以及对应的Python驱动。常见的数据库类型有：

SQLite：内嵌式数据库，适合小型应用。
MySQL：流行的开源关系型数据库。
PostgreSQL：功能强大的开源关系型数据库。
MongoDB：NoSQL数据库，适合处理大量非结构化数据。

每种数据库都有相应的Python驱动，例如：

SQLite：sqlite3
MySQL：mysql-connector-python 或 PyMySQL
PostgreSQL：psycopg2
MongoDB：pymongo

二、安装必要的库

在开始使用之前，需要安装相应的数据库驱动。例如，使用 pip 命令安装：

pip install mysql-connector-python psycopg2 pymongo

三、连接数据库

连接数据库是进行数据操作的第一步。以下是连接不同类型数据库的示例代码。

1、SQLite

import sqlite3
创建一个数据库连接
conn = sqlite3.connect('example.db')
创建一个游标对象
cur = conn.cursor()
执行SQL语句
cur.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')
提交事务
conn.commit()
关闭连接
conn.close()

2、MySQL

import mysql.connector
创建一个数据库连接
conn = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)
创建一个游标对象
cur = conn.cursor()
执行SQL语句
cur.execute('''CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), age INT)''')
提交事务
conn.commit()
关闭连接
conn.close()

3、PostgreSQL

import psycopg2
创建一个数据库连接
conn = psycopg2.connect(
    host="localhost",
    database="yourdatabase",
    user="yourusername",
    password="yourpassword"
)
创建一个游标对象
cur = conn.cursor()
执行SQL语句
cur.execute('''CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, name VARCHAR(255), age INT)''')
提交事务
conn.commit()
关闭连接
conn.close()

4、MongoDB

import pymongo
创建一个MongoDB客户端
client = pymongo.MongoClient("mongodb://localhost:27017/")
选择数据库
db = client["yourdatabase"]
选择集合
collection = db["users"]
插入数据
user = {"name": "John", "age": 30}
collection.insert_one(user)

四、插入数据

在连接数据库后，可以通过执行SQL INSERT 语句或相应的操作，将数据插入到数据库中。

1、SQLite

import sqlite3
创建一个数据库连接
conn = sqlite3.connect('example.db')
创建一个游标对象
cur = conn.cursor()
插入数据
cur.execute("INSERT INTO users (name, age) VALUES (?, ?)", ("John", 30))
提交事务
conn.commit()
关闭连接
conn.close()

2、MySQL

import mysql.connector
创建一个数据库连接
conn = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)
创建一个游标对象
cur = conn.cursor()
插入数据
cur.execute("INSERT INTO users (name, age) VALUES (%s, %s)", ("John", 30))
提交事务
conn.commit()
关闭连接
conn.close()

3、PostgreSQL

import psycopg2
创建一个数据库连接
conn = psycopg2.connect(
    host="localhost",
    database="yourdatabase",
    user="yourusername",
    password="yourpassword"
)
创建一个游标对象
cur = conn.cursor()
插入数据
cur.execute("INSERT INTO users (name, age) VALUES (%s, %s)", ("John", 30))
提交事务
conn.commit()
关闭连接
conn.close()

4、MongoDB

import pymongo
创建一个MongoDB客户端
client = pymongo.MongoClient("mongodb://localhost:27017/")
选择数据库
db = client["yourdatabase"]
选择集合
collection = db["users"]
插入数据
user = {"name": "John", "age": 30}
collection.insert_one(user)

五、查询数据

为了验证数据是否成功插入，可以进行查询操作。

1、SQLite

import sqlite3
创建一个数据库连接
conn = sqlite3.connect('example.db')
创建一个游标对象
cur = conn.cursor()
查询数据
cur.execute("SELECT * FROM users")
获取查询结果
rows = cur.fetchall()
for row in rows:
    print(row)
关闭连接
conn.close()

2、MySQL

import mysql.connector
创建一个数据库连接
conn = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)
创建一个游标对象
cur = conn.cursor()
查询数据
cur.execute("SELECT * FROM users")
获取查询结果
rows = cur.fetchall()
for row in rows:
    print(row)
关闭连接
conn.close()

3、PostgreSQL

import psycopg2
创建一个数据库连接
conn = psycopg2.connect(
    host="localhost",
    database="yourdatabase",
    user="yourusername",
    password="yourpassword"
)
创建一个游标对象
cur = conn.cursor()
查询数据
cur.execute("SELECT * FROM users")
获取查询结果
rows = cur.fetchall()
for row in rows:
    print(row)
关闭连接
conn.close()

4、MongoDB

import pymongo
创建一个MongoDB客户端
client = pymongo.MongoClient("mongodb://localhost:27017/")
选择数据库
db = client["yourdatabase"]
选择集合
collection = db["users"]
查询数据
for user in collection.find():
    print(user)

六、更新和删除数据

除了插入和查询数据，还需要了解如何更新和删除数据。

1、SQLite

import sqlite3
创建一个数据库连接
conn = sqlite3.connect('example.db')
创建一个游标对象
cur = conn.cursor()
更新数据
cur.execute("UPDATE users SET age = ? WHERE name = ?", (31, "John"))
删除数据
cur.execute("DELETE FROM users WHERE name = ?", ("John",))
提交事务
conn.commit()
关闭连接
conn.close()

2、MySQL

import mysql.connector
创建一个数据库连接
conn = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)
创建一个游标对象
cur = conn.cursor()
更新数据
cur.execute("UPDATE users SET age = %s WHERE name = %s", (31, "John"))
删除数据
cur.execute("DELETE FROM users WHERE name = %s", ("John",))
提交事务
conn.commit()
关闭连接
conn.close()

3、PostgreSQL

import psycopg2
创建一个数据库连接
conn = psycopg2.connect(
    host="localhost",
    database="yourdatabase",
    user="yourusername",
    password="yourpassword"
)
创建一个游标对象
cur = conn.cursor()
更新数据
cur.execute("UPDATE users SET age = %s WHERE name = %s", (31, "John"))
删除数据
cur.execute("DELETE FROM users WHERE name = %s", ("John",))
提交事务
conn.commit()
关闭连接
conn.close()

4、MongoDB

import pymongo
创建一个MongoDB客户端
client = pymongo.MongoClient("mongodb://localhost:27017/")
选择数据库
db = client["yourdatabase"]
选择集合
collection = db["users"]
更新数据
collection.update_one({"name": "John"}, {"$set": {"age": 31}})
删除数据
collection.delete_one({"name": "John"})

七、错误处理和事务管理

在进行数据库操作时，错误处理和事务管理非常重要，以确保数据一致性和可靠性。

1、SQLite

import sqlite3
try:
    # 创建一个数据库连接
    conn = sqlite3.connect('example.db')
    # 创建一个游标对象
    cur = conn.cursor()
    # 启动事务
    conn.execute('BEGIN TRANSACTION')
    # 执行SQL语句
    cur.execute("INSERT INTO users (name, age) VALUES (?, ?)", ("John", 30))
    # 提交事务
    conn.commit()
except sqlite3.Error as e:
    # 回滚事务
    conn.rollback()
    print(f"An error occurred: {e}")
finally:
    # 关闭连接
    conn.close()

2、MySQL

import mysql.connector
try:
    # 创建一个数据库连接
    conn = mysql.connector.connect(
        host="localhost",
        user="yourusername",
        password="yourpassword",
        database="yourdatabase"
    )
    # 创建一个游标对象
    cur = conn.cursor()
    # 启动事务
    conn.start_transaction()
    # 执行SQL语句
    cur.execute("INSERT INTO users (name, age) VALUES (%s, %s)", ("John", 30))
    # 提交事务
    conn.commit()
except mysql.connector.Error as e:
    # 回滚事务
    conn.rollback()
    print(f"An error occurred: {e}")
finally:
    # 关闭连接
    conn.close()

3、PostgreSQL

import psycopg2
try:
    # 创建一个数据库连接
    conn = psycopg2.connect(
        host="localhost",
        database="yourdatabase",
        user="yourusername",
        password="yourpassword"
    )
    # 创建一个游标对象
    cur = conn.cursor()
    # 启动事务
    conn.autocommit = False
    # 执行SQL语句
    cur.execute("INSERT INTO users (name, age) VALUES (%s, %s)", ("John", 30))
    # 提交事务
    conn.commit()
except psycopg2.Error as e:
    # 回滚事务
    conn.rollback()
    print(f"An error occurred: {e}")
finally:
    # 关闭连接
    conn.close()

4、MongoDB

import pymongo
try:
    # 创建一个MongoDB客户端
    client = pymongo.MongoClient("mongodb://localhost:27017/")
    # 选择数据库
    db = client["yourdatabase"]
    # 选择集合
    collection = db["users"]
    # 启动事务
    with client.start_session() as session:
        with session.start_transaction():
            # 插入数据
            user = {"name": "John", "age": 30}
            collection.insert_one(user, session=session)
except pymongo.errors.PyMongoError as e:
    print(f"An error occurred: {e}")

八、使用ORM框架

为了简化数据库操作，Python提供了多种ORM（对象关系映射）框架，如SQLAlchemy和Django ORM。

1、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', echo=True)
创建基类
Base = declarative_base()
定义表结构
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)
创建表
Base.metadata.create_all(engine)
创建会话
Session = sessionmaker(bind=engine)
session = Session()
插入数据
new_user = User(name="John", age=30)
session.add(new_user)
session.commit()
查询数据
for user in session.query(User).all():
    print(user.name, user.age)

2、Django ORM

首先，安装Django：

pip install django

然后，创建一个Django项目和应用：

django-admin startproject myproject cd myproject python manage.py startapp myapp

在 myapp/models.py 中定义模型：

from django.db import models
class User(models.Model):
    name = models.CharField(max_length=100)
    age = models.IntegerField()

在 myproject/settings.py 中配置数据库：

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.sqlite3',
        'NAME': BASE_DIR / "db.sqlite3",
    }
}

运行迁移命令：

python manage.py makemigrations python manage.py migrate

插入和查询数据：

from myapp.models import User
插入数据
user = User(name="John", age=30)
user.save()
查询数据
for user in User.objects.all():
    print(user.name, user.age)

九、总结

本文详细介绍了使用Python输入数据库的多种方法，涵盖了SQLite、MySQL、PostgreSQL和MongoDB等常见数据库的连接、插入、查询、更新、删除操作。同时，还介绍了错误处理、事务管理和使用ORM框架（如SQLAlchemy和Django ORM）简化数据库操作的技巧。通过本文的学习，你应该能够熟练地使用Python与各种数据库进行交互，为开发高效、可靠的应用程序打下坚实的基础。