如何更改 Sqlalchemy / PostgreSQL 中 ENUM 的字符串大小值

Question

我正在编写一个带有 SqlAlchemy 和 PostgreSQL 的 Flask 应用程序。当我尝试在一个超过六个字符的枚举字段中输入一个值时，出现以下错误： sqlalchemy.exc.DataError: (psycopg2.errors.StringDataRightTruncation) value too long for type character varying(6)

我假设我需要更改 ENUM 字段的字符串长度。 有谁知道如何做到这一点？

这里是我的model，相关栏目是role。

class User(db.Model, UserMixin): #Parent table to profile
    ROLE = OrderedDict([        
        ('admin', 'Admin'),
        ('carrier', 'Carrier'),
        ('driver', 'Driver'),
        ('dispatcher', 'Dispatcher'),
        ('super', 'Super'),
    ])
    id = db.Column(db.Integer, primary_key = True)
    
    user = db.relationship('Profile', backref='User', passive_deletes=True)
    percentage = db.relationship('Percentage', backref='User', passive_deletes=True)

    company_id = db.Column(db.Integer)
    role = db.Column(db.Enum(*ROLE, name='role_types', native_enum=False),
                     index=True, nullable=False, server_default='admin')
    active = db.Column('is_active', db.Boolean(), nullable=False,
                       server_default='1')
    username = db.Column(db.String(24), unique=True, index=True)
    firstname = db.Column(db.String(50), nullable=False)
    lastname = db.Column(db.String(50), index=True,nullable=False)
    email = db.Column(db.String(255), unique=True, index=True, nullable=False,
                      server_default='')
    password = db.Column(db.String(256), nullable=False, server_default='')

Answer 1

native_enum – 在可用时使用数据库的本机 ENUM 类型。 默认为真。 当为 False 时，对所有后端使用 VARCHAR + 检查约束。 当为 False 时，可以使用 Enum.length 控制 VARCHAR 长度； 如果 native_enum=True，当前“长度”将被忽略。

使用native_enum=false这不是一个真正的枚举，而只是一个 varchar 字段。 Python 应该猜出正确的长度。 但是， Enum需要一个列表，而不是一个字典。 使用 OrderedDict 可能会产生干扰。

您也可以尝试将native_enum为默认值，尽管Postgres 上的枚举有一些注意事项。

尝试使用PostgreSQL ENUM类型。

role = db.Column(
  ENUM('admin', 'carrier', 'driver', 'dispatcher', 'super', name='role_types'),
  index=True, nullable=False, server_default='admin'
)

Answer 2

最初设置表时，PostGreSQL 将长度设置为最长 Enum 的值。 但是，当我添加新的枚举并使用 Flask-Migrate 进行迁移时，长度限制仍然存在。 所以添加新的更长的枚举并没有改变初始长度限制。 我什至添加了长度参数，但这也不起作用。 @Schwern 评论中提到的解决方案是删除表，然后进行迁移。 如果有一种方法可以在不删除表的情况下重置枚举的最大长度，我还没有找到。