fix: media files on production

.flake8

@@ -5,4 +5,5 @@ exclude =
     manage.py,
     settings.py,
     wsgi.py,
-    venv/
+    venv/,
+    */config/*

django_project/config/production.py

@@ -1,3 +1,4 @@
+import os
 from .base import *
 
 DEBUG = False
@@ -18,5 +19,10 @@
         'PORT': DB_PORT,
     }
     }
+
 STATICFILES_STORAGE = 'whitenoise.storage.CompressedManifestStaticFilesStorage'
-SECRET_KEY = os.getenv("SECRET_KEY", "")
+SECRET_KEY = os.getenv("SECRET_KEY", "")
+
+MEDIA_ROOT = os.path.join("/var/www/","djangoblogs.onrender.com/media/")
+
+os.makedirs(MEDIA_ROOT, exist_ok=True)

django_project/urls.py

@@ -18,7 +18,9 @@
 from django.contrib import admin
 from django.contrib.auth import views as auth_views
 from django.urls import include, path
-
+from django.views.static import serve
+from django.urls import re_path
+from django.conf import settings
 
 from users import views as user_views
 
@@ -40,6 +42,14 @@
     path("", include("blog.urls")),
 ]
 
+if not settings.DEBUG:
+    # deployment fix to server media files
+    urlpatterns += [
+        re_path(r'^media/(?P<path>.*)$', serve,{
+            'document_root': settings.MEDIA_ROOT,
+        })
+    ]
+
 if settings.DEBUG:
     urlpatterns += static(
         settings.MEDIA_URL,

users/tests.py

@@ -1,5 +1,5 @@
 import os
-from django.test import TestCase  # noqa
+from django.test import TestCase 
 from django.contrib.auth.models import User
 
 from django.conf import settings
@@ -8,7 +8,12 @@
 
 class TestRegisterView(TestCase):
     @classmethod
-    def setUp(cls):
+    def setUpTestData(cls):
+        if(os.path.exists(os.path.join(settings.MEDIA_ROOT, "users", "random"))):
+            os.rmdir(os.path.join(settings.MEDIA_ROOT, "users", "random"))
+
+    @classmethod
+    def tearDownClass(cls):
         if(os.path.exists(os.path.join(settings.MEDIA_ROOT, "users", "random"))):
             os.rmdir(os.path.join(settings.MEDIA_ROOT, "users", "random"))
 
@@ -43,7 +48,8 @@ def setUpTestData(cls):
     def tearDownClass(cls):
         media_dir = settings.MEDIA_ROOT
         username = cls.user.username
-        os.rmdir(os.path.join(media_dir, 'users', username))
+        if(os.path.exists(os.path.join(media_dir, 'users', username))):
+            os.rmdir(os.path.join(media_dir, 'users', username))
 
     def test_get_login_view(self):
         response = self.client.get("/login/")