Ubuntu中Python数据库连接方法

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

1. 使用sqlite3模块（适用于SQLite数据库）

SQLite是一个轻量级的嵌入式数据库，非常适合小型应用。

import sqlite3

# 连接到SQLite数据库
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 ('Alice')''')

# 提交事务
conn.commit()

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

# 关闭连接
conn.close()

2. 使用psycopg2模块（适用于PostgreSQL数据库）

PostgreSQL是一个功能强大的开源关系型数据库。

首先，安装psycopg2：

sudo apt-get install python3-psycopg2

然后，使用以下代码连接和操作数据库：

import psycopg2

# 连接到PostgreSQL数据库
conn = psycopg2.connect(
    dbname="your_dbname",
    user="your_username",
    password="your_password",
    host="localhost",
    port="5432"
)

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

# 执行SQL查询
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, name VARCHAR(100))''')

# 插入数据
cursor.execute('''INSERT INTO users (name) VALUES ('Bob')''')

# 提交事务
conn.commit()

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

# 关闭连接
conn.close()

3. 使用mysql-connector-python模块（适用于MySQL数据库）

MySQL是一个广泛使用的开源关系型数据库。

首先，安装mysql-connector-python：

sudo apt-get install python3-mysqldb

然后，使用以下代码连接和操作数据库：

import mysql.connector

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

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

# 执行SQL查询
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(100))''')

# 插入数据
cursor.execute('''INSERT INTO users (name) VALUES ('Charlie')''')

# 提交事务
conn.commit()

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

# 关闭连接
conn.close()

4. 使用pymongo模块（适用于MongoDB数据库）

MongoDB是一个流行的NoSQL数据库。

首先，安装pymongo：

sudo apt-get install python3-pymongo

然后，使用以下代码连接和操作数据库：

from pymongo import MongoClient

# 连接到MongoDB数据库
client = MongoClient('mongodb://localhost:27017/')

# 选择数据库
db = client['your_database']

# 选择集合
users = db['users']

# 插入数据
users.insert_one({'name': 'David'})

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

总结

选择哪种数据库连接方法取决于你的具体需求和所使用的数据库类型。上述示例涵盖了常见的几种数据库及其连接方法。确保在连接数据库时使用正确的凭据和配置。