# coding:utf-8
from ctypes import *
import os
import sys
import ftplib
class myFtp:
ftp = ftplib.FTP()
bIsDir = False
path = ""
def __init__(self, host, port='21'):
# self.ftp.set_debuglevel(2) #打开调试级别2,显示详细信息
# self.ftp.set_pasv(0) #0主动模式 1 #被动模式
self.ftp.connect(host, port)
def Login(self, user, passwd):
self.ftp.login(user, passwd)
print self.ftp.welcome
def DownLoadFile(self, LocalFile, RemoteFile):
file_handler = open(LocalFile, 'wb')
self.ftp.retrbinary("RETR %s" %
RemoteFile.encode('gbk'), file_handler.write)
file_handler.close()
return True
def UpLoadFile(self, LocalFile, RemoteFile):
if os.path.isfile(LocalFile) == False:
return False
file_handler = open(LocalFile, "rb")
self.ftp.storbinary('STOR %s' %
RemoteFile.encode('gbk'), file_handler, 4096)
file_handler.close()
return True
def UpLoadFileTree(self, LocalDir, RemoteDir):
if os.path.isdir(LocalDir) == False:
return False
print "LocalDir:", LocalDir
LocalNames = os.listdir(LocalDir)
print "list:", LocalNames
print RemoteDir
self.ftp.cwd(RemoteDir)
for Local in LocalNames:
Local = Local.decode('gbk')
src = os.path.join(LocalDir, Local)
if os.path.isdir(src): self.UpLoadFileTree(src, Local)
else:
self.UpLoadFile(src, Local)
self.ftp.cwd("..")
return
def DownLoadFileTree(self, LocalDir, RemoteDir):
print "remoteDir:", RemoteDir
if os.path.isdir(LocalDir) == False:
os.makedirs(LocalDir)
self.ftp.cwd(RemoteDir)
RemoteNames = self.ftp.nlst()
i = 0
for file in RemoteNames:
print str(i) + ': %s' % file.decode('gbk').encode('utf-8')
# print self.ftp.nlst("/del1")
for file in RemoteNames:
Local = os.path.join(LocalDir, file.decode('gbk'))
if self.isDir(file):
self.DownLoadFileTree(Local, file.decode('gbk'))
else:
self.DownLoadFile(Local, file.decode('gbk'))
self.ftp.cwd("..")
return
def show(self, list):
result = list.lower().split(" ")
if self.path in result and "<dir>" in result:
self.bIsDir = True
def isDir(self, path):
self.bIsDir = False
self.path = path
# this ues callback function ,that will change bIsDir value
self.ftp.retrlines('LIST', self.show)
return self.bIsDir
def close(self):
self.ftp.quit()
class bcp:
def __init__(self, server, user, pwd, t=r'|', r=r'|\n'):
self.server = server
self.user = user
self.pwd = pwd
self.t = t
self.r = r
def unload(self, sql, file):
str = r'bcp "%s" queryout "%s" -c -t"%s" -r"%s" -S "%s" -U "%s" -P "%s"' % (
sql, file, self.t, self.r, self.server, self.user, self.pwd)
print str
os.system(str)
def load(self, sql, file):
str = r'bcp "%s" in "%s" -c -t"%s" -r"%s" -S "%s" -U "%s" -P "%s"' % (
sql, file, self.t, self.r, self.server, self.user, self.pwd)
print str
os.system(str)
if __name__ == "__main__":
'''
bcp = bcp(server='', user='',
pwd='', t=r'|', r=r'|\n')
bcp.unload('select top 10 * from test', 'E:\pytest.txt')
'''
# bcp.load('dataqry.dbo.ts_test', 'E:\pytest.txt')
ftp=myFtp('10.249.*.*')
ftp.Login('**','**')
ftp.UpLoadFile(u'D:\#20160504.txt',u'/aa.txt')
# ftp.DownLoadFile(u'而台湾.txt',u'/zhms/dandu/dd.txt')
# ftp.UpLoadFile(u'E:/zhms/aa.txt',u'/zhms/danduaa.txt')
# ftp.DownLoadFileTree(u'as','/zhms/dandu')
# ftp.UpLoadFileTree('as','/a')
'''
print 'finish'
dic = {}
for l in open('bcp_init.txt'):
print l.split('=')
print l.split('=')[0]+':'+l.split('=')[1]
dic[l.split('=')[0].strip()]=l.split('=')[1].strip().replace("'",'')
print dic
print dic['Server']
bcp = bcp(server=dic['Server'], user=dic['User'],
pwd=dic['Pwd'], t=dic['T'], r=dic['R'])
bcp.unload(dic['Sql'], 'E:\pytest.txt')
ftp=myFtp(dic['Ftp'])
ftp.Login(dic['FtpUser'],dic['FtpPwd'])
ftp.UpLoadFile('E:\pytest.txt',dic['Path'].decode('gbk'))
'''