diff --git a/sapphire/storage/database/migrations/env.py b/sapphire/storage/database/migrations/env.py index 522b68f8..00f8edaa 100644 --- a/sapphire/storage/database/migrations/env.py +++ b/sapphire/storage/database/migrations/env.py @@ -1,12 +1,11 @@ import asyncio from logging.config import fileConfig +from alembic import context from sqlalchemy import pool from sqlalchemy.engine import Connection from sqlalchemy.ext.asyncio import async_engine_from_config -from alembic import context - # this is the Alembic Config object, which provides # access to the values within the .ini file in use. config = context.config diff --git a/sapphire/storage/database/migrations/versions/f49b1a9878de_initial_migration.py b/sapphire/storage/database/migrations/versions/f49b1a9878de_initial_migration.py index 702379f3..958291f3 100644 --- a/sapphire/storage/database/migrations/versions/f49b1a9878de_initial_migration.py +++ b/sapphire/storage/database/migrations/versions/f49b1a9878de_initial_migration.py @@ -7,9 +7,8 @@ """ from typing import Sequence, Union -from alembic import op import sqlalchemy as sa - +from alembic import op # revision identifiers, used by Alembic. revision: str = 'f49b1a9878de' diff --git a/sapphire/storage/database/models.py b/sapphire/storage/database/models.py index 2d49fd3f..14221a73 100644 --- a/sapphire/storage/database/models.py +++ b/sapphire/storage/database/models.py @@ -2,9 +2,10 @@ from datetime import datetime from typing import Literal, get_args -from sqlalchemy import Enum, ForeignKey, False_, Null +from sqlalchemy import Enum, False_, ForeignKey, Null from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column + class Base(DeclarativeBase): pass