|
| 1 | +"""rework credentials |
| 2 | +
|
| 3 | +Revision ID: 39ddd3ca3a20 |
| 4 | +Revises: 618a59151523 |
| 5 | +Create Date: 2025-02-18 15:17:51.883711+00:00 |
| 6 | +
|
| 7 | +""" |
| 8 | +from alembic import op |
| 9 | +import sqlalchemy as sa |
| 10 | + |
| 11 | + |
| 12 | +# revision identifiers, used by Alembic. |
| 13 | +revision = '39ddd3ca3a20' |
| 14 | +down_revision = '615a6fdd9af4' |
| 15 | +branch_labels = None |
| 16 | +depends_on = None |
| 17 | + |
| 18 | + |
| 19 | +def upgrade(): |
| 20 | + |
| 21 | + op.execute('DROP TABLE IF EXISTS credential') |
| 22 | + op.create_table('credential', |
| 23 | + sa.Column('id', sa.Integer(), nullable=False, autoincrement=True, primary_key=True), |
| 24 | + sa.Column('password', sa.Text(), nullable=False), |
| 25 | + sa.Column('username', sa.Text(), nullable=False), |
| 26 | + sa.Column('endpoint', sa.Text(), nullable=False, server_default=''), |
| 27 | + sa.Column('leak_date', sa.DateTime(), nullable=True), |
| 28 | + sa.Column('owned', sa.Boolean(), nullable=False, server_default='false'), |
| 29 | + sa.Column('workspace_id', sa.Integer(), nullable=False), |
| 30 | + sa.Column('create_date', sa.DateTime(), nullable=False), |
| 31 | + sa.Column('update_date', sa.DateTime(), nullable=False), |
| 32 | + sa.Column('creator_id', sa.Integer(), nullable=True), |
| 33 | + sa.Column('update_user_id', sa.Integer(), nullable=True), |
| 34 | + sa.ForeignKeyConstraint(['creator_id'], ['faraday_user.id'], ondelete='SET NULL'), |
| 35 | + sa.ForeignKeyConstraint(['update_user_id'], ['faraday_user.id'], ondelete='SET NULL'), |
| 36 | + sa.ForeignKeyConstraint(['workspace_id'], ['workspace.id'], ondelete='CASCADE'), |
| 37 | + sa.UniqueConstraint('username', 'password', 'endpoint', 'workspace_id', |
| 38 | + name='uix_credential_username_password_endpoint_workspace'), |
| 39 | + sa.CheckConstraint("username != ''", name="check_username_not_empty"), |
| 40 | + sa.CheckConstraint("password != ''", name="check_password_not_empty") |
| 41 | + ) |
| 42 | + |
| 43 | + op.create_index('ix_credential_leak_date', 'credential', ['leak_date']) |
| 44 | + op.create_index('ix_credential_leak_date_workspace_id', 'credential', ['workspace_id', 'leak_date']) |
| 45 | + |
| 46 | + op.create_table('association_table_vulnerabilities_credentials', |
| 47 | + sa.Column('vulnerability_id', sa.Integer(), nullable=False), |
| 48 | + sa.Column('credential_id', sa.Integer(), nullable=False), |
| 49 | + sa.ForeignKeyConstraint(['credential_id'], ['credential.id'], ondelete='CASCADE'), |
| 50 | + sa.ForeignKeyConstraint(['vulnerability_id'], ['vulnerability.id'], ondelete='CASCADE') |
| 51 | + ) |
| 52 | + |
| 53 | + op.create_index('ix_association_vuln_creds_vuln_id', |
| 54 | + 'association_table_vulnerabilities_credentials', |
| 55 | + ['vulnerability_id']) |
| 56 | + |
| 57 | + op.create_index('ix_association_vuln_creds_cred_id', |
| 58 | + 'association_table_vulnerabilities_credentials', |
| 59 | + ['credential_id']) |
| 60 | + |
| 61 | + |
| 62 | +def downgrade(): |
| 63 | + op.drop_index('ix_association_vuln_creds_vuln_id', |
| 64 | + table_name='association_table_vulnerabilities_credentials') |
| 65 | + op.drop_index('ix_association_vuln_creds_cred_id', |
| 66 | + table_name='association_table_vulnerabilities_credentials') |
| 67 | + |
| 68 | + op.drop_table('association_table_vulnerabilities_credentials') |
| 69 | + |
| 70 | + op.drop_index('ix_credential_leak_date_workspace_id', table_name='credential') |
| 71 | + op.drop_index('ix_credential_leak_date', table_name='credential') |
| 72 | + op.drop_table('credential') |
| 73 | + # Recreate the old table to restore the previous schema during downgrade |
| 74 | + op.create_table('credential', |
| 75 | + sa.Column('id', sa.Integer(), nullable=False), |
| 76 | + sa.Column('username', sa.Text(), nullable=False), |
| 77 | + sa.Column('password', sa.Text(), nullable=False), |
| 78 | + sa.Column('description', sa.Text(), nullable=False), |
| 79 | + sa.Column('name', sa.Text(), nullable=True), |
| 80 | + sa.Column('host_id', sa.Integer(), nullable=True), |
| 81 | + sa.Column('service_id', sa.Integer(), nullable=True), |
| 82 | + sa.Column('workspace_id', sa.Integer(), nullable=False), |
| 83 | + sa.ForeignKeyConstraint(['host_id'], ['host.id'], ondelete='CASCADE'), |
| 84 | + sa.ForeignKeyConstraint(['service_id'], ['service.id'], ondelete='CASCADE'), |
| 85 | + sa.ForeignKeyConstraint(['workspace_id'], ['workspace.id'], ondelete='CASCADE'), |
| 86 | + sa.PrimaryKeyConstraint('id'), |
| 87 | + sa.UniqueConstraint( |
| 88 | + 'username', |
| 89 | + 'host_id', |
| 90 | + 'service_id', |
| 91 | + 'workspace_id', |
| 92 | + name='uix_credential_username_host_service_workspace' |
| 93 | + ) |
| 94 | + ) |
0 commit comments