[原]把boblog博客的用户导入proftpd用户认证数据库

import MySQLdb
import sys

host = ‘localhost’

# 定义进入boblog数据库的用户名和密码
sourcedb = ‘boblog’
s_user = ‘linuxing’
s_pwd = ‘redflag’

# 定义进入proftpd认证数据库的用户名和密码
targetdb = ‘proftp’
t_user = ‘linuxing’
t_pwd = ‘redflag’

# 定义这些用户的默认uid和gid
t_uid = 1000
t_gid = 1000

# 当进入数据库的信息错误时，捕捉异常并处理
try:
   #使用utf8编码连接
   s_conn = MySQLdb.Connection(host,s_user,s_pwd,sourcedb,use_unicode=1,charset=’utf8′)
   t_conn = MySQLdb.Connection(host,t_user,t_pwd,targetdb,use_unicode=1,charset=’utf8′)
except:
   print ”’Open source or target database fail.\nPlease confirm content of scripts.”’
   sys.exit(1)

#修正sys的默认编码，否则若把脚本输出导入管道会报UnicodeEncodeError错误的
reload(sys)
sys.setdefaultencoding(‘gb2312’)

s_cur = s_conn.cursor()
s_cur.execute(‘select username,userpsw from boblog_user’);
s_row = s_cur.fetchall()
# 使用for循环得到每一行的tuple
for rowone in s_row:
   ftpuser,ftppwd = rowone
   t_cur = t_conn.cursor()
   sqlstring = “select userid from FTPUSERS where userid=’%s'” % ftpuser
   t_cur.execute(sqlstring)
   #这里其实可以用简单的判断execute方法返回的值是否为0L即可
   t_row = t_cur.fetchone()
   if t_row == None:
       sqlstring = “insert into FTPUSERS(userid,passwd,uid,gid,lastime) values (‘%s’,’%s’,%i,%i,%s)” %(ftpuser,ftppwd,t_uid,t_gid,’now()’)
       t_cur.execute(sqlstring)
       print ‘insert ftp user: %s.’ %ftpuser
       t_conn.commit()
   else:
       sqlstring = “update FTPUSERS set passwd =’%s’ where userid =’%s'” %(ftppwd,ftpuser)
       t_cur.execute(sqlstring)
       print ‘update ftp user %s.’ %ftpuser
       t_conn.commit()
print ‘Everything is OK.’