zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-19 10:59:21 +02:00
Code Issues Projects Releases Packages Wiki Activity

Link bucket to generator

Browse source
This commit is contained in:
zandercymatics 2023-11-29 08:36:12 -07:00
parent f5974e00dc
commit e78a506d2b
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 26 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/api/views.py
View file

@ -9,6 +9,8 @@ from login_required import login_not_required
from cachetools.func import ttl_cache
from registrar.utility.s3_bucket import S3ClientHelper
DOMAIN_FILE_URL = "https://raw.githubusercontent.com/cisagov/dotgov-data/main/current-full.csv"
@ -94,21 +96,24 @@ def available(request, domain=""):
@require_http_methods(["GET"])
@login_not_required
def get_current_full(request, file_path="migrationdata/current-full.csv"):
return serve_file(file_path)
return serve_file(file_path, "current-full.csv")
@require_http_methods(["GET"])
@login_not_required
def get_current_federal(request, file_path="migrationdata/current-federal.csv"):
return serve_file(file_path)
return serve_file(file_path, "current-federal.csv")
def serve_file(file_path):
def serve_file(file_path, file_name):
"""Downloads a file based on a given filepath. Returns a 404 if not found."""
s3_client = S3ClientHelper()
# TODO - #1403, grab from the S3 instance instead
# TODO - check if file exists in s3, not here
if os.path.exists(file_path):
# Serve the CSV file
response = FileResponse(open(file_path, "rb"))
file = s3_client.get_file(file_name)
response = FileResponse(file)
return response
else:
return HttpResponse("File not found", status=404)