Script 465202bfb9ce_update_db_functions_py
[hide private]
[frames] | no frames]

Source Code for Script script-465202bfb9ce_update_db_functions_py

 1  """update_db_functions 
 2   
 3  Revision ID: 465202bfb9ce 
 4  Revises: 26bf5b9a4dd0 
 5  Create Date: 2018-01-24 16:38:43.500159 
 6   
 7  """ 
 8   
 9  # revision identifiers, used by Alembic. 
10  revision = '465202bfb9ce' 
11  down_revision = '26bf5b9a4dd0' 
12   
13  from alembic import op 
14  import sqlalchemy as sa 
15   
16   
17 -def upgrade():
18 query_functions = """ 19 CREATE OR REPLACE FUNCTION status_to_order (x integer) 20 RETURNS integer AS $$ BEGIN 21 RETURN CASE WHEN x = 3 THEN 1 22 WHEN x = 6 THEN 2 23 WHEN x = 7 THEN 3 24 WHEN x = 4 THEN 4 25 WHEN x = 0 THEN 5 26 WHEN x = 1 THEN 6 27 WHEN x = 5 THEN 7 28 WHEN x = 2 THEN 8 29 WHEN x = 8 THEN 9 30 WHEN x = 9 THEN 10 31 ELSE x 32 END; END; 33 $$ LANGUAGE plpgsql; 34 35 CREATE OR REPLACE FUNCTION order_to_status (x integer) 36 RETURNS integer AS $$ BEGIN 37 RETURN CASE WHEN x = 1 THEN 3 38 WHEN x = 2 THEN 6 39 WHEN x = 3 THEN 7 40 WHEN x = 4 THEN 4 41 WHEN x = 5 THEN 0 42 WHEN x = 6 THEN 1 43 WHEN x = 7 THEN 5 44 WHEN x = 8 THEN 2 45 WHEN x = 9 THEN 8 46 WHEN x = 10 THEN 9 47 ELSE x 48 END; END; 49 $$ LANGUAGE plpgsql; 50 """ 51 op.execute(sa.text(query_functions))
52 53
54 -def downgrade():
55 query_functions = """ 56 CREATE OR REPLACE FUNCTION status_to_order (x integer) 57 RETURNS integer AS $$ BEGIN 58 RETURN CASE WHEN x = 3 THEN 1 59 WHEN x = 6 THEN 2 60 WHEN x = 7 THEN 3 61 WHEN x = 4 THEN 4 62 WHEN x = 0 THEN 5 63 WHEN x = 1 THEN 6 64 WHEN x = 5 THEN 7 65 WHEN x = 2 THEN 8 66 ELSE x 67 END; END; 68 $$ LANGUAGE plpgsql; 69 70 CREATE OR REPLACE FUNCTION order_to_status (x integer) 71 RETURNS integer AS $$ BEGIN 72 RETURN CASE WHEN x = 1 THEN 3 73 WHEN x = 2 THEN 6 74 WHEN x = 3 THEN 7 75 WHEN x = 4 THEN 4 76 WHEN x = 5 THEN 0 77 WHEN x = 6 THEN 1 78 WHEN x = 7 THEN 5 79 WHEN x = 8 THEN 2 80 ELSE x 81 END; END; 82 $$ LANGUAGE plpgsql; 83 """ 84 op.execute(sa.text(query_functions))
85