Python数据库连接方法

在Python中，有多种方法可以用来连接数据库。以下是一些常用的数据库连接方法：

1. SQLite3：Python内置的SQLite3库用于连接SQLite数据库。这是一个轻量级的数据库，适用于小型项目。

import sqlite3

# 连接到数据库文件
conn = sqlite3.connect('example.db')

# 创建一个游标对象
cursor = conn.cursor()

# 执行SQL语句
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)')
cursor.execute('INSERT INTO users (name) VALUES (?)', ('John',))
conn.commit()

# 查询数据
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

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

2. MySQL/MariaDB：使用mysql-connector-python或pymysql库连接MySQL或MariaDB数据库。

安装mysql-connector-python：

pip install mysql-connector-python

import mysql.connector

# 连接到数据库
conn = mysql.connector.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='example_db'
)

# 创建一个游标对象
cursor = conn.cursor()

# 执行SQL语句
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255))')
cursor.execute('INSERT INTO users (name) VALUES (%s)', ('John',))
conn.commit()

# 查询数据
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

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

3. PostgreSQL：使用psycopg2库连接PostgreSQL数据库。

安装psycopg2：

pip install psycopg2

import psycopg2

# 连接到数据库
conn = psycopg2.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    dbname='example_db'
)

# 创建一个游标对象
cursor = conn.cursor()

# 执行SQL语句
cursor.execute('CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, name VARCHAR(255))')
cursor.execute('INSERT INTO users (name) VALUES (%s)', ('John',))
conn.commit()

# 查询数据
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

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

4. MongoDB：使用pymongo库连接MongoDB数据库。

安装pymongo：

pip install pymongo

from pymongo import MongoClient

# 连接到数据库
client = MongoClient('mongodb://localhost:27017/')
db = client['example_db']
collection = db['users']

# 插入数据
collection.insert_one({'name': 'John'})

# 查询数据
users = collection.find()
for user in users:
    print(user)

这些示例展示了如何使用Python连接不同类型的数据库。请根据您的项目需求选择合适的库和连接方法。