#coding = utf-8
import conn as con
from mysql import mysql as my

def link_ppd( pre = 'ppd' ):
#link ppd db
    ppd_config = con.data[pre];
    ppd_db = my(ppd_config['host'],ppd_config['user'],ppd_config['pw'],ppd_config['dbname'],ppd_config['port'])
    return ppd_db

ppd_db = link_ppd( 'uniprot' )

def get_uniprot_data( uniprot_id , select = '*'):
    results = ppd_db.fetchall_data("select "+select+"  from  p_uniprot where `accession` like '"+uniprot_id+"' ")
    return results

def rep(chars):
    return chars.replace(',',"，")

######根据uniprot id 获取相关的信息存储为 csv
try:
    f = open("C:/Users/li_hao/Desktop/uniprot.txt")
    con = f.read()
    import csv
    csvfile = open('C:/Users/li_hao/Desktop/un.csv', 'wb+')
    spamwriter = csv.writer(csvfile, delimiter=',',quotechar='|', quoting=csv.QUOTE_MINIMAL)
    spamwriter.writerow(['uniprot', 'antigen_source', 'function', 'reference', 'other_name'])
    if(con):        
        list_arr = con.splitlines()
        for i in list_arr :
            j = i.strip(' ')
            item = j.strip('@')
            if(item == ''):
                continue
            res = get_uniprot_data(item,"accession as uniprot,antigen_source,function,reference,other_name")
            
            if( len(res) == 0 ):
                spamwriter.writerow([item])
         
            print 'writing data...'
            for item in res:
                 #特殊字符“ ,”,替换为“，”
                data = [item[0],item[1],rep(item[2]),rep(item[3]),rep(item[4])]
                spamwriter.writerow(data)
            print 'write data ok'
    csvfile.close()
    
except IOError,msg:
    print msg

######根据uniprot id 获取相关的信息