import os
from django.conf import settings


def ensure_dirs():
    path = os.path.join(settings.MEDIA_ROOT, 's3', 'uploads')
    if not os.path.exists(path):
        os.makedirs(path)

    path = os.path.join(settings.MEDIA_ROOT, 's3', 'downloads')
    if not os.path.exists(path):
        os.makedirs(path)

@@ -54,16 +54,24 @@
 re_dirs(
+add=None
 ):%0A    p
@@ -290,28 +290,162 @@
 :%0A        os.makedirs(path)%0A
+%0A    if add:%0A        path = os.path.join(settings.MEDIA_ROOT, add)%0A        if not os.path.exists(path):%0A            os.makedirs(path)%0A

Fix test dirs