darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-11 09:28:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

strip commas from fields

Browse Source
This commit is contained in:
Kenneth Kehl
2024-08-08 13:02:51 -07:00
parent f3cab5a6cc
commit ee1dcdeb03
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/main/views/platform_admin.py
View File

@@ -77,24 +77,28 @@ def platform_admin():
def download_all_users():
# Create a CSV string from the user data
csv_data = "Name\tEmail Address\tMobile Number\tService\n"
users = user_api_client.get_all_users_detailed()
if len(users) == 0:
return "No data to download."
output = StringIO()
header = ["Name", "Email Address", "Phone Number", "Service"]
fieldnames = ["name", "email_address", "mobile_number", "service"]
writer = csv.DictWriter(
output,
fieldnames=["name", "email_address", "mobile_number", "service"],
fieldnames=fieldnames,
delimiter="\t",
)
writer.writeheader()
# Write custom header
writer.writerow(dict(zip(fieldnames, header)))
for user in users:
if user["name"].startswith("e2e"):
user_no_commas = {key: value.replace(",", "") for key, value in user.items()}
if user_no_commas["name"].startswith("e2e"):
continue
writer.writerow(user)
writer.writerow(user_no_commas)
csv_data = output.getvalue()
# Create a direct download response with the CSV data and appropriate headers
response = Response(csv_data, content_type="text/csv")
response.headers["Content-Disposition"] = "attachment; filename=users.csv"