Skip to content

Commit

Permalink
add user credentials, user credentials group,
Browse files Browse the repository at this point in the history
variable, and secret tables to migration
  • Loading branch information
arash77 committed Dec 19, 2024
1 parent 540d4b6 commit e04aa26
Showing 1 changed file with 90 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,90 @@
"""add user credentials table
Revision ID: 64ee692df15b
Revises: 75348cfb3715
Create Date: 2024-12-19 10:38:04.970502
"""

from sqlalchemy import (
Boolean,
Column,
DateTime,
ForeignKey,
Integer,
)

from galaxy.model.custom_types import TrimmedString
from galaxy.model.database_object_names import build_foreign_key_name
from galaxy.model.migrations.util import (
add_column,
create_table,
drop_column,
drop_constraint,
drop_table,
)

# revision identifiers, used by Alembic.
revision = "64ee692df15b"
down_revision = "75348cfb3715"
branch_labels = None
depends_on = None


user_credentials_table = "user_credentials"
user_credentials_group_table = "user_credentials_group"
variable_table = "variable"
secret_table = "secret"


def upgrade():
create_table(
user_credentials_table,
Column("id", Integer, primary_key=True),
Column("user_id", Integer, ForeignKey("galaxy_user.id"), index=True),
Column("reference", TrimmedString(255)),
Column("source_type", TrimmedString(255)),
Column("source_id", TrimmedString(255)),
Column("create_time", DateTime),
Column("update_time", DateTime),
)
create_table(
user_credentials_group_table,
Column("id", Integer, primary_key=True),
Column("name", TrimmedString(255)),
Column("user_credentials_id", Integer, ForeignKey("user_credentials.id"), index=True),
Column("create_time", DateTime),
Column("update_time", DateTime),
)
add_column(
user_credentials_table,
Column("current_group_id", Integer, ForeignKey("user_credentials_group.id"), index=True, nullable=True),
)
create_table(
variable_table,
Column("id", Integer, primary_key=True),
Column("user_credential_group_id", Integer, ForeignKey("user_credentials_group.id"), index=True),
Column("name", TrimmedString(255)),
Column("value", TrimmedString(255)),
Column("create_time", DateTime),
Column("update_time", DateTime),
)
create_table(
secret_table,
Column("id", Integer, primary_key=True),
Column("user_credential_group_id", Integer, ForeignKey("user_credentials_group.id"), index=True),
Column("name", TrimmedString(255)),
Column("already_set", Boolean),
Column("create_time", DateTime),
Column("update_time", DateTime),
)


def downgrade():
drop_constraint(build_foreign_key_name(user_credentials_table, "current_group_id"), user_credentials_table)
drop_column(user_credentials_table, "current_group_id")

drop_table(variable_table)
drop_table(secret_table)
drop_table(user_credentials_group_table)
drop_table(user_credentials_table)

0 comments on commit e04aa26

Please sign in to comment.