Try fixing migrations 5: remove pricetype error

Author: JoshD94

migrations/versions/eb948c31a342_create_gear_table.py

@@ -7,9 +7,7 @@
 """
 from alembic import op
 import sqlalchemy as sa
-
-# Import the PriceType enum used in the model so the Enum can be created
-from src.models.activity import PriceType
+from sqlalchemy.dialects import postgresql
 
 
 # revision identifiers, used by Alembic.
@@ -18,6 +16,9 @@
 branch_labels = None
 depends_on = None
 
+### Ensures alembic does not try to create enum
+price_type_enum = postgresql.ENUM('rate', 'gear', name='pricetype', create_type=False)
+
 
 def upgrade():
     op.create_table(
@@ -27,7 +28,7 @@ def upgrade():
         sa.Column("name", sa.String(), nullable=False),
         sa.Column("cost", sa.Float(), nullable=False),
         sa.Column("rate", sa.String(), nullable=True),
-        sa.Column("type", sa.Enum(PriceType, create_type=False), nullable=False),
+        sa.Column("type", price_type_enum, nullable=False),
     )
 
 def downgrade():