本文参考：http://blog.163.com/kongdelu2009@yeah/blog/static/1119952072009102562126194/
发送端程序：
# -*- coding: utf8 -*-
import struct
import logging.handlers
import sys;
from socket import *
import os;
import string;
LOG_FILE = 'send.log'
//通过log模块将日志输出到文件当中
handler = logging.handlers.RotatingFileHandler(LOG_FILE,mode = 'w', maxBytes = 1024*1024, backupCount = 5) # 实例化handler
fmt = '%(asctime)s - %(filename)s:%(lineno)s - %(message)s'
formatter = logging.Formatter(fmt) # 实例化formatter
handler.setFormatter(formatter) # 为handler添加formatter
logger = logging.getLogger('Send.log') # 获取名为tst的logger
logger.addHandler(handler) # 为logger添加handler
logger.setLevel(logging.DEBUG)
//增加handle，将日志输出到控制台
console = logging.StreamHandler()
console.setLevel(logging.DEBUG)
# 设置日志打印格式
console.setFormatter(formatter)
logger.addHandler(console)
//命令行参数处理
iArgLen = len(sys.argv);
if iArgLen < 4:
logger.debug('command arg is not enough')
sys.exit(0);
#网络地址
addrString = sys.argv[1];
logger.debug('addrString = ' + addrString);
#必须使用fullPath
filename = sys.argv[2];
logger.debug('filename = ' + filename);
path,fName=os.path.split(filename);
#文件传输过去的存储地址
# ServerSaveAddr = sys.argv[3];
# ServerSaveAddr = "%s\\%s"%(ServerSaveAddr,fName);
# logger.debug('ServerSaveAddr = %s'%(ServerSaveAddr));
#版本号
version = sys.argv[3];
logger.debug('version = %s'%(version));
#map类型 0-DB 1- OB
iType = int(sys.argv[4]);
logger.debug('iType = %d'%(iType));
ADDR = (addrString,5454);
BUFSIZE = 1024
FILEINFO_SIZE=struct.calcsize('128s11I')
logger.debug('fineinfo_size = %d'%(FILEINFO_SIZE));
sendSock = socket(AF_INET,SOCK_STREAM)
#try:
sendSock.connect(ADDR)
version = version.encode('utf-8');
iLen = len(version);
logger.debug('versin string Len = %d'%(iLen));
//序列号相关信息至内存
fhead=struct.pack('128s11I',version,iType,iLen,0,0,0,0,0,0,os.stat(filename).st_size,0,0)
logger.debug("fheadsize %d" %(len(fhead)));
logger.debug("filesize %d" %(os.stat(filename).st_size));
ret = sendSock.send(fhead)
logger.debug('head send success ,size = %d'%(ret));
fp = open(filename,'rb')
while 1:
filedata = fp.read(BUFSIZE)
if not filedata: break
ret = sendSock.send(filedata);
#logger.debug('file send ,size = %d'%(ret));
logger.debug('file send success');
fp.close();
sendSock.close();
##except:
# print('An exception happened ,clear and exit');
# if 'fp' in locals().keys() :
# fp.close();
#sendSock.close();
logger.debug('Aplication End');
//接受端程序
# -*- coding: utf8 -*-
from socket import *
import os
import struct
import sys
import time;
import logging
import logging.handlers
LOG_FILE = 'Recv.log'
handler = logging.handlers.RotatingFileHandler(LOG_FILE, maxBytes = 1024*1024, backupCount = 5) # ʵ��handler
fmt = '%(asctime)s - %(filename)s:%(lineno)s - %(message)s'
formatter = logging.Formatter(fmt) # ʵ��formatter
handler.setFormatter(formatter) # Ϊhandler���formatter
logger = logging.getLogger('Recv.log') # ��ȡ��Ϊtst��logger
logger.addHandler(handler) # Ϊlogger���handler
logger.setLevel(logging.DEBUG)
console = logging.StreamHandler()
console.setLevel(logging.DEBUG)
console.setFormatter(formatter)
logger.addHandler(console)
iArgLen = len(sys.argv);
if iArgLen < 2:
logger.debug('command arg is not enough')
sys.exit(0);
addrString = sys.argv[1];
logger.debug('addrString = ' + addrString);
ADDR = ('',5454);
logger.debug(ADDR);
BUFSIZE = 1024
FILEINFO_SIZE=struct.calcsize('128s11I')
logger.debug('fineinfo_size = %d'%(FILEINFO_SIZE));
recvSock = socket(AF_INET,SOCK_STREAM)
recvSock.bind(ADDR)
recvSock.listen(True)
logger.debug("wait for connect\r\n");
bTimeOut = False;
fileSize = 0;
//接受相关数据，需要进行超时处理
while 1:
conn,addr = recvSock.accept()
logger.debug("-------------------connect success------------------- ");
fhead = conn.recv(FILEINFO_SIZE)
fEnterClock = time.clock();
bTimeOut = False;
while 1:
iLen = len(fhead);
if iLen != 0:
if iLen != FILEINFO_SIZE:
logging.debug('The Recive headLen(%d) is not fit(%d)'%(iLen,FILEINFO_SIZE))
sys.exit(0);
else:
if time.clock() - fEnterClock > 3:
bTimeOut = True;
break;
continue;
logger.debug('fheadsize = %d'%(len(fhead)));
version,iType,iLen,temp3,temp4,temp5,temp6,temp7,temp8,fileSize,temp10,temp11=struct.unpack('128s11I',fhead)
logger.debug('fliesize %d'%(fileSize));
break;
if bTimeOut == True:
logger.debug('This Connect TimeOut');
conn.close();
continue;
logger.debug("file head receive success ");
version = version.decode('utf-8');
version = version[0:iLen];
logger.debug("version = %s "%(version));
filename = '%s.1zip'%(version);
logger.debug("filename = %s"%(filename));
filename = filename.strip('\00') #...
fp = open(filename,'wb')
restsize = fileSize
logger.debug("begin receice file, filesize = %d"%(fileSize));
while 1:
if restsize > BUFSIZE:
filedata = conn.recv(BUFSIZE)
else:
filedata = conn.recv(restsize)
if not filedata: break
fp.write(filedata)
restsize = restsize-len(filedata)
if restsize == 0:
break
logger.debug('file receive success');
fp.close()
conn.close()
# save the version number
fVersion = open('version.txt','w');
fVersion.write(version);
fVersion.close();
logger.debug('save version success');
# save the position number
fPos = open('CopyPostion.txt','w');
strType = ('%d'%(iType));
fPos.write(strType);
fPos.close();
logger.debug('save copyPostion success');
# open the copy bat command
os.system('test.bat');
recvSock.close()
logger.debug('socket close');
总结：
1、try/except/finally;
2、 if 'fp' in locals().keys() : 判断本地变量是否存在
3、dic = dic(), list = [], str()