From 846dbe0c4eb161ab814b76a42fcc172cd0c68c0a Mon Sep 17 00:00:00 2001 From: Andrey Date: Thu, 29 Mar 2018 16:28:58 +0300 Subject: [PATCH] finance email --- access/models/other.py | 4 ++++ access/serializers.py | 9 +++++++-- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/access/models/other.py b/access/models/other.py index 54681d8..b7e4a1c 100644 --- a/access/models/other.py +++ b/access/models/other.py @@ -53,6 +53,10 @@ class Account(models.Model): def __str__(self): return self.owner.email + def get_phone(self): + return '' if self.phone.national_number else \ + (self.phone.national_number if self.phone.country_code is None else str(self.phone)) + class Meta: verbose_name = 'Дополнительная информация о пользователе' verbose_name_plural = 'Дополнительная информация о пользователе' diff --git a/access/serializers.py b/access/serializers.py index f4bb455..9923df3 100644 --- a/access/serializers.py +++ b/access/serializers.py @@ -8,6 +8,7 @@ from progress.serializers import SecureProgressSerializer, SupportProgressSerial class AccountSerializer(serializers.ModelSerializer): gender = serializers.SerializerMethodField() + phone = serializers.SerializerMethodField() class Meta: model = Account @@ -17,6 +18,10 @@ class AccountSerializer(serializers.ModelSerializer): def get_gender(self): return self.get_gender_display() + @staticmethod + def get_phone(self): + return self.get_phone() + class UserSelfSerializer(serializers.ModelSerializer): account = serializers.SerializerMethodField() @@ -84,7 +89,7 @@ class UserSearchSerializer(serializers.ModelSerializer): @staticmethod def get_phone(self): - return None if self.account.phone is None else self.account.phone.raw_input + return self.account.get_phone() @staticmethod def get_pay(self): @@ -114,7 +119,7 @@ class UserProgressSearchSerializer(serializers.ModelSerializer): @staticmethod def get_phone(self): - return None if self.account.phone is None else self.account.phone.raw_input + return self.account.get_phone() @staticmethod def get_progresses(self):