SQLAlchemyのリレーションでULIDをデフォルトにしたときユニーク制約でエラーが出る

Question

実現したいこと

• SQLAlchemyのリレーションでULIDを使う

前提

使用するモジュールは以下のとおりです

• Flask
• Flask-Login
• SQLAlchemy
• ulid-py

発生している問題・エラーメッセージ

sqlalchemy.exc.IntegrityError: (sqlite3.IntegrityError) UNIQUE constraint failed: roles.ulid
[SQL: INSERT INTO roles (ulid, name) VALUES (?, ?)]
[parameters: ('01GTV995JF912S32TC2QG6N5WB', 'User')]
(Background on this error at: https://sqlalche.me/e/14/gkpj)

該当のソースコード

テーブルの定義

class USER_ROLE(db.Model):
    __tablename__ = 'user_role'
    user_id = db.Column(db.Integer(), ForeignKey('users.ulid', ondelete='CASCADE'), primary_key=True)
    role_id = db.Column(db.Integer(), ForeignKey('roles.ulid', ondelete='CASCADE'), primary_key=True)

class USERS(db.Model, UserMixin):
    __tablename__ = 'users'

    ulid = db.Column(db.Text(26), primary_key=True,  default = ulid.new().str)
    name = db.Column(db.String(100), unique = True)
    password = db.Column(db.String(100))
    display_name = db.Column(db.String(100))

    # Relationships
    pages = db.relationship('PAGES', secondary=UP.__tablename__, back_populates='author')
    roles = db.relationship('ROLES', secondary='ur')

class ROLES(db.Model):
    __tablename__ = 'roles'
    ulid = db.Column(db.Text(26), primary_key=True,  default = ulid.new().str)
    name = db.Column(db.String(50), unique=True)

初期化

@app.before_first_request
def init():
    try:
        tables.USERS.query.filter(tables.USERS.name=="admin").one_or_none()
    except Exception:
        db.create_all()
        admin_role = tables.ROLES(name='Admin')
        db.session.add(admin_role)
        db.session.commit()
        user_role = tables.ROLES(name='User')
        db.session.add(user_role)
        db.session.commit()
        admin = tables.USERS(
            name='admin',
            display_name='Administrator',
            password=generate_password_hash('password')
        )
        admin.roles = [admin_role, student_role]
        db.session.add(admin)
        db.session.commit()

試したこと

初期化する部分のプログラムを特に見直しましたが同じロールを追加しているわけでもないので全く原因がわかりませんでした。

検証すればいいことがわからなかったので情報が少ないかもしれませんがよろしくお願いします

Answer 1

下記の、ulidのdefaultの設定が正しくないです。

class USERS(db.Model, UserMixin):
    ulid = db.Column(db.Text(26), primary_key=True,  default = ulid.new().str)

class ROLES(db.Model):
    ulid = db.Column(db.Text(26), primary_key=True,  default = ulid.new().str)

defaultで、毎回新しいulidを作りたいのだと思うのですが、この書き方だと、先に ulid.new()で固定のulidを作成してしまって、そのstrの値をdefault値に設定するという意味になります。なので、コンストラクタで値を渡さなければ、すべて同じ値が使われます。

毎回、新しいulidを作成したい場合は、defaultに関数などのcallableをセットします。(lambdaで書いてもいいと思います)

def ulid_new_str():
    return ulid.new().str

class USERS(db.Model, UserMixin):
    ulid = db.Column(db.Text(26), primary_key=True, default=ulid_new_str)

Comments

[tom1022]

問題なく動作しました
ありがとうございます!