decode 和case when 转换函数

decode 和case when 转换函数

背景

因工作需要经常在mysql数据库和oracle数据库等切换,造成写完的代码经常要迁移,但是mysql里面没有decode,所以用python写了一个通过decode转换成 case when写法的函数

代码

def decode_case(str1):
    #对字符串进行分列
    t0 = str1.split(',')
    #取出待映射的列表
    t1 = t0[-(len(t0)-1):-1]
    #取出映射列表中的key
    t_key = [t1[i] for i in range(len(t1)) if i%2==0]
    #取出待映射列表中的value
    t_value = [t1[i] for i in range(len(t1)) if i%2==1]
    #取出待映射的列名
    t2 = str1.split(',')[0].split('(')[1]
    #取出无法匹配时的默认值
    t3 = t0[-1]
    result = 'case '
    for i in range(len(t_value)):
        result =result+ 'when '+str(t2)+'='+t_key[i]+ ' then '+t_value[i]+' '
    result = result +'end'
    return result

样例

输入

decode_case("decode(ec.customer_or_client,1,'客户',2,'代理人',ec.customer_or_client)")

输出