country name for csv users

This commit is contained in:
alex 2020-01-30 14:07:28 +03:00
parent ab8a4845d8
commit d6b8c6191d
3 changed files with 9 additions and 3 deletions

View File

@ -475,6 +475,12 @@ class User(AbstractUser):
self.roles.set(Role.objects.filter(id__in=ids))
return self
@property
def country_name(self):
if self.last_country:
return self.last_country.country.name_translated
return None
class UserRoleQueryset(models.QuerySet):
"""QuerySet for model UserRole."""

View File

@ -154,6 +154,7 @@ class UserRoleSerializer(serializers.ModelSerializer):
class UserCSVSerializer(serializers.ModelSerializer):
created_at = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True, source='date_joined')
last_seen_at = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", read_only=True, source='last_login')
country = serializers.CharField(allow_blank=True, allow_null=True, source='country_name')
class Meta:
model = models.User
@ -162,7 +163,8 @@ class UserCSVSerializer(serializers.ModelSerializer):
'username',
'email',
'locale',
'last_country',
'country',
# 'last_country',
'created_at',
'last_seen_at',
'last_ip',

View File

@ -181,8 +181,6 @@ class UserCSVViewSet(ModelViewSet):
))
for item in serializer.data:
print(item)
writer.writerow(item.values())
return response
# return Response(serializer.data)