Merge pull request #191 from cuappdev/sophie/popular-times

Popular times updated migrations

Author: TushigBili

migrations/alembic.ini

@@ -1,6 +1,7 @@
 # A generic, single database configuration.
 
 [alembic]
+script_location = migrations
 # template used to generate migration files
 # file_template = %%(rev)s_%%(slug)s
 

migrations/env.py

@@ -1,4 +1,8 @@
 from __future__ import with_statement
+import sys
+import os
+sys.path.append(os.path.abspath(os.path.dirname(__file__) + "/..")) 
+from app import app
 
 import logging
 from logging.config import fileConfig
@@ -21,10 +25,9 @@
 # for 'autogenerate' support
 # from myapp import mymodel
 # target_metadata = mymodel.Base.metadata
-from flask import current_app
-config.set_main_option('sqlalchemy.url',
-                       current_app.config.get('SQLALCHEMY_DATABASE_URI'))
-target_metadata = current_app.extensions['migrate'].db.metadata
+with app.app_context():
+    config.set_main_option('sqlalchemy.url', app.config['SQLALCHEMY_DATABASE_URI'])
+    target_metadata = app.extensions['migrate'].db.metadata
 
 # other values from the config, defined by the needs of env.py,
 # can be acquired:
@@ -82,7 +85,7 @@ def process_revision_directives(context, revision, directives):
             connection=connection,
             target_metadata=target_metadata,
             process_revision_directives=process_revision_directives,
-            **current_app.extensions['migrate'].configure_args
+            **app.extensions['migrate'].configure_args
         )
 
         with context.begin_transaction():
@@ -92,4 +95,4 @@ def process_revision_directives(context, revision, directives):
 if context.is_offline_mode():
     run_migrations_offline()
 else:
-    run_migrations_online()
+    run_migrations_online()

migrations/versions/31b1fa20772f_popular_times.py

@@ -0,0 +1,56 @@
+"""popular times
+
+Revision ID: 31b1fa20772f
+Revises: 
+Create Date: 2025-02-28 20:57:19.922403
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+
+# revision identifiers, used by Alembic.
+revision = '31b1fa20772f'
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+### Ensures alembic does not try to create enum
+day_of_week_enum = postgresql.ENUM(
+    'Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday',
+    name='dayofweekenum', create_type=False
+)
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('gear', 'cost',
+               existing_type=postgresql.DOUBLE_PRECISION(precision=53),
+               nullable=0)
+    
+    ### Alembic is running command multiple times.
+    op.execute("""
+    DO $$ 
+    BEGIN 
+        IF NOT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'hourly_average_capacity') THEN 
+            CREATE TABLE hourly_average_capacity (
+                id SERIAL PRIMARY KEY, 
+                facility_id INTEGER NOT NULL REFERENCES facility(id), 
+                average_percent FLOAT NOT NULL, 
+                hour_of_day INTEGER NOT NULL, 
+                day_of_week dayofweekenum, 
+                history NUMERIC[] DEFAULT '{}' NOT NULL
+            );
+        END IF; 
+    END $$;
+    """)
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    
+    op.drop_table('hourly_average_capacity')
+    # ### end Alembic commands ###
+