Problem inserting an enum value (Postgres + SQL Alchemy)

Question

I'm having problems to insert a value as an enum value. This is my model:

class InvitationStatusEnum(enum.Enum):
    FAILED = 1
    SENT = 2
    RESENT = 3
    ACCEPTED = 4

class Invitations(db.Model):

    __tablename__ = "tb_invitations"
    invitation_id = db.Column(db.Integer, nullable=False, primary_key=True)
    _from = db.Column("from", db.String, nullable=False)
    to = db.Column(db.String, nullable=False)
    status = db.Column("status", db.Enum(*tuple(e.name for e in tuple(InvitationStatusEnum)), name="invitation_status_enum", create_type=False))
    created_at = db.Column(db.DateTime(timezone=True), nullable=False)

    def __init__(self, **kwargs):

        self.invitation_id = kwargs["invitation_id"]
        self._from = kwargs["_from"],
        self.to = kwargs["to"],
        self.status = kwargs["status"],
        self.created_at = datetime.now()

The type is already created in Postgres as invitation_status_enum. Here I do the insert:

new_invitation = model.Invitations(
    invitation_id=1,
    _from=environ["CONCIL_NOTIFICATION_EMAIL"],
    to=to,
    status=model.InvitationStatusEnum.SENT.name
)

query = db.session.add(new_invitation)
print(query)
db.session.flush()

And I get the following error:

(builtins.LookupError) '('SENT',)' is not among the defined enum values. Enum name: invitation_status_enum. Possible values: FAILED, SENT, RESENT, ACCEPTED\n[SQL: INSERT INTO tb_invitations (invitation_id, \"from\", \"to\", status, created_at) VALUES (%(invitation_id)s, %(from)s, %(to)s, %(status)s, %(created_at)s)]\n[parameters: [{'status': ('SENT',), 'created_at': datetime.datetime(2020, 10, 15, 10, 52, 6, 819247), 'to': ('[email protected]',), 'invitation_id': 1, 'from': ('[email protected]',)}]]

If the status is model.InvitationStatusEnum.SENT or model.InvitationStatusEnum.SENT.value, it doesn't matter, I get the same error. How do I insert as a value that postgres can accept?

Ericson Willians · Accepted Answer · 2020-10-15 14:18:04Z

2

Solved, you won't believe it, but I was creating tuples accidentally:

def __init__(self, **kwargs):

    self.invitation_id = kwargs["invitation_id"]
    self._from = kwargs["_from"],
    self.to = kwargs["to"],
    self.status = kwargs["status"],
    self.created_at = datetime.now()

The commas were there, and I was thinking the tuples were related to SQL Alchemy. Anyway, it happens...

answered Oct 15, 2020 at 14:18

Ericson Willians

7,90317 gold badges70 silver badges124 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Rocco Over a year ago

Thank you! I had the same issue because I was copy-pasting some model creation code and didn't think to look at the end of the line!

Collectives™ on Stack Overflow

Problem inserting an enum value (Postgres + SQL Alchemy)

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related