python数据库操作之sqlalchemy逆向工程

文子阳

已于 2022-11-02 14:14:50 修改

阅读量8.8w

点赞数

分类专栏： python 文章标签： mysql python django

于 2021-10-19 11:41:54 首次发布

本文链接：https://blog.csdn.net/wenxingchen/article/details/120843180

版权

python 专栏收录该内容

128 篇文章 11 订阅

订阅专栏

本文介绍了如何使用 `pip install sqlacodegen` 安装工具，并通过配置数据库连接信息生成 `models.py` 文件。展示了创建数据库引擎、定义 ORM 模型以及进行数据插入和查询的基本操作。当数据库表结构变化时，只需重新生成模型文件即可。

摘要由CSDN通过智能技术生成

依赖安装

pip install sqlacodegen

数据库配置config.py

import os

from sqlalchemy import create_engine
from sqlalchemy.engine import Engine
from sqlalchemy.orm import declarative_base, sessionmaker, Session

HOST = 'localhost'
PORT = 3306
USERNAME = 'root'
PASSWORD = 'root'
DB = 'demo'

# Mysql版
DB_URI = f'mysql+pymysql://{USERNAME}:{PASSWORD}@{HOST}:{PORT}/{DB}'

# SQLite版
# DB_URI = f'sqlite:///fast.db'

# 自动生成models
os.system(f'sqlacodegen {DB_URI} > models.py')

# 操作数据句柄
engine: Engine = create_engine(DB_URI)
Base = declarative_base(engine)
session: Session = sessionmaker(engine)()

右键运行config.py,会在当前目录生成models.py模型文件如下，该文件最好不要改动，后面数据库表改动的话，直接重新生成就好。

# coding: utf-8
from sqlalchemy import Column, DateTime, Integer, SmallInteger, String, TIMESTAMP, text
from sqlalchemy.dialects.mysql import TINYINT, VARCHAR
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()
metadata = Base.metadata


class Admin(Base):
    __tablename__ = 'admin'

    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    password = Column(String(50))
    status = Column(Integer)
    create_time = Column(DateTime)
    update_time = Column(DateTime)


class AdminRole(Base):
    __tablename__ = 'admin_role'

    id = Column(Integer, primary_key=True)
    user_id = Column(Integer, nullable=False, comment='用户id')
    role_id = Column(Integer, nullable=False, comment='角色id')
    create_time = Column(TIMESTAMP, nullable=False, server_default=text("CURRENT_TIMESTAMP"), comment='创建时间')

操作数据库

from orm.config import session
from orm.models import Admin


session.add(Admin(name='张三丰', password='123456', status=1))
session.commit()
# 查询数据
admin__all = session.query(Admin.id).all()
print(admin__all)