Sqlacodegen生成混合模型和表格

时间:2016-01-08 11:08:16

标签: python sqlalchemy sqlacodegen

执行此命令:

sqlacodegen <connection-url> --outfile db.py 

db.py包含生成的表:

t_table1 = Table(...)

和课程:

Table2(Base):
    __tablename__ = 'table2'

问题是表格只以一种方式生成 - 表格或类。

我想让它只生成模型(类)但在提供的标志中我找不到这样的选项。有什么想法吗?

1 个答案:

答案 0 :(得分:11)

看起来你所描述的是一个功能本身。 sqlacodegen并不总是生成类模型。

它将仅为具有主键且不是关联表的表形成模型类,如您在source code中所见:

# Only form model classes for tables that have a primary key and are not association tables
if noclasses or not table.primary_key or table.name in association_tables:
    model = self.table_model(table)
else:
    model = self.class_model(table, links[table.name], self.inflect_engine, not nojoined)
    classes[model.name] = model

此外,在documentation中声明

  

如果表满足所有表,则将其视为关联表   以下条件:

     
      
  1. 正好有两个外键约束
  2.   
  3. 所有列都涉及所述约束
  4.   

虽然,你可以尝试快速而肮脏的黑客攻击。在源代码中找到这些行(类似/.../lib/python2.7/site-packages/sqlacodegen/codegen.py)并注释掉前三个代码行(并修复缩进):

# Only form model classes for tables that have a primary key and are not association tables
# if noclasses or not table.primary_key or table.name in association_tables:
#    model = self.table_model(table)
# else:
model = self.class_model(table, links[table.name], self.inflect_engine, not nojoined)
classes[model.name] = model

我已经为一个作为表模型生成的特定表尝试了这个。它来自

t_Admin_op = Table(
    'Admin_op', metadata,
    Column('id_admin', Integer, nullable=False),
    Column('id_op', Integer, nullable=False)
)

class AdminOp(Base):
    __tablename__ = 'Admin_op'

    id_admin = Column(Integer, nullable=False)
    id_op = Column(Integer, nullable=False)

您还可以在official tracker

中打开有关此问题的问题作为功能请求

以防万一,如果你想要相反的(只有表格模型),你可以使用--noclasses标志。