mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-06-12 17:10:54 +02:00
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
"""create usage reports table
|
|
|
|
Revision ID: bc9771dccadf
|
|
Revises: 48d14957fe80
|
|
Create Date: 2024-06-18 10:04:26.800282
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
import fastapi_users_db_sqlalchemy
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "bc9771dccadf"
|
|
down_revision = "0568ccf46a6b"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"usage_reports",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("report_name", sa.String(), nullable=False),
|
|
sa.Column(
|
|
"requestor_user_id",
|
|
fastapi_users_db_sqlalchemy.generics.GUID(),
|
|
nullable=True,
|
|
),
|
|
sa.Column(
|
|
"time_created",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("now()"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("period_from", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("period_to", sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(
|
|
["report_name"],
|
|
["file_store.file_name"],
|
|
),
|
|
sa.ForeignKeyConstraint(
|
|
["requestor_user_id"],
|
|
["user.id"],
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("usage_reports")
|