gitscope
/
← marzneshin / app/db/migrations/versions/20241129_a015ed8ce227_add_shadowtls_protocol.py / blame
Files Commits 446 Branches 1 Tags 43 Markers 4 README License
https://dev.kronoz.win/r/marzneshin.git
app/db/migrations/versions/20241129_a015ed8ce227_add_shadowtls_protocol.py · master
Source
1 70b35e2 Dawsh 1y ago """add shadowtls protocol
2 70b35e2 Dawsh 1y ago
3 70b35e2 Dawsh 1y ago Revision ID: a015ed8ce227
4 70b35e2 Dawsh 1y ago Revises: be0032100c07
5 70b35e2 Dawsh 1y ago Create Date: 2024-11-29 21:41:06.896405
6 70b35e2 Dawsh 1y ago
7 70b35e2 Dawsh 1y ago """
8 70b35e2 Dawsh 1y ago
9 70b35e2 Dawsh 1y ago from alembic import op
10 70b35e2 Dawsh 1y ago
11 70b35e2 Dawsh 1y ago # revision identifiers, used by Alembic.
12 70b35e2 Dawsh 1y ago revision = "a015ed8ce227"
13 70b35e2 Dawsh 1y ago down_revision = "be0032100c07"
14 70b35e2 Dawsh 1y ago branch_labels = None
15 70b35e2 Dawsh 1y ago depends_on = None
16 70b35e2 Dawsh 1y ago
17 70b35e2 Dawsh 1y ago
18 70b35e2 Dawsh 1y ago def upgrade() -> None:
19 70b35e2 Dawsh 1y ago bind = op.get_bind()
20 70b35e2 Dawsh 1y ago dialect = bind.dialect.name
21 70b35e2 Dawsh 1y ago
22 70b35e2 Dawsh 1y ago if dialect == "postgresql":
23 70b35e2 Dawsh 1y ago op.execute("ALTER TYPE proxytypes ADD VALUE 'ShadowTLS'")
24 70b35e2 Dawsh 1y ago
25 70b35e2 Dawsh 1y ago elif dialect in ("mariadb", "mysql"):
26 70b35e2 Dawsh 1y ago op.execute(
27 70b35e2 Dawsh 1y ago "ALTER TABLE inbounds MODIFY protocol ENUM('VMess', 'VLESS', 'Trojan', 'Shadowsocks', 'Shadowsocks2022', 'Hysteria2', 'WireGuard', 'TUIC', 'ShadowTLS')"
28 70b35e2 Dawsh 1y ago )
29 70b35e2 Dawsh 1y ago
30 70b35e2 Dawsh 1y ago
31 70b35e2 Dawsh 1y ago def downgrade() -> None:
32 70b35e2 Dawsh 1y ago bind = op.get_bind()
33 70b35e2 Dawsh 1y ago dialect = bind.dialect.name
34 70b35e2 Dawsh 1y ago
35 70b35e2 Dawsh 1y ago if dialect in ("mysql", "mariadb"):
36 70b35e2 Dawsh 1y ago op.execute(
37 70b35e2 Dawsh 1y ago "ALTER TABLE inbounds MODIFY protocol ENUM('VMess', 'VLESS', 'Trojan', 'Shadowsocks', 'Shadowsocks2022', 'Hysteria2', 'WireGuard', 'TUIC')"
38 70b35e2 Dawsh 1y ago )