first commit

db/versions/3cb6ab91fdc2_update_inverter_models_enum.py

@@ -0,0 +1,37 @@
+"""Update Inverter models enum
+
+Revision ID: 3cb6ab91fdc2
+Revises: f90d04c490dc
+Create Date: 2017-06-26 12:41:17.882779
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '3cb6ab91fdc2'
+down_revision = 'f90d04c490dc'
+branch_labels = None
+depends_on = None
+
+from alembic import op
+from helix.constants.inverter_type import InverterType
+
+
+def upgrade():
+    op.execute('COMMIT')
+
+    connection = None
+    if not op.get_context().as_sql:
+        connection = op.get_bind()
+        connection.execution_options(isolation_level='AUTOCOMMIT')
+
+    op.execute('''ALTER TYPE "InverterType" ADD VALUE IF NOT EXISTS \'%s\'''' % (str(InverterType.DELTA.MODEL_36KW.value), ))
+    op.execute('''ALTER TYPE "InverterType" ADD VALUE IF NOT EXISTS \'%s\'''' % (str(InverterType.DELTA.MODEL_42KW.value), ))
+    op.execute('''ALTER TYPE "InverterType" ADD VALUE IF NOT EXISTS \'%s\'''' % (str(InverterType.DELTA.MODEL_60KW.value), ))
+    op.execute('''ALTER TYPE "InverterType" ADD VALUE IF NOT EXISTS \'%s\'''' % (str(InverterType.DELTA.MODEL_80KW.value), ))
+
+    if connection is not None:
+        connection.execution_options(isolation_level='READ_COMMITTED')
+
+
+def downgrade():
+    pass