279 lines
9.4 KiB
Python
279 lines
9.4 KiB
Python
import typing
|
|
|
|
from django.contrib.auth.hashers import make_password
|
|
from django.db.models import Q
|
|
from rest_framework import serializers
|
|
|
|
from apps.authentication.exceptions import UserExistException, OrganizationNationalUniqueIDException
|
|
from apps.authentication.models import (
|
|
User,
|
|
City,
|
|
Province,
|
|
Organization,
|
|
OrganizationType,
|
|
OrganizationStats,
|
|
BankAccountInformation
|
|
)
|
|
from apps.authorization import models as authorize_models
|
|
from apps.authorization.api.v1.serializers import (
|
|
UserRelationSerializer
|
|
)
|
|
|
|
|
|
class CitySerializer(serializers.ModelSerializer):
|
|
""" Serialize city data """
|
|
|
|
class Meta:
|
|
model = City
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
'province'
|
|
]
|
|
|
|
|
|
class ProvinceSerializer(serializers.ModelSerializer):
|
|
""" Serialize province data """
|
|
|
|
class Meta:
|
|
model = Province
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
]
|
|
|
|
|
|
class BankAccountSerializer(serializers.ModelSerializer):
|
|
""" Serialize bank account data """
|
|
|
|
class Meta:
|
|
model = BankAccountInformation
|
|
fields = '__all__'
|
|
|
|
def update(self, instance, validated_data):
|
|
""" update user bank account information """
|
|
instance.name = validated_data.get('name', instance.name)
|
|
instance.account = validated_data.get('account', instance.account)
|
|
instance.organization = validated_data.get('organization', instance.organization)
|
|
instance.account_type = validated_data.get('account_type', instance.account_type)
|
|
instance.card = validated_data.get('card', instance.card)
|
|
instance.sheba = validated_data.get('sheba', instance.sheba)
|
|
instance.save()
|
|
return instance
|
|
|
|
|
|
class UserSerializer(serializers.ModelSerializer):
|
|
""" Serialize user data """
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = [
|
|
'id',
|
|
'username',
|
|
'password',
|
|
'first_name',
|
|
'last_name',
|
|
'is_active',
|
|
'mobile',
|
|
'phone',
|
|
'national_code',
|
|
'birthdate',
|
|
'nationality',
|
|
'ownership',
|
|
'address',
|
|
'unit_name',
|
|
'unit_national_id',
|
|
'photo',
|
|
'province',
|
|
'city',
|
|
'otp_status',
|
|
'is_herd_owner',
|
|
]
|
|
extra_kwargs = {
|
|
'password': {
|
|
'required': False
|
|
},
|
|
'username': {
|
|
'required': False
|
|
},
|
|
"national_code": {
|
|
'required': False
|
|
},
|
|
"unit_name": {
|
|
'required': False
|
|
},
|
|
"unit_national_id": {
|
|
'required': False
|
|
}
|
|
}
|
|
|
|
def validate(self, attrs):
|
|
mobile = attrs['mobile']
|
|
|
|
if not self.instance:
|
|
if self.Meta.model.objects.filter(
|
|
Q(mobile=mobile) | Q(username=attrs['username'])
|
|
).exists():
|
|
raise UserExistException()
|
|
|
|
elif self.instance:
|
|
if self.instance.mobile != mobile:
|
|
if self.Meta.model.objects.filter(mobile=mobile).exists():
|
|
raise UserExistException()
|
|
|
|
if 'username' in attrs.keys():
|
|
if self.instance.username != attrs['username']:
|
|
if self.Meta.model.objects.filter(username=attrs['username']).exists():
|
|
raise UserExistException()
|
|
|
|
return attrs
|
|
|
|
def to_representation(self, instance):
|
|
""" Custom output """
|
|
|
|
representation = super().to_representation(instance)
|
|
if isinstance(instance, User):
|
|
if instance.bank_information.filter().exists():
|
|
representation['bank_account'] = BankAccountSerializer(
|
|
instance.bank_information.all(), many=True
|
|
).data
|
|
|
|
if instance.city:
|
|
representation['city_name'] = instance.city.name
|
|
if instance.province:
|
|
representation['province_name'] = instance.province.name
|
|
|
|
return representation
|
|
|
|
def update(self, instance, validated_data):
|
|
""" update user instance """
|
|
instance.username = validated_data.get('username', instance.username)
|
|
|
|
# control password
|
|
if validated_data.get('password'):
|
|
instance.password = make_password(validated_data.get('password', instance.password))
|
|
instance.first_name = validated_data.get('first_name')
|
|
instance.last_name = validated_data.get('last_name')
|
|
instance.is_active = validated_data.get('is_active')
|
|
instance.mobile = validated_data.get('mobile')
|
|
instance.phone = validated_data.get('phone')
|
|
instance.national_code = validated_data.get('national_code')
|
|
instance.birthdate = validated_data.get('birthdate')
|
|
instance.nationality = validated_data.get('nationality')
|
|
instance.ownership = validated_data.get('ownership')
|
|
instance.address = validated_data.get('address')
|
|
instance.unit_name = validated_data.get('unit_name')
|
|
instance.unit_national_id = validated_data.get('unit_national_id')
|
|
instance.photo = validated_data.get('photo')
|
|
instance.province = validated_data.get('province', instance.province)
|
|
instance.city = validated_data.get('city', instance.city)
|
|
instance.otp_status = validated_data.get('otp_status')
|
|
instance.save()
|
|
|
|
return instance
|
|
|
|
@staticmethod
|
|
def update_relations(user: object, relation_data: dict, bank_data: dict) -> typing.Any:
|
|
"""
|
|
update user relations & bank account for user
|
|
"""
|
|
user_relation = UserRelationSerializer(data=relation_data) # Create user relation
|
|
if user_relation.is_valid(raise_exception=True):
|
|
user_relation_obj = user_relation.update(
|
|
authorize_models.UserRelations.objects.get(user=user),
|
|
validated_data=relation_data
|
|
)
|
|
|
|
bank_info = BankAccountSerializer(data=bank_data) # Create user bank information
|
|
if bank_info.is_valid(raise_exception=True):
|
|
bank_obj = bank_info.update(
|
|
BankAccountInformation.objects.get(id=bank_data['id']),
|
|
validated_data=bank_data
|
|
)
|
|
|
|
return user_relation_obj, bank_obj # noqa
|
|
|
|
|
|
class OrganizationTypeSerializer(serializers.ModelSerializer):
|
|
""" Serialize organization type data """
|
|
|
|
class Meta:
|
|
model = OrganizationType
|
|
fields = [
|
|
'id',
|
|
'key',
|
|
'name',
|
|
'org_type_field',
|
|
'is_repeatable'
|
|
]
|
|
|
|
|
|
class OrganizationSerializer(serializers.ModelSerializer):
|
|
""" Serialize organization data """
|
|
|
|
class Meta:
|
|
model = Organization
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
'type',
|
|
'province',
|
|
'city',
|
|
'parent_organization',
|
|
'national_unique_id',
|
|
'company_code',
|
|
'field_of_activity'
|
|
]
|
|
extra_kwargs = {}
|
|
|
|
def validate(self, attrs):
|
|
"""
|
|
@ validate national_unique_code to be unique
|
|
@ validate to organization type field
|
|
"""
|
|
data = self.context.get('request')
|
|
national_unique_id = (data.get('organization'))['national_unique_id']
|
|
|
|
if not self.instance:
|
|
if self.Meta.model.objects.filter(national_unique_id=national_unique_id).exists():
|
|
raise OrganizationNationalUniqueIDException()
|
|
if self.instance:
|
|
if self.instance.national_unique_id != national_unique_id:
|
|
if self.Meta.model.objects.filter(national_unique_id=national_unique_id).exists():
|
|
raise OrganizationNationalUniqueIDException()
|
|
|
|
def to_representation(self, instance):
|
|
""" Custom output """
|
|
representation = super().to_representation(instance)
|
|
if isinstance(instance, Organization):
|
|
representation['province'] = ProvinceSerializer(instance.province).data
|
|
representation['city'] = CitySerializer(instance.city).data
|
|
representation['type'] = OrganizationTypeSerializer(instance.type).data
|
|
if instance.parent_organization:
|
|
representation['parent_organization'] = OrganizationSerializer(
|
|
instance.parent_organization
|
|
).data
|
|
|
|
if instance.bank_information.exists():
|
|
representation['bank_account'] = BankAccountSerializer(
|
|
instance.bank_information.all(), many=True
|
|
).data
|
|
return representation
|
|
|
|
def update(self, instance, validated_data):
|
|
""" update user organization information """ # noqa
|
|
instance.name = validated_data.get('name', instance.name)
|
|
instance.type = validated_data.get('type', instance.type)
|
|
instance.province = validated_data.get('province', instance.province)
|
|
instance.city = validated_data.get('city', instance.city)
|
|
instance.parent_organization = validated_data.get('parent_organization', instance.parent_organization)
|
|
instance.national_unique_id = validated_data.get('national_unique_id', instance.national_unique_id)
|
|
instance.save()
|
|
return instance
|
|
|
|
|
|
class OrganizationStatsSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = OrganizationStats
|
|
fields = '__all__'
|