Python基础 SQLAlchemy

引子

数据库是一个二维表，通俗理解，即多行多列。看起来就是表格

如果Python来表示数据结构，则可以list表示行，tuple表示list的一个元素

[
    ("1", "李白")
    ("2", "杜甫")
]

[
    ("1", "爱迪生")
    ("2", "牛顿")
]

SQLAlchemy

SQLAlchemy的初始化和具体每个表的class定义

让表达更好看些

# 创建对象的基类
Base = declarative_base()

# 定义User对象
class User(Base):
    # 表的名字
    __tablename__ = "people"

    # 表的结构,包含如下字段
    id = Column(String(20), primary_key = True)
    name = Column(String(20))

运行示例

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Python基础 SQLAlchemy

from sqlalchemy import Column, String, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
import mysql.connector
import mysql.connector

# 注意把password设为你的root口令
conn = mysql.connector.connect(user='root', password='', database='test')

cursor = conn.cursor()
# 创建user表
cursor.execute('create table people (id varchar(20) primary key, name varchar(20))')

# 创建对象的基类
Base = declarative_base()

# 定义User对象
class User(Base):
    # 表的名字
    __tablename__ = "people"

    # 表的结构,包含如下字段
    id = Column(String(20), primary_key = True)
    name = Column(String(20))


# 初始化数据库连接:
# '数据库类型+数据库驱动名称://用户名:口令@机器地址:端口号/数据库名'
engine = create_engine('mysql+mysqlconnector://root:@localhost:3306/test')
# 创建DBSession类型:
DBSession = sessionmaker(bind=engine)

# 创建session对象:
session = DBSession()
# 创建新User对象:
new_user = User(id='8', name='Jom')
# 添加到session:
session.add(new_user)
# 提交即保存到数据库:
session.commit()

# 创建Query查询，filter是where条件，最后调用one()返回唯一行，如果调用all()则返回所有行:
user = session.query(User).filter(User.id=='8').one()
# 打印类型和对象的name属性:
print('type:', type(user))
print('name:', user.name)

# 关闭session:
session.close()

运行结果

D:\PythonProject>python main.py
type: <class '__main__.User'>
name: Jom