Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

apps/dpnk: add register coordinator and user profile REST API endpoints #356

Draft
wants to merge 1 commit into
base: master
Choose a base branch
from
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
201 changes: 201 additions & 0 deletions apps/dpnk/rest.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,8 @@
import requests
from rest_framework import status

import copy

organization_types = [org_type[0] for org_type in Company.ORGANIZATION_TYPE]


Expand Down Expand Up @@ -1924,6 +1926,204 @@ def post(self, request):
)


class UserAttendanceDeserializer(serializers.ModelSerializer):
class Meta:
model = UserAttendance
fields = (
"terms",
"teamId",
"merchId",
)
extra_kwargs = {
"terms": {"source": "personal_data_opt_in"},
"teamId": {"source": "team"},
"merchId": {"source": "t_shirt_size"},
}


class UserDeserializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = User
fields = (
"email",
"firstName",
"lastName",
)
extra_kwargs = {
"firstName": {"source": "first_name"},
"lastName": {"source": "last_name"},
}


class UserProfileDeserializer(serializers.HyperlinkedModelSerializer):

user_attendance = UserAttendanceDeserializer(required=False)
user = UserDeserializer(required=False)

class Meta:
model = UserProfile
fields = ["user_attendance", "user", "nickname", "gender", "newsletter"]
extra_kwargs = {
"gender": {"source": "sex"},
}

def _get_or_create_user_attendance(self):
request = self.context["request"]
user_attendance = request.user_attendance
if not user_attendance:
user_attendance = get_or_create_userattendance(request, request.subdomain)

return user_attendance

def create(self, validated_data):
user_attendance = self._get_or_create_user_attendance()
return self.update(user_attendance.userprofile, validated_data)

def update(self, instance, validated_data):
user_attendance_data = validated_data.pop("user_attendance", {})
user_data = validated_data.pop("user", {})
self._handle_user_attendance(user_attendance_data)
self._handle_user(user_data)
validated_data["user"] = self._get_user()
instance = super().update(instance, validated_data)
return instance

def _handle_user_attendance(self, data):
user_attendance = self._get_or_create_user_attendance()
if "personal_data_opt_in" in data:
user_attendance.personal_data_opt_in = data["personal_data_opt_in"]
if "team" in data:
user_attendance.team = data["team"]
if "t_shirt_size" in data:
user_attendance.t_shirt_size = data["t_shirt_size"]
user_attendance.save()

def _get_user(self):
user = self.context["request"].user
return user

def _handle_user(self, data):
user = self._get_user()
print(data)
if "email" in data:
user.email = data["email"]
if "first_name" in data:
user.first_name = data["first_name"]
if "last_name" in data:
user.last_name = data["last_name"]
user.save()

def to_representation(self, instance):
representation = super().to_representation(instance)
user_attendance = self._get_or_create_user_attendance()

user_attendance_data = UserAttendanceDeserializer(user_attendance).data
user_data = representation.pop("user")

representation["email"] = user_data["email"]
representation["personalDetails"] = {
"firstName": user_data["firstName"],
"lastName": user_data["lastName"],
"nickname": representation.pop("nickname"),
"gender": representation.pop("gender"),
"newsletter": representation.pop("newsletter"),
"terms": user_attendance_data["terms"],
}
representation["teamId"] = user_attendance_data["teamId"]
representation["merchId"] = user_attendance_data["merchId"]

return representation

def to_internal_value(self, data):
data["user"] = {}
data["user_attendance"] = {}

if "personalDetails" in data:
personal_details = data.pop("personalDetails")
if "email" in personal_details:
data["user"]["email"] = personal_details["email"]
if "firstName" in personal_details:
data["user"]["firstName"] = personal_details["firstName"]
if "lastName" in personal_details:
data["user"]["lastName"] = personal_details["lastName"]
if "nickname" in personal_details:
data["nickname"] = personal_details["nickname"]
if "gender" in personal_details:
data["gender"] = personal_details["gender"]
if "newsletter" in personal_details:
data["newsletter"] = personal_details["newsletter"]
if "terms" in personal_details:
data["user_attendance"]["terms"] = personal_details["terms"]
if "teamId" in data:
team_id = data.pop("teamId")
data["user_attendance"]["teamId"] = team_id
if "merchId" in data:
merch_id = data.pop("merchId")
data["user_attendance"]["merchId"] = merch_id

return super().to_internal_value(data)


class UserSerializer(serpy.Serializer):
email = serpy.StrField()
firstName = serpy.StrField(attr="first_name", required=False)
lastName = serpy.StrField(attr="last_name", required=False)


class UserProfileSerializer(serpy.Serializer):
user = UserSerializer()
user_attendance = RequestSpecificField(
lambda _, req: UserAttendanceSerializer(req.user_attendance).data
)
nickname = serpy.StrField(required=False)
gender = serpy.StrField(attr="sex", required=False)
newsletter = serpy.StrField(required=False)

def to_value(self, instance):
representation = super().to_value(instance)

for item in representation:
user_original = item.pop("user")
user_attendance_original = item.pop("user_attendance")

user_data_email = user_original["email"]
user_data_firstName = user_original["firstName"]
user_data_lastName = user_original["lastName"]

item["email"] = user_data_email
item["personalDetails"] = {
"firstName": user_data_firstName,
"lastName": user_data_lastName,
"nickname": item.pop("nickname"),
"gender": item.pop("gender"),
"newsletter": item.pop("newsletter"),
"terms": user_attendance_original["terms"],
}
item["teamId"] = user_attendance_original["teamId"]
item["merchId"] = user_attendance_original["merchId"]

return representation


class UserAttendanceSerializer(serpy.Serializer):
terms = serpy.BoolField(attr="personal_data_opt_in", required=False)
teamId = serpy.IntField(attr="team_id", required=False)
merchId = serpy.IntField(attr="t_shirt_size", required=False)


class UserProfileSet(viewsets.ModelViewSet, UserAttendanceMixin):
def get_queryset(self):
return UserProfile.objects.filter(pk=self.ua().userprofile.pk)

permission_classes = [permissions.IsAuthenticated]

def get_serializer_class(self):
if self.action in ["retrieve", "list"]:
return UserProfileSerializer
else:
return UserProfileDeserializer


router = routers.DefaultRouter()
router.register(r"gpx", TripSet, basename="gpxfile")
router.register(r"trips", TripRangeSet, basename="trip")
Expand Down Expand Up @@ -1983,3 +2183,4 @@ def post(self, request):
DiscountCouponSet,
basename="discount-coupon-by-code",
)
router.register(r"my_profile", UserProfileSet, basename="myprofile")
129 changes: 125 additions & 4 deletions apps/dpnk/rest_registration.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,20 @@
from dj_rest_auth.registration.serializers import RegisterSerializer
from django.contrib.auth.models import User
from rest_framework import serializers
from django.db import transaction
from dj_rest_auth.registration.views import RegisterView

from .models import (
UserProfile,
CompanyAdmin,
Campaign,
UserAttendance,
)

import copy

class CustomRegisterSerializer(RegisterSerializer):

class BasicRegisterSerializer(RegisterSerializer):
username = None # Remove the username field

def save(self, request):
Expand All @@ -16,8 +28,117 @@ def save(self, request):
return user


from dj_rest_auth.registration.views import RegisterView
class RegisterCoordinatorSerializer(BasicRegisterSerializer):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
user_fields = UserSerializer().fields
user_profile_fields = UserProfileSerializer().fields
company_admin_fields = CompanyAdminSerializer().fields
user_attendance_fields = UserAttendanceSerializer().fields

for field_name, field in user_fields.items():
self.fields[field_name] = copy.deepcopy(field)

for field_name, field in user_profile_fields.items():
self.fields[field_name] = copy.deepcopy(field)

for field_name, field in company_admin_fields.items():
self.fields[field_name] = copy.deepcopy(field)

for field_name, field in user_attendance_fields.items():
self.fields[field_name] = copy.deepcopy(field)

@transaction.atomic
def save(self, request):

user = super().save(request)
user.first_name = self.validated_data.get("first_name", "")
user.last_name = self.validated_data.get("last_name", "")
user.save()

campaign = Campaign.objects.get(slug=request.subdomain)

userprofile = UserProfile(
user=user,
telephone=self.validated_data.get("telephone", ""),
newsletter=self.validated_data.get("newsletter"),
)
userprofile.save()

company_admin = CompanyAdmin(
userprofile=userprofile,
administrated_company=self.validated_data.get("administrated_company"),
motivation_company_admin=self.validated_data.get(
"motivation_company_admin"
),
will_pay_opt_in=self.validated_data.get("will_pay_opt_in"),
campaign=campaign,
)
company_admin.save()

userattendance = UserAttendance(
campaign=campaign,
userprofile=userprofile,
personal_data_opt_in=self.validated_data.get("personal_data_opt_in"),
payment_status="None",
)
userattendance.save()

return user


class UserSerializer(serializers.ModelSerializer):
class Meta:
model = User
fields = (
"firstName",
"lastName",
)
extra_kwargs = {
"firstName": {"source": "first_name"},
"lastName": {"source": "last_name"},
}


class UserProfileSerializer(serializers.ModelSerializer):
class Meta:
model = UserProfile
fields = (
"phone",
"newsletter",
)
extra_kwargs = {
"phone": {"source": "telephone"},
}


class CompanyAdminSerializer(serializers.ModelSerializer):
class Meta:
model = CompanyAdmin
fields = (
"organizationId",
"jobTitle",
"responsibility",
)
extra_kwargs = {
"organizationId": {"source": "administrated_company"},
"jobTitle": {"source": "motivation_company_admin"},
"responsibility": {"source": "will_pay_opt_in"},
}


class UserAttendanceSerializer(serializers.ModelSerializer):
class Meta:
model = UserAttendance
fields = ("terms",)
extra_kwargs = {
"terms": {"source": "personal_data_opt_in"},
}


class BasicRegisterView(RegisterView):
serializer_class = BasicRegisterSerializer


class CustomRegisterView(RegisterView):
serializer_class = CustomRegisterSerializer
class RegisterCoordinatorView(RegisterView):
serializer_class = RegisterCoordinatorSerializer
Loading