ammar/zod-backend
1
0
Fork 0
mirror of https://github.com/HamzaSha1/zod-backend.git synced 2025-07-15 10:05:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

csv excel export api

Browse Source
This commit is contained in:
abutalib-kiwi
2023-08-16 17:15:12 +05:30
parent 4bc16c56bd
commit 043c8c2f63
4 changed files with 48 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

109
web_admin/views/analytics.py
View File

@ -44,7 +44,7 @@ class AnalyticsViewSet(GenericViewSet):
to get junior leaderboard and ranking
"""
serializer_class = None
# permission_classes = [IsAuthenticated, AdminPermission]
permission_classes = [IsAuthenticated, AdminPermission]
def get_queryset(self):
user_qs = USER.objects.filter(
@ -148,93 +148,14 @@ class AnalyticsViewSet(GenericViewSet):
serializer = self.serializer_class(paginated_queryset, many=True)
return custom_response(None, serializer.data)
# @action(methods=['get'], url_name='generate-report', url_path='generate-report', detail=False)
# def generate_report(self, request):
#
# response = HttpResponse(content_type='text/csv')
# response['Content-Disposition'] = 'attachment; filename="ZOD_Bank_Analytics.csv"'
#
# end_date = datetime.date.today()
# start_date = end_date - datetime.timedelta(days=6)
#
# if request.query_params.get('start_date') and request.query_params.get('end_date'):
# start_date = datetime.datetime.strptime(request.query_params.get('start_date'), DATE_FORMAT)
# end_date = datetime.datetime.strptime(request.query_params.get('end_date'), DATE_FORMAT)
#
# buffer = io.StringIO()
# csv_writer = csv.writer(buffer)
#
# # sheet 1 for Total Users
# user_qs = self.get_queryset()
# queryset = user_qs.filter(date_joined__range=(start_date, (end_date + datetime.timedelta(days=1))))
# serializer = UserCSVReportSerializer(queryset, many=True)
#
# # writer = csv.writer(response)
# csv_writer.writerow(['Users'])
# csv_writer.writerow([
# 'First Name', 'Last Name', 'Email', 'Phone Number', 'User Type', 'status', 'Date Joined',
# # Add more fields as needed
# ])
#
# for user_data in serializer.data:
# row = [
# user_data['first_name'], user_data['last_name'], user_data['email'],
# user_data['phone_number'], user_data['user_type'],
# user_data['is_active'], user_data['date_joined']
# ]
# csv_writer.writerow(row)
#
# response.write(buffer.getvalue())
# buffer.close()
#
# # sheet 2 for Assign Task
# assign_tasks = JuniorTask.objects.filter(
# created_at__range=[start_date, (end_date + datetime.timedelta(days=1))]
# ).exclude(task_status__in=[PENDING, EXPIRED])
# buffer = io.StringIO()
# csv_writer = csv.writer(buffer)
# csv_writer.writerow([])
# csv_writer.writerow(['Assign Tasks'])
# csv_writer.writerow([
# 'Task Name', 'Task Status'
# # Add more fields as needed
# ])
#
# for task in assign_tasks:
# row = [
# task.task_name, dict(TASK_STATUS).get(task.task_status).capitalize(),
# ]
# csv_writer.writerow(row)
# response.write(buffer.getvalue())
# buffer.close()
#
# # sheet 3 for Juniors Leaderboard and rank
# queryset = JuniorPoints.objects.prefetch_related('junior', 'junior__auth').annotate(rank=Window(
# expression=Rank(),
# order_by=[F('total_points').desc(), 'junior__created_at']
# )).order_by('-total_points', 'junior__created_at')
# buffer = io.StringIO()
# csv_writer = csv.writer(buffer)
# csv_writer.writerow([])
# csv_writer.writerow(['Top 15 Juniors'])
# csv_writer.writerow([
# 'Junior Name', 'Points', 'Rank'
# # Add more fields as needed
# ])
#
# for junior in queryset:
# row = [
# f"{junior.junior.auth.first_name}{junior.junior.auth.last_name}" if junior.junior.auth.last_name else junior.junior.auth.first_name,
# junior.total_points, junior.rank
# ]
# csv_writer.writerow(row)
# response.write(buffer.getvalue())
# buffer.close()
#
# return response
@action(methods=['get'], url_name='generate-report', url_path='generate-report', detail=False)
def generate_report(self, request):
@action(methods=['get'], url_name='export-excel', url_path='export-excel', detail=False)
def export_excel(self, request):
"""
to export users count, task details and top juniors in csv/excel file
:param request: start_date: date format (yyyy-mm-dd)
:param request: end_date: date format (yyyy-mm-dd)
:return:
"""
response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
response['Content-Disposition'] = 'attachment; filename="ZOD_Bank_Analytics.xlsx"'
@ -252,7 +173,7 @@ class AnalyticsViewSet(GenericViewSet):
workbook = xlsxwriter.Workbook(buffer)
# Add sheets
sheets = ['Users', 'Assign Tasks', 'Top 15 Juniors']
sheets = ['Users', 'Assign Tasks', 'Juniors leaderboard']
for sheet_name in sheets:
worksheet = workbook.add_worksheet(name=sheet_name)
@ -263,7 +184,13 @@ class AnalyticsViewSet(GenericViewSet):
queryset = user_qs.filter(date_joined__range=(start_date, (end_date + datetime.timedelta(days=1))))
serializer = UserCSVReportSerializer(queryset, many=True)
df_users = pd.DataFrame(serializer.data)
df_users = pd.DataFrame([
{'Name': f"{user['first_name']} {user['last_name']}",
'Email': user['email'], 'Phone Number': user['phone_number'],
'User Type': user['user_type'], 'Status': user['is_active'],
'Date Joined': user['date_joined']}
for user in serializer.data
])
for idx, col in enumerate(df_users.columns):
worksheet.write(0, idx, col) # Write header
for row_num, row in enumerate(df_users.values, start=1):
@ -288,7 +215,7 @@ class AnalyticsViewSet(GenericViewSet):
worksheet.write(row_num, col_num, value)
# sheet 3 for Juniors Leaderboard and rank
elif sheet_name == 'Top 15 Juniors':
elif sheet_name == 'Juniors leaderboard':
queryset = JuniorPoints.objects.prefetch_related('junior', 'junior__auth').annotate(rank=Window(
expression=Rank(),
order_by=[F('total_points').desc(), 'junior__created_at']