py文件传输

本文参考: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()





posted on 2013-08-28 20:00  KHacker  阅读(164)  评论(0编辑  收藏  举报