diff --git a/migrations/versions/add_docupulse_settings_table.py b/migrations/versions/add_docupulse_settings_table.py index 897f87b..dadf320 100644 --- a/migrations/versions/add_docupulse_settings_table.py +++ b/migrations/versions/add_docupulse_settings_table.py @@ -26,6 +26,14 @@ def upgrade(): type_=sa.BigInteger(), existing_nullable=False, server_default='10737418240') + + # Check if we need to insert default data + result = conn.execute("SELECT COUNT(*) FROM docupulse_settings").scalar() + if result == 0: + conn.execute(""" + INSERT INTO docupulse_settings (id, max_rooms, max_conversations, max_storage, updated_at) + VALUES (1, 10, 10, 10737418240, CURRENT_TIMESTAMP) + """) else: # Create new table op.create_table('docupulse_settings',