def zip(self, path, file=None, override=None, signal=False):
"""
Zip file or directory ``path`` to ``file`` or to ``path + '.zip'``.
On not exist raise :class:`~limited.core.files.storage.FileNotExist`.
"""
path = self.check(path)
if not self.fs.exists(path):
raise FileNotExist(u"'%s' not found" % path)
if file == None:
file = self.available_name(path + u".zip", override=override)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
newfile = file + u".part"
try:
zfile = self.fs.open(newfile, mode='wb')
archive = zipfile.ZipFile(zfile, 'w', zipfile.ZIP_DEFLATED)
if self.fs.isdir(path):
dirname = FilePath.name(path)
for abspath, name in self.fs.listfiles(path).items():
name = FilePath.join(dirname, name)
archive.write(abspath, name)
elif self.fs.isfile(path):
archive.write(self.fs.abspath(path), FilePath.name(path))
archive.close()
zfile.seek(0)
self.fs.rename(newfile, FilePath.name(file))
except EnvironmentError as e:
if e.errno == errno.EACCES:
raise FileError(u"IOError, zip. Permission denied '%s'" % path)
finally:
if self.fs.exists(newfile):
self.fs.remove(newfile)
def clear(self, path, older=None, signal=True):
"""
Remove all files and dirs in ``path`` directory.
``older`` takes seconds for max age from created_time, only top sub dirs checked.
On not exist raise :class:`~limited.core.files.storage.FileNotExist`.
On not directory raise :class:`~limited.core.files.storage.FileError`.
"""
path = self.check(path)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
if not self.fs.exists(path):
raise FileNotExist(u"'%s' not found" % path)
if not self.fs.isdir(path):
raise FileError(u"'%s' not directory" % path)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
if older == None:
for item in self.fs.list(path):
file = FilePath.join(path, item)
self.fs.remove(file)
else:
for item in self.fs.list(path):
file = FilePath.join(path, item)
chenaged = self.fs.created_time(file)
if datetime.now() - chenaged > timedelta(seconds=older):
self.fs.remove(file)
def open(self, path, mode='rb', signal=True):
"""
Return open django :class:`~django.core.files.base.File` instance
"""
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
path = self.check(path)
return self.fs.open(path, mode)
def create(self, path, content, signal=True):
"""
Write to file ``path`` same data ``content``
"""
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
newfile = self.fs.open(path, 'wb')
newfile.write(content)
newfile.close()
def remove(self, path, signal=True):
"""
Remove directory or file, on not exist raise :class:`~limited.core.files.storage.FileNotExist`
"""
path = self.check(path)
if not self.exists(path):
raise FileNotExist(u"'%s' not found" % path)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
self.fs.remove(path)
def save(self, path, file, override=True, signal=True):
"""
Return path to the file, that can be another from ``path``.
Copy to disk to ``path`` open :class:`~django.core.files.base.File` object ``file``.
Also you need to close it yourself.
"""
path = self.check(path)
path = self.available_name(path, override)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
return self.fs.save(path, file)
def totrash(self, path, signal=True):
"""
Shortcut for :func:`~limited.core.files.storage.FileStorage.move`
where second var is :ref:`LIMITED_TRASH_PATH <SETTINGS_TRASH_PATH>`.
"""
path = self.check(path)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
if not self.fs.exists(settings.LIMITED_TRASH_PATH):
self.fs.mkdir(settings.LIMITED_TRASH_PATH)
if not self.fs.exists(path):
raise FileNotExist(u"'%s' not found" % path)
self.fs.move(path, settings.LIMITED_TRASH_PATH)
def unzip(self, path, override=True, signal=False):
"""
Unzip file or directory ``path``.
On not exist raise :class:`~limited.core.files.storage.FileNotExist`.
"""
path = self.check(path)
if not self.fs.exists(path):
raise FileNotExist(u"'%s' not found" % path)
file = self.fs.abspath(path)
directory = FilePath.dirname(path)
# To lazy to do converting
# maybe chardet help later
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
zip = None
try:
zip = zipfile.ZipFile(file, 'r')
for name in zip.namelist():
try:
unicode_name = unicode(name)
except UnicodeDecodeError:
unicode_name = unicode(name.decode('cp866'))
full_path = FilePath.join(directory, unicode_name)
dir_path = FilePath.dirname(unicode_name)
if dir_path != '':
tmp = directory
for item in FilePath.split(dir_path):
tmp = FilePath.join(tmp, item)
if not self.fs.exists(tmp):
self.fs.mkdir(tmp)
if not unicode_name.endswith('/'):
with self.fs.open(full_path, 'w') as f:
f.write(zip.open(name).read())
except UnicodeDecodeError as e:
raise FileError(u"Unicode decode error '%s', try unzip yourself" %
path)
except zipfile.BadZipfile as e:
raise FileError(u"Bad zip file '%s'" % path)
except EnvironmentError as e:
if e.errno == errno.EACCES:
raise FileError(u"IOError, unzip. Permission denied '%s'" %
path)
finally:
if zip:
zip.close()
def rename(self, path, name, signal=True):
"""
Rename file or dir path ``path`` to name ``name``.
On '/' in ``name`` raise :class:`~limited.core.files.storage.FileError`.
On not exist or already exist raise :class:`~limited.core.files.storage.FileNotExist`.
"""
path = self.check(path)
if '/' in name:
raise FileError(u"'%s' contains not supported symbols" % name)
if not self.exists(path):
raise FileNotExist(u"'%s' not found" % path)
new_path = FilePath.join(FilePath.dirname(path), name)
if self.exists(new_path):
raise FileError(u"'%s' already exist!" % name)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
self.fs.rename(path, name)
def move(self, src, dst, signal=True):
"""
Move file or dir from ``src`` to ``dst``.
On the same directory raise :class:`~limited.core.files.storage.FileError`.
On not exist for both paths raise :class:`~limited.core.files.storage.FileNotExist`.
"""
src_dir = FilePath.dirname(src)
if src == dst or src_dir == dst:
raise FileError(u"Moving to the same directory")
if not self.exists(src):
raise FileNotExist(u"'%s' not found" % src)
if not self.exists(dst):
raise FileNotExist(u"'%s' not found" % dst)
src = self.check(src)
dst = self.check(dst)
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(src))
file_pre_change.send(self, basedir=dst)
self.fs.move(src, dst)
def download(self, url, path, signal=True):
"""
Download file from ``url`` to file ``path``.
The process goes to a file ``path + '.part'``.
On error file will be remove.
To get file name from url use :class:`~limited.core.utils.url_get_filename`.
"""
path = self.check(path)
path = self.fs.available_name(path)
newfile = path + u".part"
try:
# simple hook to stop File proxy access field 'name'
# that is no exists
if signal:
file_pre_change.send(self, basedir=FilePath.dirname(path))
data = urllib.urlopen(iri_to_uri(url))
data.size = int(data.info()['Content-Length'])
self.fs.save(newfile, File(data))
self.fs.rename(newfile, FilePath.name(path))
except Exception:
if self.fs.exists(newfile):
self.fs.remove(newfile)
raise
