84c7aeb1a5
<!-- .github/pull_request_template.md --> ## Description Merge changes on main to dev ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
33 lines
809 B
Python
33 lines
809 B
Python
"""Add default user
|
|
|
|
Revision ID: 482cd6517ce4
|
|
Revises: 8057ae7329c2
|
|
Create Date: 2024-10-16 22:17:18.634638
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
from sqlalchemy.util import await_only
|
|
|
|
from cognee.modules.users.methods import create_default_user, delete_user
|
|
|
|
from fastapi_users.exceptions import UserAlreadyExists
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "482cd6517ce4"
|
|
down_revision: Union[str, None] = "8057ae7329c2"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = "8057ae7329c2"
|
|
|
|
|
|
def upgrade() -> None:
|
|
try:
|
|
await_only(create_default_user())
|
|
except UserAlreadyExists:
|
|
pass # It's fine if the default user already exists
|
|
|
|
|
|
def downgrade() -> None:
|
|
await_only(delete_user("default_user@example.com"))
|