diff --git a/alembic/versions/888e96cc67bd_added_listen_updates_to_connections.py b/alembic/versions/888e96cc67bd_added_listen_updates_to_connections.py new file mode 100644 index 0000000..a11fd68 --- /dev/null +++ b/alembic/versions/888e96cc67bd_added_listen_updates_to_connections.py @@ -0,0 +1,30 @@ +"""Added listen_updates to connections. + +Revision ID: 888e96cc67bd +Revises: 1c62ff091f5c +Create Date: 2025-04-07 20:27:07.748395 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision: str = '888e96cc67bd' +down_revision: Union[str, None] = '1c62ff091f5c' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.add_column('connections', sa.Column('listen_updates', sa.Boolean(), nullable=False)) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column('connections', 'listen_updates') + # ### end Alembic commands ### diff --git a/data/models.py b/data/models.py index 8eccb0e..b49232b 100644 --- a/data/models.py +++ b/data/models.py @@ -1,5 +1,5 @@ # models.py -from sqlalchemy import Column, Integer, String, Enum, ForeignKey, JSON +from sqlalchemy import Column, Integer, String, Enum, ForeignKey, JSON, Boolean from sqlalchemy.orm import relationship from data.db import Base from core.enums import ConnectionTypes, UserRole @@ -26,6 +26,7 @@ class Connection(Base): password = Column(String) pool_minsize = Column(Integer, nullable=False, default=5) pool_maxsize = Column(Integer, nullable=False, default=10) + listen_updates = Column(Boolean, nullable=False, default=False) owner_id = Column(Integer, ForeignKey("users.id")) # owner = relationship("User", back_populates="connections")