Active status check
This commit is contained in:
24
migrations/versions/4ee23cb29001_merge_heads.py
Normal file
24
migrations/versions/4ee23cb29001_merge_heads.py
Normal file
@@ -0,0 +1,24 @@
|
||||
"""merge heads
|
||||
|
||||
Revision ID: 4ee23cb29001
|
||||
Revises: 72ab6c4c6a5f, add_status_details
|
||||
Create Date: 2025-06-09 10:04:48.708415
|
||||
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = '4ee23cb29001'
|
||||
down_revision = ('72ab6c4c6a5f', 'add_status_details')
|
||||
branch_labels = None
|
||||
depends_on = None
|
||||
|
||||
|
||||
def upgrade():
|
||||
pass
|
||||
|
||||
|
||||
def downgrade():
|
||||
pass
|
||||
@@ -7,6 +7,7 @@ Create Date: 2024-03-19 10:00:00.000000
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy import text
|
||||
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
@@ -17,23 +18,52 @@ depends_on = None
|
||||
|
||||
|
||||
def upgrade():
|
||||
op.create_table('instances',
|
||||
sa.Column('id', sa.Integer(), nullable=False),
|
||||
sa.Column('name', sa.String(length=100), nullable=False),
|
||||
sa.Column('company', sa.String(length=100), nullable=False),
|
||||
sa.Column('rooms_count', sa.Integer(), nullable=False, server_default='0'),
|
||||
sa.Column('conversations_count', sa.Integer(), nullable=False, server_default='0'),
|
||||
sa.Column('data_size', sa.Float(), nullable=False, server_default='0.0'),
|
||||
sa.Column('payment_plan', sa.String(length=20), nullable=False, server_default='Basic'),
|
||||
sa.Column('main_url', sa.String(length=255), nullable=False),
|
||||
sa.Column('status', sa.String(length=20), nullable=False, server_default='inactive'),
|
||||
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
||||
sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
sa.UniqueConstraint('name'),
|
||||
sa.UniqueConstraint('main_url')
|
||||
)
|
||||
conn = op.get_bind()
|
||||
result = conn.execute(text("""
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_name = 'instances'
|
||||
);
|
||||
"""))
|
||||
exists = result.scalar()
|
||||
if not exists:
|
||||
op.create_table('instances',
|
||||
sa.Column('id', sa.Integer(), nullable=False),
|
||||
sa.Column('name', sa.String(length=255), nullable=False),
|
||||
sa.Column('company', sa.String(length=255), nullable=False),
|
||||
sa.Column('rooms_count', sa.Integer(), nullable=False, server_default='0'),
|
||||
sa.Column('conversations_count', sa.Integer(), nullable=False, server_default='0'),
|
||||
sa.Column('data_size', sa.String(length=50), nullable=False, server_default='0 MB'),
|
||||
sa.Column('payment_plan', sa.String(length=50), nullable=False, server_default='free'),
|
||||
sa.Column('main_url', sa.String(length=255), nullable=False),
|
||||
sa.Column('status', sa.String(length=20), nullable=False, server_default='inactive'),
|
||||
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
||||
sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
sa.UniqueConstraint('name'),
|
||||
sa.UniqueConstraint('main_url')
|
||||
)
|
||||
|
||||
# Create a trigger to automatically update the updated_at column
|
||||
op.execute("""
|
||||
CREATE OR REPLACE FUNCTION update_updated_at_column()
|
||||
RETURNS TRIGGER AS $$
|
||||
BEGIN
|
||||
NEW.updated_at = CURRENT_TIMESTAMP;
|
||||
RETURN NEW;
|
||||
END;
|
||||
$$ language 'plpgsql';
|
||||
""")
|
||||
|
||||
op.execute("""
|
||||
CREATE TRIGGER update_instances_updated_at
|
||||
BEFORE UPDATE ON instances
|
||||
FOR EACH ROW
|
||||
EXECUTE FUNCTION update_updated_at_column();
|
||||
""")
|
||||
|
||||
|
||||
def downgrade():
|
||||
op.execute("DROP TRIGGER IF EXISTS update_instances_updated_at ON instances")
|
||||
op.execute("DROP FUNCTION IF EXISTS update_updated_at_column()")
|
||||
op.drop_table('instances')
|
||||
24
migrations/versions/add_status_details.py
Normal file
24
migrations/versions/add_status_details.py
Normal file
@@ -0,0 +1,24 @@
|
||||
"""add status_details column
|
||||
|
||||
Revision ID: add_status_details
|
||||
Revises: add_instances_table
|
||||
Create Date: 2024-03-19 11:00:00.000000
|
||||
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = 'add_status_details'
|
||||
down_revision = 'add_instances_table'
|
||||
branch_labels = None
|
||||
depends_on = None
|
||||
|
||||
|
||||
def upgrade():
|
||||
op.add_column('instances', sa.Column('status_details', sa.Text(), nullable=True))
|
||||
|
||||
|
||||
def downgrade():
|
||||
op.drop_column('instances', 'status_details')
|
||||
Reference in New Issue
Block a user