gitscope
https://dev.kronoz.win/r/marzneshin.git
app/db/migrations/versions/058eb11c5b17_user_reset_in_table.py · v0.5.0
Source
1 5b5d739 Dawsh 2y ago """hold user last reset and lifetime_used_traffic inside the table
2 42a1073 Dawsh 2y ago
3 42a1073 Dawsh 2y ago Revision ID: 058eb11c5b17
4 42a1073 Dawsh 2y ago Revises: 74e234a3bd4b
5 42a1073 Dawsh 2y ago Create Date: 2024-05-26 21:16:06.762089
6 42a1073 Dawsh 2y ago
7 42a1073 Dawsh 2y ago """
8 42a1073 Dawsh 2y ago
9 42a1073 Dawsh 2y ago import sqlalchemy as sa
10 42a1073 Dawsh 2y ago from alembic import op
11 42a1073 Dawsh 2y ago
12 42a1073 Dawsh 2y ago # revision identifiers, used by Alembic.
13 42a1073 Dawsh 2y ago revision = "058eb11c5b17"
14 42a1073 Dawsh 2y ago down_revision = "74e234a3bd4b"
15 42a1073 Dawsh 2y ago branch_labels = None
16 42a1073 Dawsh 2y ago depends_on = None
17 42a1073 Dawsh 2y ago
18 42a1073 Dawsh 2y ago
19 42a1073 Dawsh 2y ago def upgrade() -> None:
20 42a1073 Dawsh 2y ago # ### commands auto generated by Alembic - please adjust! ###
21 42a1073 Dawsh 2y ago op.drop_table("user_usage_logs")
22 42a1073 Dawsh 2y ago op.add_column(
23 49457ba Dawsh 2y ago "users",
24 49457ba Dawsh 2y ago sa.Column(
25 49457ba Dawsh 2y ago "lifetime_used_traffic",
26 49457ba Dawsh 2y ago sa.BigInteger(),
27 8ee2242 Dawsh 2y ago server_default="0",
28 8ee2242 Dawsh 2y ago nullable=False,
29 49457ba Dawsh 2y ago ),
30 49457ba Dawsh 2y ago )
31 49457ba Dawsh 2y ago op.add_column(
32 49457ba Dawsh 2y ago "users",
33 49457ba Dawsh 2y ago sa.Column(
34 49457ba Dawsh 2y ago "traffic_reset_at",
35 49457ba Dawsh 2y ago sa.DateTime(),
36 49457ba Dawsh 2y ago server_default=sa.sql.null(),
37 49457ba Dawsh 2y ago nullable=True,
38 49457ba Dawsh 2y ago ),
39 42a1073 Dawsh 2y ago )
40 42a1073 Dawsh 2y ago # ### end Alembic commands ###
41 42a1073 Dawsh 2y ago
42 42a1073 Dawsh 2y ago
43 42a1073 Dawsh 2y ago def downgrade() -> None:
44 42a1073 Dawsh 2y ago # ### commands auto generated by Alembic - please adjust! ###
45 42a1073 Dawsh 2y ago op.drop_column("users", "traffic_reset_at")
46 42a1073 Dawsh 2y ago op.drop_column("users", "lifetime_used_traffic")
47 42a1073 Dawsh 2y ago op.create_table(
48 42a1073 Dawsh 2y ago "user_usage_logs",
49 42a1073 Dawsh 2y ago sa.Column("id", sa.INTEGER(), nullable=False),
50 42a1073 Dawsh 2y ago sa.Column("user_id", sa.INTEGER(), nullable=True),
51 42a1073 Dawsh 2y ago sa.Column("used_traffic_at_reset", sa.BIGINT(), nullable=False),
52 42a1073 Dawsh 2y ago sa.Column("reset_at", sa.DATETIME(), nullable=True),
53 42a1073 Dawsh 2y ago sa.ForeignKeyConstraint(
54 42a1073 Dawsh 2y ago ["user_id"],
55 42a1073 Dawsh 2y ago ["users.id"],
56 42a1073 Dawsh 2y ago ),
57 42a1073 Dawsh 2y ago sa.PrimaryKeyConstraint("id"),
58 42a1073 Dawsh 2y ago )
59 42a1073 Dawsh 2y ago # ### end Alembic commands ###