153 lines
7.6 KiB
Python
153 lines
7.6 KiB
Python
"""initial schema
|
|
|
|
Revision ID: 854325b9cbe6
|
|
Revises:
|
|
Create Date: 2026-03-14 13:48:44.859635
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '854325b9cbe6'
|
|
down_revision: Union[str, Sequence[str], None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('proxy_sources',
|
|
sa.Column('url', sa.String(length=2048), nullable=False),
|
|
sa.Column('parser_name', sa.String(length=64), nullable=False),
|
|
sa.Column('cron_schedule', sa.String(length=64), nullable=True),
|
|
sa.Column('default_protocol', postgresql.ENUM('HTTP', 'HTTPS', 'SOCKS4', 'SOCKS5', name='proxy_protocol'), nullable=False),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False),
|
|
sa.Column('last_scraped_at', sa.DateTime(), nullable=True),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('url')
|
|
)
|
|
op.create_table('users',
|
|
sa.Column('email', sa.String(length=320), nullable=False),
|
|
sa.Column('display_name', sa.String(length=128), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('email')
|
|
)
|
|
op.create_table('api_keys',
|
|
sa.Column('user_id', sa.Uuid(), nullable=False),
|
|
sa.Column('key_hash', sa.String(length=128), nullable=False),
|
|
sa.Column('prefix', sa.String(length=8), nullable=False),
|
|
sa.Column('label', sa.String(length=128), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False),
|
|
sa.Column('last_used_at', sa.DateTime(), nullable=True),
|
|
sa.Column('expires_at', sa.DateTime(), nullable=True),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('ix_api_keys_hash', 'api_keys', ['key_hash'], unique=True)
|
|
op.create_index('ix_api_keys_prefix', 'api_keys', ['prefix'], unique=False)
|
|
op.create_table('credit_ledger',
|
|
sa.Column('user_id', sa.Uuid(), nullable=False),
|
|
sa.Column('amount', sa.Integer(), nullable=False),
|
|
sa.Column('tx_type', postgresql.ENUM('PURCHASE', 'ACQUIRE', 'REFUND', 'ADMIN_ADJUST', name='credit_tx_type'), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('reference_id', sa.Uuid(), nullable=True),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('ix_ledger_user_created', 'credit_ledger', ['user_id', 'created_at'], unique=False)
|
|
op.create_table('proxies',
|
|
sa.Column('ip', postgresql.INET(), nullable=False),
|
|
sa.Column('port', sa.Integer(), nullable=False),
|
|
sa.Column('protocol', postgresql.ENUM('HTTP', 'HTTPS', 'SOCKS4', 'SOCKS5', name='proxy_protocol'), nullable=False),
|
|
sa.Column('source_id', sa.Uuid(), nullable=False),
|
|
sa.Column('status', postgresql.ENUM('UNCHECKED', 'ACTIVE', 'DEAD', name='proxy_status'), nullable=False),
|
|
sa.Column('anonymity', postgresql.ENUM('TRANSPARENT', 'ANONYMOUS', 'ELITE', name='anonymity_level'), nullable=True),
|
|
sa.Column('exit_ip', postgresql.INET(), nullable=True),
|
|
sa.Column('country', sa.String(length=2), nullable=True),
|
|
sa.Column('score', sa.Float(), nullable=False),
|
|
sa.Column('avg_latency_ms', sa.Float(), nullable=True),
|
|
sa.Column('uptime_pct', sa.Float(), nullable=True),
|
|
sa.Column('first_seen_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('last_checked_at', sa.DateTime(), nullable=True),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['source_id'], ['proxy_sources.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('ix_proxies_ip_port_proto', 'proxies', ['ip', 'port', 'protocol'], unique=True)
|
|
op.create_index('ix_proxies_status_score', 'proxies', ['status', 'score'], unique=False)
|
|
op.create_table('proxy_checks',
|
|
sa.Column('proxy_id', sa.Uuid(), nullable=False),
|
|
sa.Column('checker_name', sa.String(length=64), nullable=False),
|
|
sa.Column('stage', sa.Integer(), nullable=False),
|
|
sa.Column('passed', sa.Boolean(), nullable=False),
|
|
sa.Column('latency_ms', sa.Float(), nullable=True),
|
|
sa.Column('detail', sa.Text(), nullable=True),
|
|
sa.Column('exit_ip', postgresql.INET(), nullable=True),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['proxy_id'], ['proxies.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('ix_checks_proxy_created', 'proxy_checks', ['proxy_id', 'created_at'], unique=False)
|
|
op.create_table('proxy_leases',
|
|
sa.Column('user_id', sa.Uuid(), nullable=False),
|
|
sa.Column('proxy_id', sa.Uuid(), nullable=False),
|
|
sa.Column('acquired_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('expires_at', sa.DateTime(), nullable=False),
|
|
sa.Column('is_released', sa.Boolean(), nullable=False),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.ForeignKeyConstraint(['proxy_id'], ['proxies.id'], ),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('ix_leases_proxy_active', 'proxy_leases', ['proxy_id', 'is_released'], unique=False)
|
|
op.create_index('ix_leases_user', 'proxy_leases', ['user_id'], unique=False)
|
|
op.create_table('proxy_tags',
|
|
sa.Column('proxy_id', sa.Uuid(), nullable=False),
|
|
sa.Column('key', sa.String(length=64), nullable=False),
|
|
sa.Column('value', sa.String(length=256), nullable=False),
|
|
sa.Column('id', sa.Uuid(), nullable=False),
|
|
sa.ForeignKeyConstraint(['proxy_id'], ['proxies.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('ix_tags_proxy_key', 'proxy_tags', ['proxy_id', 'key'], unique=True)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index('ix_tags_proxy_key', table_name='proxy_tags')
|
|
op.drop_table('proxy_tags')
|
|
op.drop_index('ix_leases_user', table_name='proxy_leases')
|
|
op.drop_index('ix_leases_proxy_active', table_name='proxy_leases')
|
|
op.drop_table('proxy_leases')
|
|
op.drop_index('ix_checks_proxy_created', table_name='proxy_checks')
|
|
op.drop_table('proxy_checks')
|
|
op.drop_index('ix_proxies_status_score', table_name='proxies')
|
|
op.drop_index('ix_proxies_ip_port_proto', table_name='proxies')
|
|
op.drop_table('proxies')
|
|
op.drop_index('ix_ledger_user_created', table_name='credit_ledger')
|
|
op.drop_table('credit_ledger')
|
|
op.drop_index('ix_api_keys_prefix', table_name='api_keys')
|
|
op.drop_index('ix_api_keys_hash', table_name='api_keys')
|
|
op.drop_table('api_keys')
|
|
op.drop_table('users')
|
|
op.drop_table('proxy_sources')
|
|
# ### end Alembic commands ###
|