darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 18:01:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add support for CSV files

Browse Source
This commit is contained in:
Ali Zaidi
2020-05-06 12:21:16 +01:00
parent 9f6bfb1b4e
commit 642ab1ad1e
3 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/clients/document_download.py
View File

@@ -1,5 +1,3 @@
import base64
import requests
from flask import current_app
@@ -26,15 +24,16 @@ class DocumentDownloadClient:
def get_upload_url(self, service_id):
return "{}/services/{}/documents".format(self.api_host, service_id)
def upload_document(self, service_id, file_contents):
def upload_document(self, service_id, file_contents, is_csv=None):
try:
response = requests.post(
self.get_upload_url(service_id),
headers={
'Authorization': "Bearer {}".format(self.auth_token),
},
files={
'document': base64.b64decode(file_contents)
json={
'document': file_contents,
'is_csv': is_csv or False,
}
)