1111from alembic import op
1212import sqlalchemy as sa
1313
14-
1514revision : str = "6b3b2a7f804c"
1615down_revision : Union [str , Sequence [str ], None ] = None
1716branch_labels : Union [str , Sequence [str ], None ] = None
@@ -23,7 +22,9 @@ def upgrade() -> None:
2322 "waitlist" ,
2423 sa .Column ("id" , sa .Integer (), primary_key = True ),
2524 sa .Column ("email" , sa .String (length = 255 ), nullable = False ),
26- sa .Column ("source" , sa .String (length = 100 ), nullable = False , server_default = "landing" ),
25+ sa .Column (
26+ "source" , sa .String (length = 100 ), nullable = False , server_default = "landing"
27+ ),
2728 sa .Column (
2829 "created_at" ,
2930 sa .TIMESTAMP (timezone = True ),
@@ -60,7 +61,9 @@ def upgrade() -> None:
6061 DO $$
6162 BEGIN
6263 IF NOT EXISTS (
63- SELECT 1 FROM pg_policies WHERE schemaname = 'public' AND tablename = 'waitlist' AND policyname = 'Allow public insert on waitlist'
64+ SELECT 1 FROM pg_policies WHERE schemaname = 'public'
65+ AND tablename = 'waitlist'
66+ AND policyname = 'Allow public insert on waitlist'
6467 ) THEN
6568 CREATE POLICY "Allow public insert on waitlist"
6669 ON waitlist FOR INSERT
@@ -75,7 +78,9 @@ def upgrade() -> None:
7578 DO $$
7679 BEGIN
7780 IF NOT EXISTS (
78- SELECT 1 FROM pg_policies WHERE schemaname = 'public' AND tablename = 'waitlist' AND policyname = 'Allow public select count on waitlist'
81+ SELECT 1 FROM pg_policies WHERE schemaname = 'public'
82+ AND tablename = 'waitlist'
83+ AND policyname = 'Allow public select count on waitlist'
7984 ) THEN
8085 CREATE POLICY "Allow public select count on waitlist"
8186 ON waitlist FOR SELECT
@@ -90,7 +95,9 @@ def upgrade() -> None:
9095 DO $$
9196 BEGIN
9297 IF NOT EXISTS (
93- SELECT 1 FROM pg_policies WHERE schemaname = 'public' AND tablename = 'support' AND policyname = 'Allow public insert on support'
98+ SELECT 1 FROM pg_policies WHERE schemaname = 'public'
99+ AND tablename = 'support'
100+ AND policyname = 'Allow public insert on support'
94101 ) THEN
95102 CREATE POLICY "Allow public insert on support"
96103 ON support FOR INSERT
@@ -105,7 +112,9 @@ def upgrade() -> None:
105112 DO $$
106113 BEGIN
107114 IF NOT EXISTS (
108- SELECT 1 FROM pg_policies WHERE schemaname = 'public' AND tablename = 'support' AND policyname = 'Allow public select on support'
115+ SELECT 1 FROM pg_policies WHERE schemaname = 'public'
116+ AND tablename = 'support'
117+ AND policyname = 'Allow public select on support'
109118 ) THEN
110119 CREATE POLICY "Allow public select on support"
111120 ON support FOR SELECT
@@ -134,10 +143,12 @@ def upgrade() -> None:
134143def downgrade () -> None :
135144 op .execute ("DROP FUNCTION IF EXISTS public.waitlist_count();" )
136145
137- op .execute ("DROP POLICY IF EXISTS \" Allow public insert on support\" ON support;" )
138- op .execute ("DROP POLICY IF EXISTS \" Allow public select on support\" ON support;" )
139- op .execute ("DROP POLICY IF EXISTS \" Allow public insert on waitlist\" ON waitlist;" )
140- op .execute ("DROP POLICY IF EXISTS \" Allow public select count on waitlist\" ON waitlist;" )
146+ op .execute ('DROP POLICY IF EXISTS "Allow public insert on support" ON support;' )
147+ op .execute ('DROP POLICY IF EXISTS "Allow public select on support" ON support;' )
148+ op .execute ('DROP POLICY IF EXISTS "Allow public insert on waitlist" ON waitlist;' )
149+ op .execute (
150+ 'DROP POLICY IF EXISTS "Allow public select count on waitlist" ON waitlist;'
151+ )
141152
142153 op .drop_index ("ix_support_status" , table_name = "support" )
143154 op .drop_index ("ix_support_email" , table_name = "support" )
0 commit comments