Files
vpn/alembic/versions/0002_ssh_fields.py
T

33 lines
1.1 KiB
Python

"""Add SSH connection fields to vpn_servers
Revision ID: 0002_ssh_fields
Revises: 0001_initial
Create Date: 2026-07-25
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
revision: str = "0002_ssh_fields"
down_revision: Union[str, None] = "0001_initial"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.add_column("vpn_servers", sa.Column("ssh_host", sa.String(length=255), nullable=True))
op.add_column("vpn_servers", sa.Column("ssh_port", sa.Integer(), nullable=False, server_default="22"))
op.add_column("vpn_servers", sa.Column("ssh_username", sa.String(length=128), nullable=True))
op.add_column("vpn_servers", sa.Column("ssh_password", sa.Text(), nullable=True))
op.add_column("vpn_servers", sa.Column("ssh_private_key", sa.Text(), nullable=True))
def downgrade() -> None:
op.drop_column("vpn_servers", "ssh_private_key")
op.drop_column("vpn_servers", "ssh_password")
op.drop_column("vpn_servers", "ssh_username")
op.drop_column("vpn_servers", "ssh_port")
op.drop_column("vpn_servers", "ssh_host")