| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- import boto3
- import json
- import config as cf
- class S3Service:
- """
- This class is intended to have helper methods for operations in s3.
- ...
- Methods
- --------
- validFileChecker(file_name)
- Checks the extension of the file and returns if the file is valid or not.
- """
- def vald_file_checker(self, file_name):
- """Returns "valid" if the file extension is of csv, hl7, json else returns invalid.
- Parameters
- ----------
- file_name : str
- File_name in string format
- Returns
- --------
- string
- Returns "valid" or "invalid" based on the extensionw
- """
- if("." in file_name):
- file_name_split = file_name.split(".")
- extension = file_name_split[1]
- check = ["csv","hl7","json"]
- if(extension not in check):
- print(cf.respomseMessage['fileFormat'])
- return "invalid"
- else:
- #if the file-name doesn't have any extension return invalid
- return "invalid"
- return extension
- def upload_file_tos3(self,bucket_name,file_name,file_content):
- """Methof to upload file to s3.
- Parameters
- ----------
- bucket_name : string
- Name of the bucket, where the file need to be uploaded.
- file_name : string
- file which need to be uploaded to bucket
- file_content: json
- The content which is being uploaded to s3 with it's name as file_name
- """
- s3 = boto3.client("s3")
- s3_file_path = cf.constants['s3_folder_path']
- #s3.put_object is the method provided by boto3 to upload file.
- response_upload_object = s3.put_object(Bucket=bucket_name, Key=s3_file_path+file_name,Body=file_content)
- print("The response upload object is",response_upload_object)
-
- # #print("The extension of the file is ",magic.from_file(fileContent, mime = True))
- return response_upload_object
-
-
|
