Python连接数据库|极客教程

Python连接数据库

在数据处理和存储领域，数据库是不可或缺的重要组成部分。Python作为一种高效、灵活的编程语言，提供了丰富的库和模块，可以方便地连接和操作各种类型的数据库。本文将介绍如何使用Python连接数据库，包括MySQL、PostgreSQL和SQLite等常见数据库。

MySQL数据库连接

MySQL是一种流行的关系型数据库管理系统，在Python中可以使用mysql-connector库来连接MySQL数据库。

首先，我们需要安装mysql-connector库，可以使用以下命令进行安装：

pip install mysql-connector

接下来，我们可以通过以下代码连接MySQL数据库：

import mysql.connector

# 连接数据库
mydb = mysql.connector.connect(
  host="localhost",
  user="root",
  password="password",
  database="geek_docs"
)

print(mydb)

运行以上代码，如果连接成功，将输出类似如下结果：

<mysql.connector.connection.MySQLConnection object at 0x7f9993ad11f0>

我们还可以执行SQL语句来操作数据库，例如查询数据库中的表：

import mysql.connector

mydb = mysql.connector.connect(
  host="localhost",
  user="root",
  password="password",
  database="geek_docs"
)

mycursor = mydb.cursor()

mycursor.execute("SHOW TABLES")

for table in mycursor:
  print(table)

以上代码将输出数据库中的所有表名。通过这种方式，我们可以轻松连接MySQL数据库，并对数据库进行操作。

PostgreSQL数据库连接

另一种常见的关系型数据库是PostgreSQL，在Python中可以使用psycopg2库来连接PostgreSQL数据库。

首先，我们需要安装psycopg2库，可以使用以下命令进行安装：

pip install psycopg2

接下来，我们可以通过以下代码连接PostgreSQL数据库：

import psycopg2

# 连接数据库
conn = psycopg2.connect(
  dbname="geek_docs",
  user="postgres",
  password="password",
  host="localhost"
)

print(conn)

运行以上代码，如果连接成功，将输出类似如下结果：

<connection object at 0x7f9993ad11f0; dsn: user=postgres password=xxx dbname=geek_docs host=localhost>

我们还可以执行SQL语句来操作数据库，例如查询数据库中的表：

import psycopg2

conn = psycopg2.connect(
  dbname="geek_docs",
  user="postgres",
  password="password",
  host="localhost"
)

cur = conn.cursor()

cur.execute("SELECT table_name FROM information_schema.tables WHERE table_schema='public'")

for table in cur:
  print(table)

以上代码将输出数据库中的所有表名。通过这种方式，我们可以轻松连接PostgreSQL数据库，并对数据库进行操作。

SQLite数据库连接

除了MySQL和PostgreSQL之外，SQLite是另一种轻量级的数据库管理系统，在Python中可以使用内置的sqlite3库来连接SQLite数据库。

我们可以通过以下代码连接SQLite数据库：

import sqlite3

# 连接数据库
conn = sqlite3.connect('geek_docs.db')

print(conn)

运行以上代码，如果连接成功，将输出类似如下结果：

<sqlite3.Connection object at 0x7f9993ad11f0>

我们还可以执行SQL语句来操作数据库，例如创建表并插入数据：

import sqlite3

conn = sqlite3.connect('geek_docs.db')
cur = conn.cursor()

cur.execute('''CREATE TABLE IF NOT EXISTS users (
               id INTEGER PRIMARY KEY,
               name TEXT NOT NULL)''')

cur.execute("INSERT INTO users (name) VALUES ('Alice')")
conn.commit()

cur.execute("SELECT * FROM users")
print(cur.fetchall())