101 lines
3.7 KiB
Python
101 lines
3.7 KiB
Python
|
|
|||
|
|
from datetime import datetime
|
|||
|
|
from io import BytesIO
|
|||
|
|
|
|||
|
|
from django.db.models import Q
|
|||
|
|
from django.http import HttpResponse
|
|||
|
|
from openpyxl import Workbook
|
|||
|
|
from rest_framework import viewsets
|
|||
|
|
from rest_framework.decorators import action
|
|||
|
|
|
|||
|
|
from apps.authorization.api.v1.serializers import UserRelationSerializer
|
|||
|
|
from apps.authorization.models import UserRelations
|
|||
|
|
from apps.product import models as product_models
|
|||
|
|
from apps.product.web.api.v1.serializers import quota_distribution_serializers as distribution_serializers
|
|||
|
|
from apps.product.web.api.v1.serializers.product_serializers import IncentivePlanSerializer
|
|||
|
|
from common.helper_excel import create_header, excel_description, create_header_freez, create_value, shamsi_date, \
|
|||
|
|
convert_str_to_date
|
|||
|
|
from common.helpers import get_organization_by_user
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AuthExcelViewSet(viewsets.ModelViewSet):
|
|||
|
|
queryset = UserRelations.objects.all()
|
|||
|
|
serializer_class = UserRelationSerializer
|
|||
|
|
|
|||
|
|
# noqa # کاربران
|
|||
|
|
@action(
|
|||
|
|
methods=['get'],
|
|||
|
|
detail=False,
|
|||
|
|
url_path='user_relations_excel',
|
|||
|
|
url_name='user_relations_excel',
|
|||
|
|
name='user_relations_excel'
|
|||
|
|
)
|
|||
|
|
def user_relations_excel(self, request):
|
|||
|
|
output = BytesIO()
|
|||
|
|
workbook = Workbook()
|
|||
|
|
worksheet = workbook.active
|
|||
|
|
worksheet.sheet_view.rightToLeft = True
|
|||
|
|
worksheet.insert_rows(1)
|
|||
|
|
|
|||
|
|
queryset = self.filter_queryset(self.get_queryset().order_by('-create_date')) # noqa
|
|||
|
|
|
|||
|
|
ser_data = self.get_serializer(queryset, many=True).data
|
|||
|
|
|
|||
|
|
excel_options = [
|
|||
|
|
"ردیف",
|
|||
|
|
"نام کاربری",
|
|||
|
|
"نام",
|
|||
|
|
"نام خانوادگی",
|
|||
|
|
"سازمان",
|
|||
|
|
"نقش",
|
|||
|
|
"موبایل",
|
|||
|
|
"کد ملی",
|
|||
|
|
"استان/شهر",
|
|||
|
|
"آدرس",
|
|||
|
|
"وضعیت",
|
|||
|
|
"دسترسی ها"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
excel_description(worksheet, 'B1', f'کاربران', row2='C3')
|
|||
|
|
|
|||
|
|
create_header_freez(worksheet, excel_options, 1, 6, 7, height=22, width=20)
|
|||
|
|
l = 6
|
|||
|
|
m = 1
|
|||
|
|
if ser_data:
|
|||
|
|
for data in ser_data:
|
|||
|
|
is_active = 'فعال' if (data.get('user') or {}).get('is_active', False) else 'غیرفعال'
|
|||
|
|
|
|||
|
|
city = str((data.get('user') or {}).get('city_name', '')) or '-'
|
|||
|
|
province = str((data.get('user') or {}).get('province_name', '')) or '-'
|
|||
|
|
city_province = f"{city}/{province}" if city != '-' or province != '-' else '-'
|
|||
|
|
|
|||
|
|
list1 = [
|
|||
|
|
m,
|
|||
|
|
(data.get('user') or {}).get('username', '') or '-',
|
|||
|
|
(data.get('user') or {}).get('first_name', '') or '-',
|
|||
|
|
(data.get('user') or {}).get('last_name', '') or '-',
|
|||
|
|
(data.get('organization') or {}).get('name', '') or '-',
|
|||
|
|
(data.get('role') or {}).get('role_name', '') or '-',
|
|||
|
|
str((data.get('user') or {}).get('mobile', '')) or '-',
|
|||
|
|
str((data.get('user') or {}).get('national_code', '')) or '-',
|
|||
|
|
city_province,
|
|||
|
|
str((data.get('user') or {}).get('address', '')) or '-',
|
|||
|
|
is_active,
|
|||
|
|
]
|
|||
|
|
create_value(worksheet, list1, l + 1, 1, m=m)
|
|||
|
|
|
|||
|
|
m += 1
|
|||
|
|
l += 1
|
|||
|
|
|
|||
|
|
workbook.save(output)
|
|||
|
|
output.seek(0)
|
|||
|
|
|
|||
|
|
response = HttpResponse(
|
|||
|
|
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet') # noqa
|
|||
|
|
response[
|
|||
|
|
'Content-Disposition'] = f'attachment; filename="کاربران.xlsx"'.encode( # noqa
|
|||
|
|
'utf-8')
|
|||
|
|
response.write(output.getvalue())
|
|||
|
|
return response
|
|||
|
|
|