text: #
  # Copyright (c) 2021 the Hugging Face team.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.#
  """add_keys
  Revision ID: ba788d7c81bf
  Revises: abc051ececd5
  Create Date: 2021-05-03 15:29:01.414138
  """
  import sqlalchemy as sa
  from alembic import op
  
  
  # revision identifiers, used by Alembic
  revision = "ba788d7c81bf"
  down_revision = "abc051ececd5"
  branch_labels = None
  depends_on = None
  
  
  def upgrade() -> None:
      op.add_column("whitelist", sa.Column("peer_public_key", sa.Text))
      op.add_column("experiments", sa.Column("auth_server_public_key", sa.LargeBinary))
      op.add_column("experiments", sa.Column("auth_server_private_key", sa.LargeBinary))
  
  
  def downgrade() -> None:
      op.drop_column("whitelist", "peer_public_key")
      op.drop_column("experiments", "auth_server_public_key")
      op.drop_column("experiments", "auth_server_private_key")
  