add index in report

fga-eps-mds · Dec 6, 2023 · f6af411 · f6af411
1 parent e4f23fc
commit f6af411
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/gestao/web/api/document/utils.py b/gestao/web/api/document/utils.py
@@ -5,6 +5,7 @@
 
 def generate_report_users_file(all_users_list: list, filter_list: list):
     template_data = {
+        "index": [],
         "fullName": [],
         "warName": [],
         "registration": [],
@@ -38,6 +39,7 @@ def generate_report_users_file(all_users_list: list, filter_list: list):
     }
 
     for user in all_users_list:
+        template_data["index"].append(all_users_list.index(user) + 1)
         template_data["fullName"].append(user.fullName)
         template_data["warName"].append(user.warName)
         template_data["registration"].append(user.registration)

diff --git a/gestao/web/api/document/views.py b/gestao/web/api/document/views.py
@@ -10,12 +10,16 @@
 
 
 @router.get("/report-users/")
-async def get_report_users(filter_list: List[str]):
+async def get_report_users(filter_list: List[str] = []):
     users_list = await User.objects.all()
     file_stream = generate_report_users_file(list(users_list), filter_list)
+    file_name = "report-users.xlsx"
     return StreamingResponse(
         file_stream,
         media_type=(
             "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
         ),
+        headers={
+            "Content-Disposition": f"attachment; filename={file_name}",
+        },
     )