def _GetTrashcanDirectory(self, id):
aP = DvPath()
aP.SetStr(str(self.root))
aP.AppendSeperator()
aP.AppendDirectory(self.Trashcan)
aP.AppendSeperator()
aP.AppendDirectory(self._GetDirectory(id))
aP.AppendSeperator()
return aP
def _CreatePath(self, key, filename):
"""
Create the physical path of the file
"""
root = str(self.fileentry().pool.root)
aP = DvPath(root)
aP.AppendSeperator()
aP.AppendDirectory(self.fileentry().pool._GetDirectory(self.fileentry().id))
aP.AppendSeperator()
aP.SetName(u"%06d_%s_" % (self.fileentry().id, key))
aP.SetExtension(DvPath(filename).GetExtension())
return aP.GetStr()
def test_Dirs(self):
n = self.base
p = DvPath(n)
self.assert_(str(p) == n)
p.Append("another_dir" + os.sep + "and.file")
self.assert_(str(p) == n + "another_dir" + os.sep + "and.file", str(p))
p = DvPath(n)
p.AppendDirectory("the_last")
self.assert_(str(p) == n + "the_last" + os.sep, str(p))
p.RemoveLastDirectory()
self.assert_(str(p) == n)
p = DvPath(n[:-1])
p.AppendSeperator()
self.assert_(str(p) == self.base)
def _Path(self, absolute = True):
"""
Get the physical path of the file. Checks the database.
"""
if self.tempfile or not self.path:
return u""
root = str(self.fileentry().pool.root)
if absolute and self.path[:len(root)] != root:
path = DvPath(root)
path.AppendSeperator()
path.Append(self.path)
else:
path = DvPath(self.path)
return path.GetStr()
class FileManager(object):
"""
Data Pool File Manager class for SQL Database with version support.
Files are stored in filesystem, aditional information in database table.
Table "pool_files" ("id", "fileid", "filekey", "path", "filename", "size", "extension", "version").
Field path stores internal path to the file in filesystem without root.
Preperty descriptions are dictionaries with key:value pairs.
Property values:
id = unit id to store file for (id is required)
version = the version of the file
filekey = custom value
key:
id_filekey_version
directory structure:
root/id[-4:-2]00/id_filekey_version.ext
"""
DirectoryCnt = -4 # directory id range limit
FileTable = u"pool_files" # file table name
FileTableFields = (u"id", u"fileid", u"filekey", u"path", u"filename", u"size", u"extension", u"version")
Trashcan = u"_trashcan"
def GetFileClass(self):
"""
Returns the required file class for File object instantiation
"""
return File
def InitFileStorage(self, root, connectionParam):
"""
Set the local root path for files
"""
self.root = DvPath()
self.root.SetStr(root)
if root == u"":
return
self.root.AppendSeperator()
self.root.CreateDirectoriesExcp()
def SearchFilename(self, filename):
"""
search for filename
"""
return self.SearchFiles({u"filename": filename})
def SearchFiles(self, parameter, sort=u"filename", start=0, max=100, ascending = 1, **kw):
"""
search files
"""
flds = self.FileTableFields
kw["singleTable"] = 1
sql, values = self.FmtSQLSelect(flds, parameter, dataTable=self.FileTable, sort = sort, start=start, max=max, ascending = ascending, **kw)
files = self.Query(sql, values)
f2 = []
for f in files:
f2.append(self.ConvertRecToDict(f, flds))
return f2
def DeleteFiles(self, id, cursor=None, version=None):
"""
Delete the file with the prop description
"""
files = self.SearchFiles({u"id":id}, sort=u"id")
if not files:
return True
entry = self.GetEntry(id, version=version)
for f in files:
file = self.GetFileClass()(filedict=f,fileentry=entry)
file.delete()
if len(files):
sql = u"delete from %s where id = %d" % (self.FileTable, id)
self.Query(sql, cursor=cursor, getResult=False)
return True
# Internal --------------------------------------------------------------
def _GetDirectory(self, id):
"""
construct directory path without root
"""
return (u"%06d" % (id))[self.DirectoryCnt:-2] + u"00/" + (u"%06d" % (id))[self.DirectoryCnt+2:]
def _MoveToTrashcan(self, path, id):
if not self.useTrashcan:
return path.Delete()
aP = self._GetTrashcanDirectory(id)
aP.SetNameExtension(path.GetNameExtension())
if aP.Exists():
aP.Delete()
return path.Rename(str(aP))
def _GetTrashcanDirectory(self, id):
aP = DvPath()
aP.SetStr(str(self.root))
aP.AppendSeperator()
aP.AppendDirectory(self.Trashcan)
aP.AppendSeperator()
aP.AppendDirectory(self._GetDirectory(id))
aP.AppendSeperator()
return aP
