# 63 Refactoring

* refactors django User model to custom User model to provide further attributes and methods directly on the user model

konova/management/commands/setup.py

@@ -7,7 +7,8 @@ Created on: 15.12.20
 """
 from getpass import getpass
 
-from django.contrib.auth.models import User, Group
+from user.models import User
+from django.contrib.auth.models import Group
 from django.core.management import BaseCommand, call_command
 from django.db import transaction
 

konova/management/commands/setup_data.py

@@ -23,8 +23,7 @@ GROUPS_DATA = [
 
 # Must match with UserNotificationEnum
 USER_NOTIFICATIONS_NAMES = {
-    "NOTIFY_ON_NEW_RELATED_DATA": _("On new related data"),
-    "NOTIFY_ON_SHARE_LINK_DISABLED": _("On disabled share link"),
+    "NOTIFY_ON_SHARED_ACCESS_GAINED": _("On shared access gained"),
     "NOTIFY_ON_SHARED_ACCESS_REMOVED": _("On shared access removed"),
     "NOTIFY_ON_SHARED_DATA_RECORDED": _("On shared data recorded"),
     "NOTIFY_ON_SHARED_DATA_DELETED": _("On shared data deleted"),