Contents
p2p 试开发
最近,用python编了一个p2p文件共享系统,是在windows下的 源代码如下:
>>> import sys,os,os.path,xml.dom.minidom
>>> import struct,socket,threading,Queue
>>> #获取本地共享文件列表
>>> def LocalFileList():
#共享当前工作目录下的文件
cwdlist=os.listdir(os.getcwd())
cwdfilelist=[]
for afile in cwdlist:
#判断是文件还是目录
if os.path.isfile(afile):
cwdfilelist.append(afile)
else:
continue
return(cwdfilelist,os.getcwd())
>>> #生成共享文件XML文档
>>> def GenFileXML(cwdfilelist,cwdir):
XMLFileName='SharedFile.xml'
if os.path.isfile(XMLFileName):
os.remove(XMLFileName)
xmlfile=open(XMLFileName,'w')
fileid=1
filestr=''
#对于简单的XML文档可以使用输出重定向来生成
print>>xmlfile,'<FileList>'
for afile in cwdfilelist:
filestr='<File FileID="'+str(fileid)
+'"FileName="'+afile
+'"FileDir="'+cwdir+'"/>'
print>>xmlfile,filestr
fileid=fileid+1
print>>xmlfile,'</FileList>'
xmlfile.close()
>>> #解析XML文档
>>> def ParsFileXML(xmlfile):
RFileDict={}
fh=open(xmlfile,'r')
doc=xml.dom.minidom.parse(fh)
FileTagList=doc.getElementsByTagName("File")
for aFileTag in FileTagList:
FileID=aFileTag.getAttribute('FileID')
FileName=aFileTag.getAttribute('FileName')
RFileDict[int(FileID)]=FileName
fh.close()
return RFileDict
>>> #命令监听线程
>>> def cmdListenThread(queue):
#监听端口12007
cmdport=12007
#计算一条命令的长度
CMD_SIZE=struct.calcsize('cI')
cmdsock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
cmdsock.bind(('',cmdport))
while(True):
cmdsock.listen(1)
conn,addr=cmdsock.accept()
data=conn.recv(CMD_SIZE)
#解析命令包
cmdhead,cmdtail=struct.unpack('cI',data)
conn.close()
#'r'是请求连接命令
if cmdhead=='r':
#发共享文件列表
senddata(addr,'SharedFile.xml')
#'d'是下载文件命令
elif cmdhead=='d':
RFileDict={}
RFileDict=ParsFileXML('SharedFile.xml')
keys=RFileDict.keys()
for key in keys:
if int(key)==cmdtail:
senddata(addr,RFileDict[key])
else:
continue
>>> #文件监听线程
>>> def fileListenThread(queue):
#监听端口是12008
fileport=12008
#用于判断发送的是文件信息还是文件内容
IS_FILEINFO=1
#计算文件信息包的大小
FILEINFO_SIZE=struct.calcsize('128s32sI8s')
#接收文件时，每次读取1KB
blksz=1024
filename,filesize='',0
#已经接收的文件大小
recved=0
filesock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
filesock.bind(('',fileport))
while(True):
filesock.listen(1)
conn,addr=filesock.accept()
while(True):
#接收文件信息
if(IS_FILEINFO==1):
data=conn.recv(FILEINFO_SIZE)
#解析文件信息包
filename,temp1,filesize,temp2=struct.unpack
('128s32sI8s',data)
filename=filename.strip('\00')
if filename=='SharedFile.xml':
filename='RSharedFile.xml'
IS_FILEINFO=0
else:
file=open("./"+filename,'web')
while(True):
remainf=filesize-recved
if remainf>=blksz:
data=conn.recv(blksz)
file.write(data)
recved=recved+len(data)
elif remainf<blksz:
data=conn.recv(remainf)
file.write(data)
recved=recved+len(data)
if recved==filesize:
break
IS_FILEINFO=2
recved=0
file.close()
filename=""
if IS_FILEINFO==2:
IS_FILEINFO=1
break
IS_FILEINFO=1
filename,filesize='',0
recved=0
conn.close()
>>> #发送命令
>>> def sendcmd(ip,cmdtype,cmdarg):
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((ip,12007))
if cmdtype=='connect':
cmdpkt=struct.pack('cI','r',0)
s.send(cmdpkt)
elif cmdtype=='down':
cmdpkt=struct.pack('cI','d',int(cmdarg))
s.send(cmdpkt)
s.close()
>>> #发送文件
>>> def senddata(ip,filename):
FILEINFO_SIZE=struct.calcsize('128s32sI18s')
blksz=1024
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((ip,12008))
fp='./'+filename
fs=os.stat(fp)
fhead=struct.pack('128s11I',f,0,0,0,0,0,0,0,0,fs.st_size,0,0)
s.send(fhead)
fd=open(fp,'rb')
while 1:
bytes=fd.read(blksz)
if not bytes:break
sent=s.send(bytes)
assert sent==len(bytes)
fd.close()
s.close()
>>> queue=Queue.Queue(0)
>>> tcmdlisten=threading.Thread(target=cmdListenThread,args=(queue,))
>>> tcmdlisten.setDaemon(True)
>>> tcmdlisten.start()
>>> tfilelisten=threading.Thread(target=fileListenThread,args=(queue,))
>>> tfilelisten.setDaemon(True)
>>> tfilelisten.start()
>>> cwdfilelist,cwdir=LocalFileList()
>>> GenFileXML(cwdfilelist,cwdir)
>>> while(True):
commandstr=raw_input('p2p File Transport?')
if commandstr=='help':
print 'This a simple p2p fileshare program in
python.'
elif commandstr=='exit':
print 'Exit the program'
break
elif commandstr=='Islocal':
print 'Local shared file:'
for afile in cwdfilelist:
print afile
elif(commandstr.find('connect')!=-1):
RemoteIP=commandstr.split('')[1]
sendcmd(RemoteIP,'connect',0)
elif(commandstr.find('down')!=-1):
dfileid=commandstr.split('')[1]
sendcmd(RemoteIP,'connect',int(dfileid))
elif commandstr=='Isremote':
print 'Remote shared file:'
RFileDict={}
RFileDict=ParsFileXML()
keys=RFileDict.keys()
for key in keys:
print key,':',RFileDict[key]
else:
print 'unknown command!'
print 'please using help command'
{{{
运行后,出错在第一行,
{{{
C:\Python26\packbag>minitransport.py
File "C:\Python26\packbag\minitransport.py", line 2
>>> import sys,os,os.path,xml.dom.minidom
^
SyntaxError: invalid syntax
这是什么原因,一般是不会出这样的错吧 麻烦各位了，我在网上找了一下原因，说有可能是路径的问题，即把os.path改为os.path.dirname(argv[0]),可是，运行后， 仍然报错，和上面那个一样，这是什么原因？
反馈
创建 by -- ZoomQuiet [2009-04-23 13:47:36]