Skip to content

Commit

Permalink
added hotdrop replacement for django native get_user_model instead of…
Browse files Browse the repository at this point in the history
… outdated User
  • Loading branch information
agriev authored and javrasya committed Feb 1, 2021
1 parent ec5dd2b commit 25ca879
Show file tree
Hide file tree
Showing 3 changed files with 17 additions and 3 deletions.
7 changes: 6 additions & 1 deletion demo/management/commands/bootstrap_river_admin_demo.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,11 @@
import random

from django.contrib.auth.models import Group, User, Permission
from django.contrib.auth.models import Group, Permission

#custom user support patch
from django.contrib.auth import get_user_model
User = get_user_model()

from django.contrib.contenttypes.models import ContentType
from django.core.management.base import BaseCommand
from django.db import transaction
Expand Down
7 changes: 6 additions & 1 deletion examples/issue_tracker_example/models.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
from django.contrib.auth.models import User

#Custom user model support replacement
# from django.contrib.auth.models import User
from django.contrib.auth import get_user_model
User = get_user_model()

from django.db import models
from django.db.models import SET_NULL
from river.models.fields.state import StateField
Expand Down
6 changes: 5 additions & 1 deletion river_admin/views/serializers.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,13 @@
from django.contrib.auth.models import Permission, Group, User
from django.contrib.auth.models import Permission, Group
from django.contrib.contenttypes.models import ContentType
from rest_framework import serializers
from river.models import Function, OnApprovedHook, State, TransitionApprovalMeta, \
OnTransitHook, TransitionMeta, Transition, Workflow, TransitionApproval, DONE, CANCELLED

# Custom Usermodel support patch
from django.contrib.auth import get_user_model
User = get_user_model()

from river.models.hook import AFTER, BEFORE


Expand Down

0 comments on commit 25ca879

Please sign in to comment.