--- versionedFile/extVersionedFile.py 2007/09/10 16:34:16 1.7 +++ versionedFile/extVersionedFile.py 2007/10/26 22:44:30 1.14 @@ -29,6 +29,7 @@ import os.path import urllib import time import logging +import types try: from Products.ECHO_content.ECHO_collection import ECHO_basis @@ -134,8 +135,9 @@ class extVersionedFileFolder(Folder,ECHO {'label':'Generate Index.html','action':'generateIndexHTML'}, {'label':'Generate Image Index.html','action':'generateIndexHTML_image'}, {'label':'Generate history_template.html','action':'generateHistoryHTML'}, - {'label':'Import Folder','action':'importFolderForm'}, - {'label':'Export Folder','action':'exportFolder'}, + {'label':'Import directory','action':'importFolderForm'}, + {'label':'Export as file','action':'exportFolder'}, + {'label':'Import versionedFileFolder','action':'importVersionedFileFolderForm'}, {'label':'Position of version number','action':'changeHistoryFileNamesForm'}, ) @@ -186,17 +188,68 @@ class extVersionedFileFolder(Folder,ECHO return pt() def importFolder(self,path,comment="",author=None,lockedBy=None,RESPONSE=None): - """importiere inhalt eines folders""" - + """import contents of a folder on the server""" for fileName in os.listdir(path): - if os.path.isfile(os.path.join(path,fileName)): - manage_addextVersionedFile(self,fileName,'','') - id=fileName - ob=self._getOb(fileName) - ob.title=id - file2=file(os.path.join(path,fileName)) + fn = os.path.join(path,fileName) + if os.path.isfile(fn): + f = file(fn) + self.addFile(vC=comment, file=f, author=author) + + if RESPONSE: + RESPONSE.redirect(self.REQUEST['URL1']) + + def importVersionedFileFolderForm(self): + """form fuer versionedFileFolder import""" + pt=PageTemplateFile(os.path.join(package_home(globals()),'zpt','importVersionedFileFolderForm.zpt')).__of__(self) + return pt() + + def importVersionedFileFolder(self,path,RESPONSE=None): + """import contents of a versionedFileFolder on the server""" + vff = getattr(self.aq_parent, path, None) + if vff is None: + return "SORRY, unable to import %s"%path + + tmpPath=tempfile.mktemp() + if not os.path.exists(tempfile.tempdir): + os.mkdir(tempfile.tempdir) + + if not os.path.exists(tmpPath): + os.mkdir(tmpPath) + + for (vfn, vf) in vff.getVersionedFiles(): + if vf.meta_type == 'versionedFile': + logging.error("importvff: importing %s of type %s!"%(vfn,vf.meta_type)) + title = vf.title + fob = vf.getLastVersion() + author = fob.getLastEditor() + vc = fob.getVersionComment() + # save file to filesystem + savePath=os.path.join(tmpPath,title) + fh=file(savePath,"w") + data = vf.getLastVersion().data + if isinstance(data, str): + # simple data object + fh.write(data) + else: + # chained data objects + while data is not None: + fh.write(data.data) + data = data.next + fh.close() + # and read in again + fh = file(savePath) + logging.error("importvff: comment=%s author=%s!"%(vc,author)) + self.addFile(vC=vc, file=fh, author=author) + # copy more fields + newfob = getattr(self, vfn).getContentObject() + newfob.vComment = fob.vComment + newfob.time = fob.time + logging.error("importvff: vc=%s time=%s of %s!"%(fob.vComment,fob.time,fob.getId())) + + else: + logging.error("importvff: unable to import %s of type %s!"%(vfn,vf.meta_type)) - obj=ob.manage_addextVersionedFileObject(id,comment,author,file2,content_type='') + shutil.rmtree(tmpPath) if RESPONSE: RESPONSE.redirect(self.REQUEST['URL1']) @@ -216,6 +269,7 @@ class extVersionedFileFolder(Folder,ECHO downloadZip=generateDownloadZip(self,self.absolute_url()) downloadZip() return downloadZip.getResult() + ## if not threadName or threadName=="": ## threadStart=generateDownloadZip(self,self.absolute_url()) ## thread=Thread(target=threadStart) @@ -435,10 +489,18 @@ class extVersionedFileFolder(Folder,ECHO return out() - def addFile(self,vC,file,author,newName='',content_type='',RESPONSE=None): + def addFile(self,vC,file,author='',newName='',content_type='',RESPONSE=None): """ add a new file""" + # is file is a real file or a zope download object? + isRealFile = type(file) is types.FileType + if newName=='': - filename=file.filename + logging.debug("fileobject: %s real:%s"%(repr(file),repr(isRealFile))) + if isRealFile: + filename = file.name + else: + filename=file.filename + id=filename[max(filename.rfind('/'), filename.rfind('\\'), filename.rfind(':'), @@ -447,24 +509,32 @@ class extVersionedFileFolder(Folder,ECHO else: id=newName - vC=self.REQUEST.form['vC'] - manage_addextVersionedFile(self,id,'','') + if vC is None: + vC=self.REQUEST.form['vC'] + + # get new extVersionedFile + vf = self._newVersionedFile(id,title=id) #if (getattr(self,'commentNonEmpty',0)==1) and vC.strip()=="": + # add file to this folder + self._setObject(id,vf) + # add its content + logging.info("ADD: %s"%repr(vf)) + obj=vf.addContentObject(id,vC,author=author,file=file,content_type=content_type,from_tmp=isRealFile) - ob=self._getOb(id) - ob.title=id - file2=file - - logging.info("ADD: %s"%repr(ob)) - obj=ob.manage_addVersionedFileObject(id,vC,author,file2,content_type=content_type) - self.REQUEST.SESSION['objID']=ob.getId() + self.REQUEST.SESSION['objID']=vf.getId() self.REQUEST.SESSION['objID_parent']=None if obj.getSize()==0: pt=PageTemplateFile(os.path.join(package_home(globals()),'zpt','errorUploadFile')).__of__(self) return pt() - RESPONSE.redirect(self.REQUEST['URL1']) + if RESPONSE is not None: + RESPONSE.redirect(self.REQUEST['URL1']) + + + def _newVersionedFile(self, id, title='', lockedBy=None, author=None): + """factory for versioned files. to be overridden in derived classes.""" + return extVersionedFile(id, title, lockedBy=lockedBy, author=author) def deleteEmptyObject(self,submit,RESPONSE=None): @@ -485,7 +555,11 @@ class extVersionedFileFolder(Folder,ECHO """fix last version number of all files""" for (id,vf) in self.getVersionedFiles(): vf.fixVersionNumbers() + # recursively + for (id,vf) in self.objectItems(self.meta_type): + vf.fixVersionNumbers() + manage_addextVersionedFileFolderForm=DTMLFile('dtml/extfolderAdd', globals()) @@ -847,7 +921,8 @@ class extVersionedFile(CatalogAware,Fold def getContentObject(self): """returns the last version object""" - if not self.lastVersionId: + if not hasattr(self, 'lastVersionId'): + # find last version and save it lv = self.findLastVersion() if lv is None: return None @@ -998,10 +1073,10 @@ class extVersionedFile(CatalogAware,Fold self.lastVersionNumber = versNum self.lastVersionId = id - logging.debug("addcontentobject: lastversion=%s"%self.getData()) + #logging.debug("addcontentobject: lastversion=%s"%self.getData()) logging.debug("reindex1: %s in %s"%(repr(self),repr(self.default_catalog))) self.reindex_object() - logging.debug("addcontentobject: fob_data=%s"%fob.getData()) + #logging.debug("addcontentobject: fob_data=%s"%fob.getData()) return fob