Script 4b57794e2b5_fix_stored_procudure_status_order_bug_py
[hide private]
[frames] | no frames]

Source Code for Script script-4b57794e2b5_fix_stored_procudure_status_order_bug_py

 1  """fix stored procudure (status order)  bug 
 2   
 3  Revision ID: 4b57794e2b5 
 4  Revises: 19ca0c14096e 
 5  Create Date: 2015-11-20 11:57:25.079854 
 6   
 7  """ 
 8   
 9  # revision identifiers, used by Alembic. 
10  revision = '4b57794e2b5' 
11  down_revision = '19ca0c14096e' 
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 = 0 THEN 0 22 WHEN x = 3 THEN 1 23 WHEN x = 6 THEN 2 24 WHEN x = 7 THEN 3 25 WHEN x = 4 THEN 4 26 WHEN x = 1 THEN 5 27 WHEN x = 5 THEN 6 28 WHEN x = 2 THEN 7 29 ELSE x 30 END; END; 31 $$ LANGUAGE plpgsql; 32 33 CREATE OR REPLACE FUNCTION order_to_status (x integer) 34 RETURNS integer AS $$ BEGIN 35 RETURN CASE WHEN x = 0 THEN 0 36 WHEN x = 1 THEN 3 37 WHEN x = 2 THEN 6 38 WHEN x = 3 THEN 7 39 WHEN x = 4 THEN 4 40 WHEN x = 5 THEN 1 41 WHEN x = 6 THEN 5 42 WHEN x = 7 THEN 2 43 ELSE x 44 END; END; 45 $$ LANGUAGE plpgsql; 46 """ 47 op.execute(sa.text(query_functions))
48 49
50 -def downgrade():
51 query_functions = """ 52 CREATE OR REPLACE FUNCTION status_to_order (x integer) 53 RETURNS integer AS $$ BEGIN 54 RETURN CASE WHEN x = 0 THEN 0 55 WHEN x = 3 THEN 1 56 WHEN x = 6 THEN 2 57 WHEN x = 7 THEN 3 58 WHEN x = 4 THEN 4 59 WHEN x = 1 THEN 5 60 WHEN x = 5 THEN 6 61 ELSE 1000 62 END; END; 63 $$ LANGUAGE plpgsql; 64 65 CREATE OR REPLACE FUNCTION order_to_status (x integer) 66 RETURNS integer AS $$ BEGIN 67 RETURN CASE WHEN x = 0 THEN 0 68 WHEN x = 1 THEN 3 69 WHEN x = 2 THEN 6 70 WHEN x = 3 THEN 7 71 WHEN x = 4 THEN 4 72 WHEN x = 5 THEN 1 73 WHEN x = 6 THEN 5 74 ELSE 1000 75 END; END; 76 $$ LANGUAGE plpgsql; 77 """ 78 op.execute(sa.text(query_functions))
79