第一个python程序

　　版本上线，有个洗数据的流程。

　　根据一个文件，把sql生成出来。

　　习惯了利用php，非常简单的几句代码就写出来。因为没写过python，就想用py写下，虽然没啥技术含量，也就当练习下。

原文件格式如下：　　

30001,BACDH45101
30001,BACHXZ5101
30002,BACJD45101
30003,BACTG45101
30003,BAGLP45101
30004,BAYXJ45101
30004,BFAKJ41101
30005,BFBDZ41102
30005,BFBDZ41119
...
...
...

 

 

php 代码：

$data = file_get_contents("flushdata.log");

$arr = explode(PHP_EOL, $data);

foreach ($arr as $item){
    $items = explode(',', $item);
    if($tagc = $items[0]){
        $temp[$tagc][] = $items[1];
    }
}
$str = '';
foreach ($temp as $key => $val){
    $ins = "'".implode("','", $val)."'";
    $str .= "update list set tag=". $key. " where mid in (". $ins. ");".PHP_EOL;
}

file_put_contents("flushdata_php.sql", $str);

 

 

py代码：

#! /usr/bin/python

file_object = open('flushdata.log')
try:
     contents = file_object.readlines()
     d = {}
     for line in contents:
        sp = line.split(',')
        tagc = sp[0]
        mid = sp[1].strip('\n')
        if tagc in d:
            d[tagc].append(mid)
        else :
            d[tagc] = [mid]

     arr = sorted(d.keys())

     str = ""
     for tagc in arr:
        str += "update list set tag=" + tagc + " where mid in ("
        str += "'" + "','".join(d[tagc]) + "'"
        str += ");\n"

     file_object = open('flushdata_py.sql', 'w')
     file_object.write(str)
     file_object.close()

finally:
     file_object.close()

 

执行后，diff下俩文件一致

diff -b -B  flushdata_php.sql flushdata_py.sql

 

python的语法不太熟悉，第一次写比较生硬...

各种冒号 TAB，dict 和list的用法 简单应用了下