zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-17 18:09:25 +02:00
Code Issues Projects Releases Packages Wiki Activity

Linting + fix tests

Browse source
This commit is contained in:
zandercymatics 2023-11-22 14:00:06 -07:00
parent 1d70c5058a
commit bcf5c6bc5b
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/api/views.py
View file

@ -91,25 +91,28 @@ def available(request, domain=""):
except Exception: except Exception:
return JsonResponse({"available": False, "message": DOMAIN_API_MESSAGES["error"]}) return JsonResponse({"available": False, "message": DOMAIN_API_MESSAGES["error"]})
@require_http_methods(["GET"]) @require_http_methods(["GET"])
@login_not_required @login_not_required
def get_current_full(request): def get_current_full(request):
# Open the CSV file # Open the CSV file
file_path = 'migrationdata/current-full.csv' file_path = "migrationdata/current-full.csv"
return serve_file(file_path) return serve_file(file_path)
@require_http_methods(["GET"]) @require_http_methods(["GET"])
@login_not_required @login_not_required
def get_current_federal(request): def get_current_federal(request):
# Open the CSV file # Open the CSV file
file_path = 'migrationdata/current-federal.csv' file_path = "migrationdata/current-federal.csv"
return serve_file(file_path) return serve_file(file_path)
def serve_file(file_path): def serve_file(file_path):
"""Downloads a file based on a given filepath. Returns a 404 if not found.""" """Downloads a file based on a given filepath. Returns a 404 if not found."""
if os.path.exists(file_path): if os.path.exists(file_path):
# Serve the CSV file # Serve the CSV file
response = FileResponse(open(file_path, 'rb')) response = FileResponse(open(file_path, "rb"))
return response return response
else: else:
return HttpResponse("File not found", status=404) return HttpResponse("File not found", status=404)

2
src/registrar/tests/test_url_auth.py
View file

@ -111,6 +111,8 @@ class TestURLAuth(TestCase):
"/openid/callback/login/", "/openid/callback/login/",
"/openid/callback/logout/", "/openid/callback/logout/",
"/api/v1/available/whitehouse.gov", "/api/v1/available/whitehouse.gov",
"/api/v1/get-report/current-federal",
"/api/v1/get-report/current-full"
] ]
def assertURLIsProtectedByAuth(self, url): def assertURLIsProtectedByAuth(self, url):