Context Navigation

← Previous Revision
Latest Revision
Next Revision →
Normal
Revision Log

fileServer.py @ 4

Revision 4, 31.0 KB checked in by ajaworski, 13 years ago (diff)
Added modified SAGE sources

Rev	Line
[4]	1	#!/usr/bin/env python
	2	############################################################################
	3	#
	4	# -= FileViewer =-
	5	#
	6	# FileServer - receives, provides and organizes multimedia files for use with SAGE
	7	#
	8	# Copyright (C) 2005 Electronic Visualization Laboratory,
	9	# University of Illinois at Chicago
	10	#
	11	# All rights reserved.
	12	#
	13	# Redistribution and use in source and binary forms, with or without
	14	# modification, are permitted provided that the following conditions are met:
	15	#
	16	# * Redistributions of source code must retain the above copyright
	17	# notice, this list of conditions and the following disclaimer.
	18	# * Redistributions in binary form must reproduce the above
	19	# copyright notice, this list of conditions and the following disclaimer
	20	# in the documentation and/or other materials provided with the distribution.
	21	# * Neither the name of the University of Illinois at Chicago nor
	22	# the names of its contributors may be used to endorse or promote
	23	# products derived from this software without specific prior written permission.
	24	#
	25	# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
	26	# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
	27	# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
	28	# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
	29	# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
	30	# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
	31	# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
	32	# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
	33	# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
	34	# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
	35	# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	36	#
	37	# Direct questions, comments etc about FileViewer to www.evl.uic.edu/cavern/forum
	38	#
	39	# Author: Ratko Jagodic
	40	#
	41	############################################################################
	42
	43	import os, base64, os.path, socket, shutil, sys, SocketServer, wx, stat, urllib
	44	from SimpleXMLRPCServer import *
	45	from threading import Thread
	46	from time import asctime, time
	47	import makeThumbs
	48
	49	# miscellaneous stuff (3rd party supporting tools)
	50	sys.path.append("misc") #so that we can import packages from "misc" folder
	51	from imsize import imagesize # reads the image header and gets the size from it
	52	import mmpython
	53	import countPDFpages
	54
	55	# a shortcut
	56	opj = os.path.join
	57
	58	# for loading the sagePath helper file
	59	sys.path.append( opj(os.environ["SAGE_DIRECTORY"], "bin" ) )
	60	from sagePath import getUserPath, getPath
	61
	62
	63	## some globals defining the environment
	64	SCRIPT_PATH = sys.path[0]
	65	CONFIG_FILE = getPath("fileServer", "fileServer.conf")
	66	CACHE_DIR = getUserPath("fileServer", "file_server_cache")
	67	FILES_DIR = opj(SCRIPT_PATH, "file_library")
	68	REDIRECT = False
	69	THUMB_DIR = opj(FILES_DIR, "thumbnails")
	70	RUN_SERVER = True
	71
	72	## holds information about the types that we support (read from the config file)
	73	dirHash = {}
	74	viewers = {}
	75	types = {}
	76
	77
	78	def ParseConfigFile():
	79	global FILES_DIR
	80	global THUMB_DIR
	81	global dirHash
	82	global types
	83	global viewers
	84
	85	# reinitialize everything
	86	FILES_DIR = ""
	87	THUMB_DIR = ""
	88	dirHash = {}
	89	types = {}
	90	viewers = {}
	91
	92	# read the config file
	93	f = open(CONFIG_FILE, "r")
	94	for line in f:
	95	line = line.strip()
	96	if line.startswith("FILES_DIR"):
	97	FILES_DIR = line.split("=")[1].strip()
	98	if not os.path.isabs(FILES_DIR):
	99	FILES_DIR = getUserPath("fileServer", FILES_DIR)
	100	FILES_DIR = os.path.realpath(FILES_DIR) #expand any symbolic links in the library directory
	101	THUMB_DIR = opj(FILES_DIR, "thumbnails")
	102	elif line.startswith("type:"):
	103	line = line.split(":",1)[1]
	104	(type, extensions) = line.split("=")
	105	types[type.strip()] = extensions.strip().split(" ")
	106	dirHash[type.strip()] = opj(FILES_DIR, type.strip())
	107	elif line.startswith("app:"):
	108	line = line.split(":", 1)[1].strip()
	109	(type, app) = line.split("=")
	110	tpl = app.strip().split(" ", 1)
	111	if len(tpl) == 1: params = ""
	112	else: params = tpl[1].strip()
	113	app = tpl[0].strip()
	114	viewers[type.strip()] = (app, params)
	115	f.close()
	116
	117
	118	# create the folders first if they dont exist
	119	if not os.path.isdir(FILES_DIR):
	120	os.makedirs(FILES_DIR)
	121	if not os.path.isdir(CACHE_DIR):
	122	os.makedirs(CACHE_DIR)
	123	for type, typeDir in dirHash.items():
	124	if not os.path.isdir( ConvertPath(typeDir) ):
	125	os.makedirs(typeDir)
	126	if not os.path.isdir( ConvertPath( opj(typeDir, "Trash")) ):
	127	os.makedirs( opj(typeDir, "Trash") )
	128	if not os.path.isdir(THUMB_DIR):
	129	os.makedirs(THUMB_DIR)
	130
	131
	132
	133	# convert the path between the systems (Windows vs. Linux/Mac)
	134	def ConvertPath(path):
	135	return apply(opj, tuple(path.split('\\')))
	136
	137
	138
	139
	140
	141	import traceback as tb
	142	from SocketServer import ThreadingTCPServer, StreamRequestHandler
	143
	144
	145
	146	class Listener(ThreadingTCPServer):
	147	""" the main server listening for connections from HWCapture """
	148
	149	allow_reuse_address = True
	150	request_queue_size = 2
	151
	152	def __init__(self, port, onMsgCallback, fileServer):
	153	self.onMsgCallback = onMsgCallback
	154	self.fileServer = fileServer
	155	ThreadingTCPServer.__init__(self, ('', int(port)), SingleFileUpload)
	156	self.socket.settimeout(0.5)
	157
	158
	159	def serve_forever(self):
	160	while RUN_SERVER:
	161	try:
	162	self.handle_request()
	163	except socket.timeout:
	164	pass
	165	except KeyboardInterrupt:
	166	break
	167	except:
	168	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	169	break
	170
	171
	172	class SingleFileUpload(StreamRequestHandler):
	173	""" one of these per file upload """
	174
	175
	176	def handle(self):
	177	""" this is where we parse the incoming messages """
	178	self.request.settimeout(0.5) # so that we can quit properly
	179
	180	# read the header
	181	header = self.rfile.readline().strip()
	182	if len(header) == 0:
	183	return
	184	(fileName, previewWidth, previewHeight, fileSize) = header.strip().split()
	185	previewSize = (int(previewWidth), int(previewHeight))
	186	fileSize = int(fileSize)
	187
	188	# make sure the filename is ok
	189	fileName = os.path.basename(fileName) # strip any directory and just keep the filename (for security reasons)
	190	fileType = self.server.fileServer.GetFileType(fileName)
	191	if not fileType:
	192	return #dont allow upload of anything that doesn't have an extension
	193
	194	# make the file
	195	fullPath = opj(dirHash[fileType], fileName)
	196	try:
	197	f=open(fullPath, "wb")
	198	except:
	199	WriteLog("Cannot create file (have write permissions?):" + filePath)
	200	return
	201
	202	bytesRead = 0
	203	while RUN_SERVER and bytesRead < fileSize:
	204	try:
	205	line = self.rfile.readline()
	206	if not line: # readline doesn't fail when socket is closed... it returns an empty string
	207	f.close()
	208	os.remove(fullPath)
	209	break
	210
	211	bytesRead+=len(line)
	212	if bytesRead > fileSize:
	213	line = line.rstrip() # remove the newline char at the end of transmission
	214	f.write(line)
	215
	216	except socket.timeout:
	217	continue
	218	except:
	219	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	220	f.close()
	221	os.remove(fullPath)
	222	break
	223	else:
	224	f.flush()
	225	f.close()
	226
	227	# tell the client that all went well
	228	self.wfile.write("1")
	229
	230	# make the preview if necessary and everything went well
	231	self.server.onMsgCallback(fileName, previewSize)
	232
	233
	234
	235
	236
	237
	238
	239
	240	class FileLibrary:
	241	def __init__(self):
	242	ParseConfigFile()
	243
	244	# start the server that will accept the file data
	245	self.fServer = Listener(8802, self.__MakePreview, self)
	246	fServerThread = Thread(target=self.fServer.serve_forever)
	247	fServerThread.start()
	248
	249
	250	### used by clients to test the connection with the server
	251	def TestConnection(self):
	252	return (FILES_DIR, 0)
	253
	254
	255	### gets the file info:
	256	### full path, file type, application info, size (for images...), file size
	257	def GetFileInfo(self, filename, fileSize=-1, path=None):
	258	try:
	259	ParseConfigFile()
	260	fileType = self.GetFileType(filename)
	261	size = (-1, -1)
	262	fileExists = False
	263
	264	if fileType: #is file even supported
	265	appName = viewers[fileType][0]
	266	params = viewers[fileType][1]
	267	if path == None: # the user dropped a new file on there
	268	fullPath = opj(self.__GetFilePath(fileType), filename) # return a default path if not passed in
	269	res = self.__FileAlreadyThere(filename, fileType, fileSize)
	270	if res: #if the file was found it will return its directory
	271	if fileType == "image": #if the file is an image, return its size
	272	try:
	273	imsize = imagesize(fullPath)
	274	size = (imsize[1], imsize[0])
	275	except:
	276	size = (-1,-1)
	277
	278	fullPath = opj(res, filename) #the path of the found file
	279	fileExists = True
	280
	281	elif not self.__LegalPath(path):
	282	return False
	283	else: # the user is trying to show an existing file
	284	path = ConvertPath(path)
	285	fullPath = opj(path, filename)
	286	## get the image size... FIX later (remove)
	287	if os.path.isfile(fullPath):
	288	fileExists = True
	289	if fileType == "image": #if the file is an image, return its size
	290	try:
	291	imsize = imagesize(fullPath)
	292	size = (imsize[1], imsize[0])
	293	except:
	294	size = (-1,-1)
	295	return (fileType, size, fullPath, appName, params, fileExists)
	296	else:
	297	return False #file type not supported
	298	except:
	299	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	300	return False
	301
	302
	303	### gets the metadata information about the file
	304	def GetMetadata(self, fullPath):
	305	fullPath = ConvertPath(fullPath)
	306	if not self.__LegalPath(fullPath):
	307	return False
	308
	309	try:
	310	fileType = self.GetFileType(fullPath)
	311	if not fileType:
	312	return False
	313	elif fileType == "pdf": # for pdfs
	314	return "Pages: "+str(countPDFpages.getPDFPageCount(fullPath))
	315	else: # for all other types
	316	metadata = mmpython.parse(fullPath)
	317	if metadata: # metadata extracted successfully
	318	return unicode(metadata).encode('latin-1', 'replace')
	319	else:
	320	return False
	321	except:
	322	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1]) )
	323	return False
	324
	325
	326	### construct a hash of all the files keyed by the directory in which the files
	327	### are and return it
	328	def GetFiles(self):
	329	ParseConfigFile()
	330	numDirs = 0
	331	numFiles = 0
	332
	333	fileHash = {}
	334	for fileType in dirHash.iterkeys():
	335	fileHash[fileType] = self.__TraverseDir( self.__GetFilePath(fileType), fileType )
	336	return fileHash
	337
	338
	339	### deletes a folder and all files and dirs in it
	340	# FIX to delete the thumbnails for all the images in this folder
	341	def DeleteFolder(self, fullPath):
	342	try:
	343	if self.__LegalPath(fullPath): # prevent users from deleting anything
	344	shutil.rmtree(ConvertPath(fullPath))
	345	return True
	346	else:
	347	return False
	348	except:
	349	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	350	return False
	351
	352
	353	### makes a new folder
	354	def NewFolder(self, fullPath):
	355	try:
	356	if self.__LegalPath(fullPath): # prevent users from deleting anything
	357	os.mkdir(ConvertPath(fullPath))
	358	self.__SetWritePermissions(ConvertPath(fullPath))
	359	return True
	360	else:
	361	return False
	362	except:
	363	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	364	return False
	365
	366
	367	### move the file to Trash
	368	def DeleteFile(self, fullPath):
	369	try:
	370	fullPath = ConvertPath(fullPath)
	371	if not self.__LegalPath(fullPath): # restrict the users to the FILES_DIR folder
	372	return False
	373	fileType = self.GetFileType(fullPath)
	374	if not fileType: # no extension so don't allow deletion
	375	return False
	376
	377	trashPath = opj( opj(self.__GetFilePath(fileType), "Trash"), os.path.split(fullPath)[1] )
	378	self.MoveFile( fullPath, trashPath )
	379	return True
	380
	381	except:
	382	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	383	return False
	384
	385
	386	### deletes the file from the library (also deletes it's thumbnail preview if it exists)
	387	def DeleteFilePermanently(self, fullPath):
	388	try:
	389	fullPath = ConvertPath(fullPath)
	390	if not self.__LegalPath(fullPath): # restrict the users to the FILES_DIR folder
	391	return False
	392	fileType = self.GetFileType(fullPath)
	393	if not fileType: # no extension so don't allow deletion
	394	return False
	395
	396	previewPath = self.__GetPreviewName(fullPath)
	397	if os.path.isfile(fullPath): # prevent users from deleting anything
	398	os.remove( fullPath ) #remove the file itself
	399	dxtFilePath = fullPath.rstrip(os.path.splitext(fullPath)[1])+".dxt"
	400	if os.path.isfile(dxtFilePath):
	401	os.remove( dxtFilePath ) # remove the dxt file if it exists
	402	if os.path.isfile( previewPath ):
	403	os.remove( previewPath ) #remove its preview thumbnail
	404	return True
	405	except:
	406	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	407	return False
	408
	409
	410	### moves the file from one directory to another
	411	def MoveFile(self, oldFullPath, newFullPath):
	412	oldFullPath = ConvertPath(oldFullPath)
	413	newFullPath = ConvertPath(newFullPath)
	414	if not self.__LegalPath(oldFullPath, newFullPath): # restrict the movement to the FILES_DIR
	415	return False
	416	if os.path.isfile(oldFullPath):
	417	try:
	418	shutil.move(oldFullPath, newFullPath)
	419	oldDxtFilePath = oldFullPath.rstrip(os.path.splitext(oldFullPath)[1])+".dxt"
	420	newDxtFilePath = newFullPath.rstrip(os.path.splitext(newFullPath)[1])+".dxt"
	421	if os.path.isfile(oldDxtFilePath):
	422	shutil.move(oldDxtFilePath, newDxtFilePath)
	423	return True
	424	except:
	425	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	426	return False
	427	else:
	428	return False
	429
	430
	431	### accepts a file encoded as a base64 string and puts it in
	432	### the default directory for that file type
	433	def UploadFile(self, name, data, previewSize=(150, 150)):
	434	name = os.path.basename(name) # strip any directory and just keep the filename (for security reasons)
	435	fileType = self.GetFileType(name)
	436	if not fileType:
	437	return False #dont allow upload of anything that doesn't have an extension
	438	try:
	439	# write the file
	440	f=open( opj(dirHash[fileType], name), "wb")
	441	f.write( base64.decodestring(data) )
	442	f.close()
	443	del data
	444
	445	self.__MakePreview(name, previewSize)
	446
	447	return True
	448	except:
	449	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	450	return False
	451
	452
	453	def UploadLink(self, url):
	454	name = os.path.basename(url) # strip any directory and just keep the filename (for security reasons)
	455	fileType = self.GetFileType(name)
	456	if not fileType:
	457	return False #dont allow upload of anything that doesn't have an extension
	458
	459	try:
	460	filename, headers = urllib.urlretrieve(url, opj(dirHash[fileType], name))
	461	return True
	462	except:
	463	return False
	464
	465
	466	### looks if there is a preview already saved... if not, create one and send it back
	467	def GetPreview(self, fullPath, previewSize=(150,150)):
	468	try:
	469	res = self.MakeThumbnail(fullPath)
	470	if res:
	471	return (self.__ReadFile(res), True)
	472	else:
	473	return False
	474	except:
	475	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	476	return False
	477
	478
	479	## fullPath = ConvertPath(fullPath)
	480	## if not self.__LegalPath(fullPath):
	481	## return False
	482	## fileType = self.GetFileType(fullPath)
	483
	484	## try:
	485	## if fileType != "image" or not os.path.isfile(fullPath):
	486	## return False
	487	## previewPath = self.__GetPreviewName(fullPath)
	488	## wxImageType = self.__GetWxBitmapType(os.path.basename(fullPath))
	489	## if not os.path.isfile(previewPath): #if the preview doesnt exist, make it
	490	## if wxImageType: # is writing this file type supported?
	491	## im = wx.Image(fullPath) # read the original image
	492	## if not im.Ok():
	493	## return False # the image is probably corrupted
	494	## preview = im.Rescale(previewSize[0], previewSize[1]) # resize it
	495	## im.SaveFile(previewPath, wxImageType) # save it back to a file
	496	## self.__SetWritePermissions(previewPath)
	497	## else: #if writing is not supported, try and read it and resize it on the fly
	498	## im = wx.Image(fullPath) # read the original image
	499	## if not im.Ok():
	500	## return False # the image is probably corrupted
	501	## preview = im.Rescale(previewSize[0], previewSize[1]) # resize it
	502	## data = preview.GetData()
	503	## return (base64.encodestring(data), False) #return the pixels themselves (isBinary=False)
	504
	505	## return (self.__ReadFile(previewPath), True)
	506	## except:
	507	## WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	508	## return False
	509
	510
	511	def MakeThumbnail(self, fullPath):
	512	fullPath = ConvertPath(fullPath)
	513	if not self.__LegalPath(fullPath):
	514	return False
	515	fileType = self.GetFileType(fullPath)
	516
	517	try:
	518	if not os.path.isfile(fullPath):
	519	return False
	520
	521	previewPath = self.__GetPreviewName(fullPath)
	522	if not os.path.isfile(previewPath): #if the preview doesnt exist, make it
	523	makeThumbs.makeThumbnail(fullPath, fileType, previewPath)
	524	if not os.path.isfile(previewPath):
	525	return False
	526
	527	return previewPath
	528	except:
	529	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]))
	530	return False
	531
	532
	533
	534	### get the file ready for loading by the app on the "host" machine
	535	def PrepareFile(self, fullPath, host):
	536	if not self.__LegalPath(fullPath):
	537	return False
	538
	539	# make the connection with the remote FileServer
	540	fileServer = xmlrpclib.ServerProxy("http://"+str(host)+":8800")
	541
	542	try:
	543	# if the file exists already, there's no need to transfer it
	544	filePath = fileServer.FileExists(fullPath, os.stat(fullPath).st_size)
	545	except:
	546	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	547	return False
	548	if filePath:
	549	return filePath # if the file is already cached, just send the path back
	550	else:
	551	try:
	552	# file is not cached so read it and send it to the remote FileServer
	553	fileData = self.__ReadFile(fullPath)
	554	except:
	555	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	556	return False
	557	else:
	558	return fileServer.PutFile(os.path.basename(fullPath), fileData)
	559
	560
	561	### gets the extension of the file that was passed in
	562	def GetFileType(self, thefile):
	563	try:
	564	filename = os.path.basename( thefile )
	565	extension = os.path.splitext(filename)[1].lower()
	566	fileType = False
	567	for fType in types.iterkeys():
	568	if extension in types[ fType ]:
	569	fileType = fType
	570	return fileType
	571	except:
	572	return False
	573
	574
	575	### does file exist in the cache already?? if not, someone will have to send it here
	576	def FileExists(self, fullPath, size):
	577	fullPath = ConvertPath(fullPath)
	578	if os.path.isfile(fullPath): # if the file server is on the same machine then we can
	579	return fullPath # just read the file directly from the file_library
	580	filename = "_".join(os.path.basename(fullPath).split())
	581	if filename == "": # if it's a directory
	582	return False
	583	path = os.path.abspath( opj(CACHE_DIR, filename) )
	584	if os.path.isfile(path) and os.stat(path).st_size==size: #compare the sizes as well
	585	return path
	586	else:
	587	return False
	588
	589
	590	### called by the remote FileServer to upload the file
	591	def PutFile(self, filename, data):
	592	try:
	593	filename = "_".join(os.path.basename(filename).split()) # strip the directory and keep just the name (security reasons)
	594	if filename == "":
	595	return False
	596	f=open( opj(CACHE_DIR, filename), "wb")
	597	f.write( base64.decodestring(data) )
	598	f.close()
	599	self.__SetWritePermissions(opj(CACHE_DIR, filename))
	600
	601	del data
	602	return os.path.abspath( opj(CACHE_DIR, filename) ) #return the path of the file to be passed to the app
	603	except:
	604	del data
	605	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	606	return False
	607
	608
	609	### so that the server can be quit remotely
	610	def Quit(self, var):
	611	global RUN_SERVER
	612	RUN_SERVER = False
	613	return 1
	614
	615
	616
	617
	618
	619	#=======================================#
	620	######## Auxiliary methods ########
	621
	622
	623	### constructs the preview name from the filename
	624	### preview name is: filename+fileSize+original_extension
	625	### (such as trees56893.jpg where 56893 is file size in bytes)
	626	def __GetPreviewName(self, fullPath):
	627	fileType = self.GetFileType(fullPath)
	628
	629	fileSize = os.stat(fullPath).st_size
	630	(root, ext) = os.path.splitext( os.path.basename(fullPath) )
	631	previewName = root+str(fileSize)+ext #construct the preview name
	632
	633	# all the files have .jpg tacked onto the end
	634	previewName += ".jpg"
	635
	636	return opj( THUMB_DIR, previewName )
	637
	638
	639	def __MakePreview(self, name, previewSize):
	640	try:
	641	wxImageType = self.__GetWxBitmapType(name)
	642	if wxImageType: # some types just cant be saved by wxPython
	643	fileType = self.GetFileType(name)
	644	# write the preview but name it with a filename and filesize so that they are unique
	645	previewPath = self.__GetPreviewName(opj(dirHash[fileType], name))
	646	im = wx.Image(opj(dirHash[fileType], name)) # read the original image
	647
	648	if im.Ok(): #the image may be corrupted...
	649	im.Rescale(previewSize[0], previewSize[1]) # resize it
	650	im.SaveFile(previewPath, wxImageType) # save it back to a file
	651	self.__SetWritePermissions(previewPath, opj(dirHash[fileType], name))
	652	except:
	653	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	654	pass # no preview was saved... oh well
	655
	656
	657
	658	### change the permission of the settings file to allow everyone to write to it
	659	def __SetWritePermissions(self, *files):
	660	for filePath in files:
	661	try:
	662	flags = stat.S_IWUSR \| stat.S_IRUSR \| stat.S_IWGRP \| stat.S_IRGRP \| stat.S_IROTH
	663	os.chmod(filePath, flags)
	664	except:
	665	WriteLog( str(sys.exc_info()[0])+" "+str(sys.exc_info()[1])+" "+str(sys.exc_info()[2]) )
	666
	667
	668	### reads a file and encodes it in a string (for transfer over the network)
	669	def __ReadFile(self, fullPath):
	670	f = open(fullPath, "rb")
	671	fileData = base64.encodestring( f.read() )
	672	f.close()
	673	return fileData
	674
	675
	676	### looks through all the files of the same type
	677	### and figures out if the files is already there (optionally compare the file sizes as well)
	678	### if the same file is found, it will return its path
	679	def __FileAlreadyThere(self, filename, fileType, size):
	680	dirPath = self.__GetFilePath(fileType)
	681	fileDir = False
	682
	683	def LookForFile(dirPath, fileType, fileDir):
	684	if fileDir:
	685	return fileDir #this will prevent the recursion to go on if we found a file (base case)
	686	dirPath = ConvertPath(dirPath)
	687	items = os.listdir(dirPath)
	688	files = []
	689	dirs = {}
	690	for item in items: #loop through all the items in this directory (files or dirs)
	691	if os.path.isfile( opj(dirPath, item) ): #is it a file
	692	if item.lower() == filename.lower(): #are their filenames equal
	693	if size==-1 or size==os.stat(opj(dirPath, item)).st_size:
	694	fileDir = dirPath
	695	return dirPath
	696	elif os.path.isdir( opj(dirPath, item) ): #is it a directory
	697	LookForFile( opj(dirPath, item), fileType, fileDir) #if so, search in it
	698	return False
	699
	700	return LookForFile(dirPath, fileType, fileDir)
	701
	702
	703	### recursively traverses the given directory
	704	### it return a tuple of (dir-path, list-of-dirs, files)
	705	def __TraverseDir(self, dirPath, fileType):
	706	dirPath = ConvertPath(dirPath)
	707	items = os.listdir(dirPath)
	708	files = []
	709	dirs = {}
	710	for item in items:
	711	if os.path.isfile( opj(dirPath, item) ):
	712	if self.GetFileType(item): #if the file is supported, add it to the list, otherwise disregard it
	713	files.append(item)
	714	elif os.path.isdir( opj(dirPath, item) ):
	715	dirs[item] = self.__TraverseDir( opj(dirPath, item), fileType )
	716	return (dirPath, dirs, files, fileType)
	717
	718
	719	### you pass it in a filename and it figures out the wx.BITMAP_TYPE and returns it
	720	def __GetWxBitmapType(self, name):
	721	ext = os.path.splitext(name)[1]
	722	ext = ext.lower()
	723	if ext==".jpg": return wx.BITMAP_TYPE_JPEG
	724	elif ext==".png": return wx.BITMAP_TYPE_PNG
	725	elif ext==".gif": return wx.BITMAP_TYPE_GIF
	726	elif ext==".bmp": return wx.BITMAP_TYPE_BMP
	727	elif ext==".pcx": return wx.BITMAP_TYPE_PCX
	728	elif ext==".pnm": return wx.BITMAP_TYPE_PNM
	729	elif ext==".tif": return wx.BITMAP_TYPE_TIF
	730	elif ext==".tiff": return wx.BITMAP_TYPE_TIF
	731	elif ext==".xpm": return wx.BITMAP_TYPE_XPM
	732	elif ext==".ico": return wx.BITMAP_TYPE_ICO
	733	elif ext==".cur": return wx.BITMAP_TYPE_CUR
	734	else:
	735	return False
	736
	737
	738	### gets the default path for a certain file type
	739	def __GetFilePath(self, type=None):
	740	if type == None:
	741	return FILES_DIR
	742	else:
	743	return dirHash[type]
	744
	745
	746
	747	### Checks whether all the paths are still within FILES_DIR (for security issues)
	748	def __LegalPath(self, *pathList):
	749	try:
	750	normedPathList = []
	751	normedPathList.append(FILES_DIR)
	752	for path in pathList:
	753	if path != "":
	754	normedPathList.append( os.path.normpath( os.path.realpath(path) ) ) # normalize all the paths (remove ../ kinda stuff)
	755	return (os.path.commonprefix(normedPathList) == FILES_DIR)
	756	except:
	757	return False #if anything fails, it's safer to assume that it's an illegal action
	758
	759
	760	def __IsFolder(self, fullPath):
	761	return (os.path.splitext(fullPath)[1] == '')
	762
	763
	764
	765
	766	# to output all the error messages to a file
	767	def WriteLog(message):
	768	try:
	769	print message
	770	except:
	771	pass
	772
	773
	774
	775
	776
	777
	778	# a mix-in class for adding the threading support to the XMLRPC server
	779	class ThreadedXMLRPCServer(SocketServer.ThreadingMixIn, SimpleXMLRPCServer):
	780	allow_reuse_address = True
	781
	782
	783	# so that we can quit the server properly
	784	class MyServer(ThreadedXMLRPCServer):
	785	def serve_forever(self):
	786	while RUN_SERVER:
	787	self.handle_request()
	788
	789
	790
	791
	792	def main(argv):
	793
	794	## global REDIRECT
	795	## if len(argv) > 2:
	796	## if "-v" in argv:
	797	## REDIRECT = False
	798	## else:
	799
	800	## # redirect all the output to nothing
	801	## class dummyStream:
	802	## ''' dummyStream behaves like a stream but does nothing. '''
	803	## def __init__(self): pass
	804	## def write(self,data): pass
	805	## def read(self,data): pass
	806	## def flush(self): pass
	807	## def close(self): pass
	808	## # and now redirect all default streams to this dummyStream:
	809	## sys.stdout = dummyStream()
	810	## sys.stderr = dummyStream()
	811	## sys.stdin = dummyStream()
	812	## sys.__stdout__ = dummyStream()
	813	## sys.__stderr__ = dummyStream()
	814	## sys.__stdin__ = dummyStream()
	815
	816
	817	wx.InitAllImageHandlers()
	818
	819	# start the XMLRPC server
	820	server = MyServer(("", 8800), SimpleXMLRPCRequestHandler, logRequests=False)
	821	server.register_instance(FileLibrary())
	822	server.serve_forever()
	823
	824
	825
	826	if __name__ == '__main__':
	827	import sys, os
	828	main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])
	829

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: trunk/src/testing/app/FileViewer/FileServer/fileServer.py @ 4

Download in other formats: