01fe99da5716_initial.py 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. """initial
  2. Revision ID: 01fe99da5716
  3. Revises:
  4. Create Date: 2018-03-20 10:29:28.864971
  5. """
  6. from alembic import op
  7. import sqlalchemy as sa
  8. # revision identifiers, used by Alembic.
  9. revision = "01fe99da5716"
  10. down_revision = None
  11. branch_labels = None
  12. depends_on = None
  13. def upgrade():
  14. # ### commands auto generated by Alembic - please adjust! ###
  15. op.create_table(
  16. "asset_type",
  17. sa.Column("name", sa.String(length=80), nullable=False),
  18. sa.Column("is_consumer", sa.Boolean(), nullable=False),
  19. sa.Column("is_producer", sa.Boolean(), nullable=False),
  20. sa.Column("can_curtail", sa.Boolean(), nullable=False),
  21. sa.Column("can_shift", sa.Boolean(), nullable=False),
  22. sa.Column("daily_seasonality", sa.Boolean(), nullable=False),
  23. sa.Column("weekly_seasonality", sa.Boolean(), nullable=False),
  24. sa.Column("yearly_seasonality", sa.Boolean(), nullable=False),
  25. sa.PrimaryKeyConstraint("name"),
  26. )
  27. op.create_index(
  28. op.f("ix_asset_type_can_curtail"), "asset_type", ["can_curtail"], unique=False
  29. )
  30. op.create_index(
  31. op.f("ix_asset_type_can_shift"), "asset_type", ["can_shift"], unique=False
  32. )
  33. op.create_table(
  34. "asset",
  35. sa.Column("id", sa.Integer(), nullable=False),
  36. sa.Column("asset_type_name", sa.String(length=80), nullable=False),
  37. sa.Column("name", sa.String(length=80), nullable=True),
  38. sa.Column("display_name", sa.String(length=80), nullable=True),
  39. sa.Column("capacity_in_mw", sa.Float(), nullable=False),
  40. sa.Column("latitude", sa.Float(), nullable=False),
  41. sa.Column("longitude", sa.Float(), nullable=False),
  42. sa.ForeignKeyConstraint(["asset_type_name"], ["asset_type.name"]),
  43. sa.PrimaryKeyConstraint("id"),
  44. sa.UniqueConstraint("display_name"),
  45. sa.UniqueConstraint("name"),
  46. )
  47. # ### end Alembic commands ###
  48. def downgrade():
  49. # ### commands auto generated by Alembic - please adjust! ###
  50. op.drop_table("asset")
  51. op.drop_index(op.f("ix_asset_type_can_shift"), table_name="asset_type")
  52. op.drop_index(op.f("ix_asset_type_can_curtail"), table_name="asset_type")
  53. op.drop_table("asset_type")
  54. # ### end Alembic commands ###