little update

This commit is contained in:
2026-08-12 21:11:39 +03:00
parent e3d5de12fe
commit 3af436d48a
37 changed files with 2299 additions and 9 deletions

48
alembic/env.py Normal file
View File

@@ -0,0 +1,48 @@
from logging.config import fileConfig
import os
from pathlib import Path
import sys
from alembic import context
from dotenv import load_dotenv
from sqlalchemy import engine_from_config, pool
PROJECT_ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(PROJECT_ROOT / "src"))
load_dotenv(PROJECT_ROOT / ".env")
from database.models import Base
config = context.config
if database_url := os.getenv("DATABASE_URL"):
config.set_main_option("sqlalchemy.url", database_url)
if config.config_file_name:
fileConfig(config.config_file_name)
target_metadata = Base.metadata
def run_migrations_offline() -> None:
context.configure(
url=config.get_main_option("sqlalchemy.url"),
target_metadata=target_metadata,
literal_binds=True,
dialect_opts={"paramstyle": "named"},
)
with context.begin_transaction():
context.run_migrations()
def run_migrations_online() -> None:
connectable = engine_from_config(
config.get_section(config.config_ini_section, {}), prefix="sqlalchemy.", poolclass=pool.NullPool
)
with connectable.connect() as connection:
context.configure(connection=connection, target_metadata=target_metadata)
with context.begin_transaction():
context.run_migrations()
if context.is_offline_mode():
run_migrations_offline()
else:
run_migrations_online()

View File

@@ -0,0 +1,32 @@
"""create network pool
Revision ID: 20260812_01
Revises:
Create Date: 2026-08-12
"""
from alembic import op
import sqlalchemy as sa
revision = "20260812_01"
down_revision = None
branch_labels = None
depends_on = None
def upgrade() -> None:
op.create_table(
"network_pool",
sa.Column("cidr", sa.String(length=18), nullable=False),
sa.Column("cloud_subnet_id", sa.String(length=64), nullable=False),
sa.Column("game_id", sa.Uuid(), nullable=True),
sa.Column("allocated_at", sa.DateTime(timezone=True), nullable=True),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("CURRENT_TIMESTAMP"), nullable=False),
sa.PrimaryKeyConstraint("cidr"),
sa.UniqueConstraint("cloud_subnet_id"),
sa.UniqueConstraint("game_id"),
)
def downgrade() -> None:
op.drop_table("network_pool")