check

apps/establishment/models.py

@@ -138,6 +138,9 @@ class EstablishmentQuerySet(models.QuerySet):
                                                         public_mark__isnull=False)
         if establishment_qs.exists():
             establishment = establishment_qs.first()
+
+            # TODO fix error:
+            # AttributeError: 'NoneType' object has no attribute 'coordinates'
             return self.exclude(slug=establishment_slug) \
                        .filter(is_publish=True,
                                image_url__isnull=False,

apps/establishment/serializers/back.py

@@ -7,6 +7,9 @@ from establishment.serializers import (
     EstablishmentBaseSerializer, PlateSerializer, ContactEmailsSerializer,
     ContactPhonesSerializer, SocialNetworkRelatedSerializers, EstablishmentDetailSerializer
 )
+
+from utils.decorators import with_base_attributes
+
 from main.models import Currency
 from utils.serializers import TJSONSerializer
 
@@ -124,7 +127,10 @@ class ContactEmailBackSerializers(PlateSerializer):
         ]
 
 
+# TODO: test decorator
+@with_base_attributes
 class EmployeeBackSerializers(serializers.ModelSerializer):
+
     """Social network serializers."""
     class Meta:
         model = models.Employee
@@ -132,4 +138,5 @@ class EmployeeBackSerializers(serializers.ModelSerializer):
             'id',
             'user',
             'name'
-        ]
+        ]
+

apps/establishment/serializers/common.py

@@ -315,20 +315,22 @@ class EstablishmentFavoritesCreateSerializer(serializers.ModelSerializer):
     def validate(self, attrs):
         """Override validate method"""
         # Check establishment object
-        establishment_id = self.context.get('request').parser_context.get('kwargs').get('pk')
+        establishment_slug = self.context.get('request').parser_context.get('kwargs').get('slug')
-        establishment_qs = models.Establishment.objects.filter(id=establishment_id)
+        establishment_qs = models.Establishment.objects.filter(slug=establishment_slug)
 
-        # Check establishment obj by pk from lookup_kwarg
+        # Check establishment obj by slug from lookup_kwarg
         if not establishment_qs.exists():
             raise serializers.ValidationError({'detail': _('Object not found.')})
+        else:
+            establishment = establishment_qs.first()
 
         # Check existence in favorites
         if self.get_user().favorites.by_content_type(app_label='establishment',
                                                      model='establishment')\
-                                    .by_object_id(object_id=establishment_id).exists():
+                                    .by_object_id(object_id=establishment.id).exists():
             raise utils_exceptions.FavoritesError()
 
-        attrs['establishment'] = establishment_qs.first()
+        attrs['establishment'] = establishment
         return attrs
 
     def create(self, validated_data, *args, **kwargs):

apps/establishment/tests.py

@@ -26,7 +26,7 @@ class BaseTestCase(APITestCase):
         self.establishment_type = EstablishmentType.objects.create(name="Test establishment type")
 
 
-class EstablishmentTests(BaseTestCase):
+class EstablishmentBTests(BaseTestCase):
     def test_establishment_CRUD(self):
         params = {'page': 1, 'page_size': 1,}
         response = self.client.get('/api/back/establishments/', params, format='json')
@@ -92,7 +92,8 @@ class ChildTestCase(BaseTestCase):
         self.establishment = Establishment.objects.create(
             name="Test establishment",
             establishment_type_id=self.establishment_type.id,
-            is_publish=True
+            is_publish=True,
+            slug="test"
         )
 
 
@@ -262,3 +263,89 @@ class EstablishmentShedulerTests(ChildTestCase):
 
         response = self.client.delete(f'/api/back/establishments/{self.establishment.id}/schedule/1/')
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+# Web tests
+class EstablishmentWebTests(BaseTestCase):
+
+    def test_establishment_Read(self):
+        params = {'page': 1, 'page_size': 1,}
+        response = self.client.get('/api/web/establishments/', params, format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class EstablishmentWebTagTests(BaseTestCase):
+
+    def test_tag_Read(self):
+        response = self.client.get('/api/web/establishments/tags/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class EstablishmentWebSlugTests(ChildTestCase):
+
+    def test_slug_Read(self):
+        response = self.client.get(f'/api/web/establishments/slug/{self.establishment.slug}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class EstablishmentWebSimilarTests(ChildTestCase):
+
+    def test_similar_Read(self):
+        response = self.client.get(f'/api/web/establishments/slug/{self.establishment.slug}/similar/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class EstablishmentWebCommentsTests(ChildTestCase):
+
+    def test_comments_CRUD(self):
+
+        response = self.client.get(f'/api/web/establishments/slug/{self.establishment.slug}/comments/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = {
+            'text': 'test',
+            'user': self.user.id,
+            'mark': 4
+        }
+
+        response = self.client.post(f'/api/web/establishments/slug/{self.establishment.slug}/comments/create/',
+                                    data=data)
+
+        comment = response.json()
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.get(f'/api/web/establishments/slug/{self.establishment.slug}/comments/{comment["id"]}/',
+                                   format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        update_data = {
+            'text': 'Test new establishment'
+        }
+
+        response = self.client.patch(f'/api/web/establishments/slug/{self.establishment.slug}/comments/{comment["id"]}/',
+                                     data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        response = self.client.delete(
+            f'/api/web/establishments/slug/{self.establishment.slug}/comments/{comment["id"]}/',
+            format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+class EstablishmentWebFavoriteTests(ChildTestCase):
+
+    def test_favorite_CR(self):
+        data = {
+            "user": self.user.id,
+            "object_id": self.establishment.id
+        }
+
+        response = self.client.post(f'/api/web/establishments/slug/{self.establishment.slug}/favorites/',
+                                    data=data)
+
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.delete(
+            f'/api/web/establishments/slug/{self.establishment.slug}/favorites/',
+            format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)

apps/establishment/urls/common.py

@@ -8,13 +8,13 @@ app_name = 'establishment'
 urlpatterns = [
     path('', views.EstablishmentListView.as_view(), name='list'),
     path('tags/', views.EstablishmentTagListView.as_view(), name='tags'),
-    path('<slug:slug>/', views.EstablishmentRetrieveView.as_view(), name='detail'),
+    path('slug/<slug:slug>/', views.EstablishmentRetrieveView.as_view(), name='detail'),
-    path('<slug:slug>/similar/', views.EstablishmentSimilarListView.as_view(), name='similar'),
+    path('slug/<slug:slug>/similar/', views.EstablishmentSimilarListView.as_view(), name='similar'),
-    path('<slug:slug>/comments/', views.EstablishmentCommentListView.as_view(), name='list-comments'),
+    path('slug/<slug:slug>/comments/', views.EstablishmentCommentListView.as_view(), name='list-comments'),
-    path('<slug:slug>/comments/create/', views.EstablishmentCommentCreateView.as_view(),
+    path('slug/<slug:slug>/comments/create/', views.EstablishmentCommentCreateView.as_view(),
          name='create-comment'),
-    path('<slug:slug>/comments/<int:comment_id>/', views.EstablishmentCommentRUDView.as_view(),
+    path('slug/<slug:slug>/comments/<int:comment_id>/', views.EstablishmentCommentRUDView.as_view(),
          name='rud-comment'),
-    path('<slug:slug>/favorites/', views.EstablishmentFavoritesCreateDestroyView.as_view(),
+    path('slug/<slug:slug>/favorites/', views.EstablishmentFavoritesCreateDestroyView.as_view(),
          name='add-to-favorites')
 ]

apps/location/tests.py

@@ -1,3 +1,193 @@
-from django.test import TestCase
+import json
 
-# Create your tests here.
+from rest_framework.test import APITestCase
+from account.models import User
+from rest_framework import status
+from http.cookies import SimpleCookie
+
+from location.models import City, Region, Country
+
+
+class BaseTestCase(APITestCase):
+
+    def setUp(self):
+        self.username = 'sedragurda'
+        self.password = 'sedragurdaredips19'
+        self.email = 'sedragurda@desoz.com'
+        self.newsletter = True
+        self.user = User.objects.create_user(
+            username=self.username, email=self.email, password=self.password)
+
+        # get tokens
+
+        tokkens = User.create_jwt_tokens(self.user)
+        self.client.cookies = SimpleCookie(
+            {'access_token': tokkens.get('access_token'),
+             'refresh_token': tokkens.get('refresh_token')})
+
+
+class CountryTests(BaseTestCase):
+
+    def test_country_CRUD(self):
+        response = self.client.get('/api/back/location/countries/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = {
+            'name': 'Test country',
+            'code': 'test'
+        }
+
+        response = self.client.post('/api/back/location/countries/', data=data, format='json')
+        response_data = response.json()
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.get(f'/api/back/location/countries/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        update_data = {
+            'name': json.dumps({"en-GB": "Test new country"})
+        }
+
+        response = self.client.patch(f'/api/back/location/countries/{response_data["id"]}/', data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        response = self.client.delete(f'/api/back/location/countries/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+class RegionTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+        self.country = Country.objects.create(
+            name=json.dumps({"en-GB": "Test country"}),
+            code="test"
+        )
+
+    def test_region_CRUD(self):
+        response = self.client.get('/api/back/location/regions/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = {
+            'name': 'Test country',
+            'code': 'test',
+            'country_id': self.country.id
+        }
+
+        response = self.client.post('/api/back/location/regions/', data=data, format='json')
+        response_data = response.json()
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.get(f'/api/back/location/regions/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        update_data = {
+            'name': json.dumps({"en-GB": "Test new country"})
+        }
+
+        response = self.client.patch(f'/api/back/location/regions/{response_data["id"]}/', data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        response = self.client.delete(f'/api/back/location/regions/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+class CityTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
+        self.country = Country.objects.create(
+            name=json.dumps({"en-GB": "Test country"}),
+            code="test"
+        )
+
+        self.region = Region.objects.create(
+            name="Test region",
+            code="812",
+            country=self.country
+        )
+
+    def test_city_CRUD(self):
+        response = self.client.get('/api/back/location/cities/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = {
+            'name': 'Test country',
+            'code': 'test',
+            'country_id': self.country.id,
+            'region_id': self.region.id
+        }
+
+        response = self.client.post('/api/back/location/cities/', data=data, format='json')
+        response_data = response.json()
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.get(f'/api/back/location/cities/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        update_data = {
+            'name': json.dumps({"en-GB": "Test new country"})
+        }
+
+        response = self.client.patch(f'/api/back/location/cities/{response_data["id"]}/', data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        response = self.client.delete(f'/api/back/location/cities/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+class AddressTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
+        self.country = Country.objects.create(
+            name=json.dumps({"en-GB": "Test country"}),
+            code="test"
+        )
+
+        self.region = Region.objects.create(
+            name="Test region",
+            code="812",
+            country=self.country
+        )
+
+        self.city = City.objects.create(
+            name="Test region",
+            code="812",
+            region=self.region,
+            country=self.country
+        )
+
+    def test_address_CRUD(self):
+        response = self.client.get('/api/back/location/addresses/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = {
+            'city_id': self.city.id,
+            'number': '+79999999',
+            "coordinates": {
+                "latitude": 37.0625,
+                "longitude": -95.677068
+            },
+            "geo_lon": -95.677068,
+            "geo_lat": 37.0625
+        }
+
+        response = self.client.post('/api/back/location/addresses/', data=data, format='json')
+        response_data = response.json()
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.get(f'/api/back/location/addresses/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        update_data = {
+            'number': '+79999991'
+        }
+
+        response = self.client.patch(f'/api/back/location/addresses/{response_data["id"]}/', data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        response = self.client.delete(f'/api/back/location/addresses/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)

apps/main/serializers.py

@@ -136,8 +136,8 @@ class CarouselListSerializer(serializers.ModelSerializer):
     """Serializer for retrieving list of carousel items."""
     model_name = serializers.CharField()
     name = serializers.CharField()
-    toque_number = serializers.CharField()
+    toque_number = serializers.IntegerField()
-    public_mark = serializers.CharField()
+    public_mark = serializers.IntegerField()
     image = serializers.URLField(source='image_url')
     awards = AwardBaseSerializer(many=True)
     vintage_year = serializers.IntegerField()

apps/news/tests.py

@@ -35,11 +35,18 @@ class NewsTestCase(BaseTestCase):
         response = self.client.get("/api/web/news/")
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
-    def test_news_detail(self):
+    def test_news_web_detail(self):
-        response = self.client.get(f"/api/web/news/{self.test_news.slug}/")
+        response = self.client.get(f"/api/web/news/slug/{self.test_news.slug}/")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+    def test_news_back_detail(self):
+        response = self.client.get(f"/api/back/news/{self.test_news.id}/")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+    def test_news_list_back(self):
+        response = self.client.get("/api/back/news/")
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
     def test_news_type_list(self):
         response = self.client.get("/api/web/news/types/")
         self.assertEqual(response.status_code, status.HTTP_200_OK)
-

apps/news/urls/web.py

@@ -7,5 +7,5 @@ app_name = 'news'
 urlpatterns = [
     path('', views.NewsListView.as_view(), name='list'),
     path('types/', views.NewsTypeListView.as_view(), name='type'),
-    path('<slug:slug>/', views.NewsDetailView.as_view(), name='rud'),
+    path('slug/<slug:slug>/', views.NewsDetailView.as_view(), name='rud'),
 ]

apps/notification/tests.py

@@ -0,0 +1,116 @@
+from http.cookies import SimpleCookie
+
+from django.test import TestCase
+from rest_framework.test import APITestCase
+from rest_framework import status
+
+from account.models import User
+from notification.models import Subscriber
+
+
+class BaseTestCase(APITestCase):
+
+    def setUp(self):
+        self.username = 'sedragurda'
+        self.password = 'sedragurdaredips19'
+        self.email = 'sedragurda@desoz.com'
+        self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
+        # get tokkens
+        tokkens = User.create_jwt_tokens(self.user)
+        self.client.cookies = SimpleCookie({'access_token': tokkens.get('access_token'),
+                                            'refresh_token': tokkens.get('refresh_token')})
+
+
+class NotificationAnonSubscribeTestCase(APITestCase):
+
+    def test_subscribe(self):
+
+        test_data = {
+            "email": "test@email.com",
+            "state": 1
+        }
+
+        response = self.client.post("/api/web/notifications/subscribe/", data=test_data, format="json")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.json()["email"], test_data["email"])
+        self.assertEqual(response.json()["state"], test_data["state"])
+
+
+class NotificationSubscribeTestCase(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
+        self.test_data = {
+            "email": self.email,
+            "state": 1
+        }
+
+    def test_subscribe(self):
+
+        response = self.client.post("/api/web/notifications/subscribe/", data=self.test_data, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.json()["email"], self.email)
+        self.assertEqual(response.json()["state"], self.test_data["state"])
+
+    def test_subscribe_info_auth_user(self):
+
+        Subscriber.objects.create(user=self.user, email=self.email, state=1)
+
+        response = self.client.get("/api/web/notifications/subscribe-info/", data=self.test_data, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class NotificationSubscribeInfoTestCase(APITestCase):
+
+    def test_subscribe_info(self):
+
+        self.username = 'sedragurda'
+        self.password = 'sedragurdaredips19'
+        self.email = 'sedragurda@desoz.com'
+        self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
+
+        test_subscriber = Subscriber.objects.create(user=self.user, email=self.email, state=1)
+
+        response = self.client.get(f"/api/web/notifications/subscribe-info/{test_subscriber.update_code}/")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class NotificationUnsubscribeAuthUserTestCase(BaseTestCase):
+
+    def test_unsubscribe_auth_user(self):
+
+        Subscriber.objects.create(user=self.user, email=self.email, state=1)
+
+        self.test_data = {
+            "email": self.email,
+            "state": 1
+        }
+
+        response = self.client.patch("/api/web/notifications/unsubscribe/", data=self.test_data, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+
+class NotificationUnsubscribeTestCase(APITestCase):
+
+    def test_unsubscribe(self):
+        self.username = 'sedragurda'
+        self.password = 'sedragurdaredips19'
+        self.email = 'sedragurda@desoz.com'
+        self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
+
+        self.test_data = {
+            "email": self.email,
+            "state": 1
+        }
+
+        test_subscriber = Subscriber.objects.create(user=self.user, email=self.email, state=1)
+
+        response = self.client.patch(f"/api/web/notifications/unsubscribe/{test_subscriber.update_code}/",
+                                     data=self.test_data, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)

apps/search_indexes/documents/establishment.py

@@ -84,6 +84,8 @@ class EstablishmentDocument(Document):
             'name',
             'toque_number',
             'price_level',
+            'preview_image_url',
+            'slug',
         )
 
     def get_queryset(self):

apps/search_indexes/serializers.py

@@ -63,6 +63,8 @@ class EstablishmentDocumentSerializer(DocumentSerializer):
             'collections',
             'establishment_type',
             'establishment_subtypes',
+            'preview_image_url',
+            'slug',
         )
 
     @staticmethod

apps/utils/!tests.py

@@ -8,12 +8,16 @@ from http.cookies import SimpleCookie
 from account.models import User
 from news.models import News, NewsType
 
+
 from django.test import TestCase
 from translation.models import Language
 from django.core import exceptions
 
 from .serializers import validate_tjson
 
+from establishment.models import Establishment, EstablishmentType, Employee
+
+
 class BaseTestCase(APITestCase):
 
     def setUp(self):
@@ -33,6 +37,12 @@ class BaseTestCase(APITestCase):
              'locale': "en"
              })
 
+
+class TranslateFieldTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
         self.news_type = NewsType.objects.create(name="Test news type")
 
         self.news_item = News.objects.create(
@@ -50,15 +60,9 @@ class BaseTestCase(APITestCase):
             news_type=self.news_type
         )
 
-
-class TranslateFieldModel(BaseTestCase):
-
     def test_model_field(self):
         self.assertIsNotNone(getattr(self.news_item, "title_translated", None))
 
-
-class TranslateFieldReview(BaseTestCase):
-
     def test_read_locale(self):
         response = self.client.get(f"/api/web/news/{self.news_item.id}/", format='json')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
@@ -69,6 +73,59 @@ class TranslateFieldReview(BaseTestCase):
         self.assertEqual(news_data['title_translated'], "Test news item")
 
 
+class BaseAttributeTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
+        self.establishment_type = EstablishmentType.objects.create(name="Test establishment type")
+        self.establishment = Establishment.objects.create(
+            name="Test establishment",
+            establishment_type_id=self.establishment_type.id,
+            is_publish=True
+        )
+
+    def test_base_attr_api(self):
+        data = {
+            'user': self.user.id,
+            'name': 'Test name'
+        }
+
+        response = self.client.post('/api/back/establishments/employees/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response_data = response.json()
+        self.assertIn("id", response_data)
+
+        employee = Employee.objects.get(id=response_data['id'])
+
+        self.assertEqual(self.user, employee.created_by)
+        self.assertEqual(self.user, employee.modified_by)
+
+        modify_user = User.objects.create_user(
+            username='sedragurda2',
+            password='sedragurdaredips192',
+            email='sedragurda2@desoz.com',
+        )
+
+        modify_tokkens = User.create_jwt_tokens(modify_user)
+        self.client.cookies = SimpleCookie(
+            {'access_token': modify_tokkens.get('access_token'),
+             'refresh_token': modify_tokkens.get('refresh_token'),
+             'locale': "en"
+             })
+
+        update_data = {
+            'name': 'Test new name'
+        }
+
+        response = self.client.patch('/api/back/establishments/employees/1/', data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        employee.refresh_from_db()
+        self.assertEqual(modify_user, employee.modified_by)
+        self.assertEqual(self.user, employee.created_by)
+
 class ValidJSONTest(TestCase):
 
     def test_valid_json(self):
@@ -99,4 +156,4 @@ class ValidJSONTest(TestCase):
             validate_tjson(data)
             self.assertTrue(True)
         except exceptions.ValidationError:
-            self.assert_(False, "Test json translated FAILED")
+            self.assert_(False, "Test json translated FAILED")

apps/utils/decorators.py

@@ -0,0 +1,21 @@
+
+def with_base_attributes(cls):
+
+    def validate(self, data):
+        user = None
+        request = self.context.get("request")
+
+        if request and hasattr(request, "user"):
+            user = request.user
+
+        if user is not None:
+            data.update({'modified_by': user})
+
+            if not self.instance:
+                data.update({'created_by': user})
+
+        return data
+
+    setattr(cls, "validate", validate)
+
+    return cls