【Python】python練習

 #!/usr/bin/python
# -*- coding: utf-8 -*-
import MySQLdb,os
conn=MySQLdb.connect(user="root",passwd="XXXXXXXXX",db="interface_hd_com",port=3307,host="127.0.0.1",charset="utf8")
cursor =conn.cursor()

def get_all_uid():
       v_sql ="select distinct(uid) FROM b_citywide_peer limit 10"
       cursor.execute(v_sql)
       row=cursor.fetchall()
       return row



def get_name(x):
       v_sql ="select name FROM b_users where uid= %d" % x
       cursor.execute(v_sql)
       row=cursor.fetchall()
       return  row[0][0]

def get_tel(x):
       v_sql ="select tel  FROM b_users where uid= %d" % x
       cursor.execute(v_sql)
       row=cursor.fetchall()
       return  row[0][0]


def get_name_tel(x):
     v_sql ="select name,tel  FROM b_users where uid= %d" % x
     cursor.execute(v_sql)
     row=cursor.fetchall()
     return  row

def get_my_foc(id):
        sql ="select uid ,group_concat(c_uid)  FROM b_citywide_peer group by uid having uid=%d" % id
        cursor.execute(sql)
        row=cursor.fetchall()
        row2=row[0]
        my_foc=set(row2[1].split(','))
        return my_foc

def get_foc_me(id):
        sql2="select c_uid,group_concat(uid) from  b_citywide_peer group by c_uid having c_uid= %d" % id
        cursor.execute(sql2)
        row=cursor.fetchall()
        print row
        row2=row[0]
        foc_me=set(row2[1].split(','))
        return foc_me



def save_to_txt(path,data):
                file_path = os.path.split(path)
                if not os.path.exists( file_path[0] ):
                        os.makedirs( file_path[0] )
                        print '建立目錄成功'
                if len(data)>0:
                        try:
                                with open(  path.encode('utf-8'),'ab' ) as f:
                                        data = data+"\n"
                                        f.write( data.encode('utf-8') )
                        except:
                                print '資料插入失敗'
                else:
                        print '沒有資料'


def get_erdu_link():
   for m in get_all_uid():
        print m
        uid=m[0]
        Sec_fid=get_foc_me(uid).difference(get_my_foc(uid))
        for i in Sec_fid:
              j=int(i)
              lists=get_name_tel(j)
              erdu_name=lists[0][0]
              erdu_tel=lists[0][1]
              self_name=get_name(uid)
              self_tel=get_tel(uid)
              str= "%s %s %s %s %s" % (self_name,self_tel,u"二度人脈",erdu_name,erdu_tel)
              print str
              save_to_txt("./tt.log",str)



#print get_foc_me(140)
#print get_all_uid()





get_erdu_link()
cursor.close()
conn.close()