Python一键生成所有数据库表结构的简单函数

在Python中，我们可以使用ORM（对象关系映射）库来简化数据库表结构的生成。ORM库可以将数据库表结构映射为Python类的结构，使我们能够直接使用Python类的方式操作数据库。

下面是一个简单的函数，用于根据Python类生成数据库表结构：

import sqlalchemy
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

def create_tables():
    # 创建数据库连接
    engine = create_engine('mysql://username:password@host:port/database')
    connection = engine.connect()

    # 继承Base类来创建新的类
    class User(Base):
        __tablename__ = 'users'
        id = Column(Integer, primary_key=True)
        name = Column(String(50))
        email = Column(String(50))

    class Product(Base):
        __tablename__ = 'products'
        id = Column(Integer, primary_key=True)
        name = Column(String(50))
        price = Column(Integer)

    # 创建所有表
    Base.metadata.create_all(bind=engine)

    # 关闭数据库连接
    connection.close()

# 调用函数生成数据库表
create_tables()

上述代码使用SQLAlchemy库来创建数据库表结构。首先，我们需要创建一个数据库连接字符串，并通过create_engine函数创建一个数据库引擎。然后，我们定义每个表的Python类，使用Column定义表的字段。最后，调用Base.metadata.create_all函数来创建所有表。

在使用这个函数之前，需要先安装SQLAlchemy库。可以使用pip来安装：

pip install SQLAlchemy

下面是一个简单的例子，使用这个函数来生成用户表和产品表的数据库表结构：

import sqlalchemy
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

def create_tables():
    # 创建数据库连接
    engine = create_engine('mysql://username:password@host:port/database')
    connection = engine.connect()

    # 继承Base类来创建新的类
    class User(Base):
        __tablename__ = 'users'
        id = Column(Integer, primary_key=True)
        name = Column(String(50))
        email = Column(String(50))

    class Product(Base):
        __tablename__ = 'products'
        id = Column(Integer, primary_key=True)
        name = Column(String(50))
        price = Column(Integer)

    # 创建所有表
    Base.metadata.create_all(bind=engine)

    # 关闭数据库连接
    connection.close()

# 调用函数生成数据库表
create_tables()

这个例子中，我们创建了两个表：用户表和产品表。用户表包含id、name和email字段，而产品表包含id、name和price字段。

通过使用这个函数，我们可以方便地一键生成所有数据库表结构。这在开发过程中非常有用，特别是在创建数据库或者重构表结构时。