Feito upload de arquivos publicos ou privados

backend/README.md

@@ -43,3 +43,19 @@ acordo com sua necessidade.
 Para ler configurações de instância esse projeto usa a lib [python decouple](https://pypi.org/project/python-decouple/).
 Ela é importada no arquivo settings.py
 
+## Uploads de arquivos
+
+O projeto está configurado para gravar arquivos publicos ou privados. Por isso, se usar ImageField ou FileField,
+O parâmetro "upldat_to" precisar ter prefixo "public" para arquivos que sejam publicos e "private" para privados. Exemplos:
+
+```python
+from django.db.models import Model, ImageField, FileField
+
+class UserProfile(Model):
+    avatar = ImageField(upload_to='public/base/avatars/', null=True, blank=True)
+
+class NotaFiscal(Model):
+    arquivo = FileField(upload_to='private/base/notas_fiscais/')
+
+```
+

backend/devpro/base/admin.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+from django.contrib.admin import ModelAdmin
 from django_min_custom_user.admin import MinUserAdmin
 
 from devpro.base.models import User

backend/devpro/base/models.py

@@ -1,3 +1,4 @@
+from django.db.models import Model
 from django_min_custom_user.models import MinAbstractUser
 
 

backend/devpro/s3_file_handlers.py

@@ -0,0 +1,29 @@
+from storages.backends.s3 import S3Storage
+
+
+class FileConfigurationError(Exception):
+    pass
+
+
+class S3FileStorage(S3Storage):
+    def url(self, name, parameters=None, expire=None, http_method=None):
+        url = super().url(name, parameters, expire, http_method)
+        if name.startswith('public'):
+            url = url.split('?')[0]
+        return url
+
+    def get_object_parameters(self, name):
+        parameters = super().get_object_parameters(name)
+        self._validate(name)
+        if self._startswith(name, 'public'):
+            parameters['ACL'] = 'public-read'
+        else:
+            parameters['ACL'] = 'private'
+        return parameters
+
+    def _validate(self, name: str):
+        if not (self._startswith(name, 'public') or self._startswith(name, 'private')):
+            raise FileConfigurationError(f'property "upload_to" must start with "public" or "private" and it is {name}')
+
+    def _startswith(self, name, prefix):
+        return name.startswith(f'{self.location}/{prefix}')

backend/devpro/settings.py

@@ -124,17 +124,27 @@
     # Configuração para coletar estáticos para o Nginx
     STATIC_URL = 'static/'
     STATIC_ROOT = BASE_DIR.parent / 'docker/staticfiles/static'
+    MEDIA_ROOT = BASE_DIR.parent / 'docker/mediafiles'
+    MEDIA_URL = '/mediafiles/'
 else:
     STATIC_URL = f'//{AWS_STORAGE_BUCKET_NAME}.s3.amazonaws.com/static/'
 
     AWS_S3_ACCESS_KEY_ID = config('AWS_S3_ACCESS_KEY_ID')
     AWS_S3_SECRET_ACCESS_KEY = config('AWS_S3_SECRET_ACCESS_KEY')
     STORAGES = {
+        "default": {
+            "BACKEND": "devpro.s3_file_handlers.S3FileStorage",
+            "OPTIONS": {
+                'default_acl': 'private',
+                'location': 'media',
+            },
+        },
         "staticfiles": {
             "BACKEND": "storages.backends.s3.S3Storage",
             "OPTIONS": {
                 'default_acl': 'public-read',
                 'location': 'static',
+                'querystring_auth': False
             },
         },
     }

backend/devpro/urls.py

@@ -14,9 +14,14 @@
     1. Import the include() function: from django.urls import include, path
     2. Add a URL to urlpatterns:  path('blog/', include('blog.urls'))
 """
+from django.conf import settings
+from django.conf.urls.static import static
 from django.contrib import admin
 from django.urls import path
 
 urlpatterns = [
     path('admin/', admin.site.urls),
 ]
+
+if settings.AWS_STORAGE_BUCKET_NAME == '':
+    urlpatterns.extend(static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT))

backend/pyproject.toml

@@ -14,6 +14,7 @@ dj-database-url = "^2.2.0"
 python-decouple = "^3.8"
 django-min-custom-user = "^0.2.0"
 django-storages = {extras = ["s3"], version = "^1.14.4"}
+pillow = "^10.4.0"
 
 
 [tool.poetry.group.dev]