Kaydet (Commit) e4025563 authored tarafından chillaranand's avatar chillaranand Kaydeden (comit) Tim Graham

Fixed #27836 -- Allowed FileSystemStorage.delete() to remove directories.

üst 339d526d
......@@ -293,12 +293,15 @@ class FileSystemStorage(Storage):
def delete(self, name):
assert name, "The name argument is not allowed to be empty."
name = self.path(name)
# If the file exists, delete it from the filesystem.
# If the file or directory exists, delete it from the filesystem.
try:
os.remove(name)
if os.path.isdir(name):
os.rmdir(name)
else:
os.remove(name)
except FileNotFoundError:
# If os.remove() fails with FileNotFoundError, the file may have
# been removed concurrently.
# If removal fails, the file or directory may have been removed
# concurrently.
pass
def exists(self, name):
......
......@@ -495,6 +495,11 @@ class FileStorageTests(SimpleTestCase):
with self.assertRaises(AssertionError):
self.storage.delete('')
def test_delete_deletes_directories(self):
tmp_dir = tempfile.TemporaryDirectory(dir=self.storage.location)
self.storage.delete(tmp_dir.name)
self.assertFalse(os.path.exists(tmp_dir.name))
@override_settings(
MEDIA_ROOT='media_root',
MEDIA_URL='media_url/',
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment