python 从共享文件中自动拷贝指定文件到本地【续】

前几天写过一篇文章：python 从共享文件中自动拷贝指定文件到本地

今天把它完善一下，整个代码分为4块：测试代码脚本、命令行脚本、copy 功能脚本以及辅助工具脚本。

【备注：我是用的是python2.7，由于python3.x与2.x有很多地方不同，所以不要尝试在3.x上，以免出现未知错误】

首先是测试脚本：

在这里，由于是拷贝脚本的编写，必然涉及到源文件以及目标文件，那么先从源文件入手，想获得共享文件，首先得有一个共享文件所在服务器的ip地址，然后有一个已知源文件路径这样就有了srcFilename了，至于目标路径，随便设定。代码：

test.py:

import os , sys , Mycommand

def  main():
hostIp = '192.168.0.111'
sharePath = '\\path'
filename = 'file_testname'

resultStr = []
resultStr.append([])
srcFilename = '\\\\' + hostIp + sharePath + '\\' + filename
desFilename = 'd:\\ftp\\ftp_download'

cmd = [
'C:\\python27\\python.exe',
'C:\\Users\\admin\\Desktop\\copy\\copyShareFile.py',
srcFilename,
desFilename
]
res = Mycommand.command(resultStr , cmd , 3)
if res == True:
print ('download success!')
else :
print ('download failed!')

if __name__ == '__main__':
main()

接下来是MyCommand模块，这个模块是使用subprocess.Popen创建命令行线程向copy那个自定义的脚本传参，这个模块为什么会有呢？为什么不使用copy直接拷贝呢？因为当考虑的时候，应该考虑共享文件不可访问或者路径不存在这种情况，那么如何识别不能顺利获取到想要的文件呢？开始的时候想用ping 命令判断是否可以访问到，但是实际做的时候发现，服务器那边可能将ping 功能关闭，这样当可以访问的时候，也会出现ping 不通的可能。后来想到，如果可以顺利找到共享文件那么就可以获得，之后使用os.path.exists()找的时候发现，如果找得到的时候，会瞬间找到，即使不能很快，至少3秒也应该能够找到了。但是如果不能找到，这个函数会阻塞，一直在那尝试找，直到达到了函数默认的timeout（大约是10秒以上）才会停止查找，但是我们不希望那样让它一直在那“傻傻的”找，而想自己定时，以至于我们可以根据具体情况设置一下界限，当限制时间内还没有找到共享文件那么就判断是不可访问，则立刻停止搜索，也就是将搜索所在的脚本kill掉（之前一直找如何控制函数运行时间，超时就kill，但是找到如何控制时间了，但是同一个线程中，感觉没办法kill一个函数，所以但提出来使用了这个方法），用户体验也就提升了，好了，贴代码：

Mycommand.py:

import sys , time , subprocess

class TimeoutError(Exception):
pass

def command(resultStr , cmd , timeout = 20):
print ('command start...')
process = subprocess.Popen(cmd , stderr=subprocess.STDOUT, stdout=subprocess.PIPE, shell = True)
beginTime = time.time()
secondsPass = 0

while True:
if process.poll() is not None:
break
secondsPass = time.time() - beginTime
if timeout and timeout < secondsPass:
process.terminate()
print ('running timeout!')
return False
#raise TimeoutError(cmd , timeout)
time.sleep(0.1)
if len(resultStr) > 0:
resultStr[0] = process.stdout.read()
return True

最后是比较常规的拷贝代码：

# -*- coding: cp936 -*-
import sys , os , re

def copyFileDir(srcFilename , desFilename):
status = False
try:
fileList = os.listdir(srcFilename)
for eachFile in fileList:
sourceF = os.path.join(srcFilename,eachFile)
targetF = os.path.join(desFilename,eachFile)

if os.path.isdir(sourceF):
if not os.path.exists(targetF):
os.makedir(targetF)
status = copyFileDir(sourceF,targetF)
else :
status = copyFile(sourceF,targetF)
except Exception,e:
print (e)
status = False
finally:
print ('copyFileDir function is quit!')
return status

def copyFile(srcFilename , desFilename):
status = False
copyCommand = 'copy %s %s'%(srcFilename,desFilename)

try:
if(os.popen(copyCommand)):  #不用op.system(copyCommand),因为这个会弹出命令行界面
print ('copy done!')
status = True
else :
print ('copy failed!')
status = False
except Exception,e:
print (e)
status = False
finally:
print ('copyFile function is quit!')
return status

def copyFromSharePath(srcFilename,desFilename):
if not os.path.exists(srcFilename):
print ('no found '+srcFilename)
return False
if not os.path.exists(desFilename):
print ('no found '+desFilename)
os.makedirs(str(desFilename))
print ('create '+desFilename)

copyStatus = False
if os.path.isdir(srcFilename):
copyStatus = copyFileDir(srcFilename,desFilename)
else :
copyStatus = copyFile(srcFilename,desFilename)
return copyStatus

def main(argv = sys.argv):
if not len(argv) == 3:
print ('input parameters\'s count should be 3,not %s'%(len(argv)))
return
print ('脚本名字是：' + argv[0])
srcFilename = argv[1]
print ('源目录：' + argv[1])
desFilename = argv[2]
print ('目标目录：' + argv[2])

if os.path.isdir(srcFilename):
if os.path.isfile(desFilename):
print ('can not copy a folder to a file')
return
copyFromSharePath(srcFilename,desFilename)

if __name__=='__main__':
main()

里边依然存在一些没哟考虑到的地方，希望朋友们发现了说一下啊