def validateInput(self, args):
event = self.getInputParameters(args)
ips = {}
if (not 'documents' in event):
raise Exception(
"Document or path to a foler or S3 bucket containing documents is required."
)
inputDocument = event['documents']
idl = inputDocument.lower()
bucketName = None
documents = []
awsRegion = 'us-east-1'
if (idl.startswith("s3://")):
o = urlparse(inputDocument)
bucketName = o.netloc
path = o.path[1:]
ar = S3Helper.getS3BucketRegion(bucketName)
if (ar):
awsRegion = ar
if (idl.endswith("/")):
allowedFileTypes = ["jpg", "jpeg", "png", "pdf"]
documents = S3Helper.getFileNames(awsRegion, bucketName,
path, 1,
allowedFileTypes)
else:
documents.append(path)
else:
if (idl.endswith("/")):
allowedFileTypes = ["jpg", "jpeg", "png"]
documents = FileHelper.getFileNames(
inputDocument, allowedFileTypes)
else:
documents.append(inputDocument)
if ('region' in event):
awsRegion = event['region']
ips["bucketName"] = bucketName
ips["documents"] = documents
ips["awsRegion"] = awsRegion
ips["text"] = ('text' in event)
ips["forms"] = ('forms' in event)
ips["tables"] = ('tables' in event)
ips["insights"] = ('insights' in event)
ips["medical-insights"] = ('medical-insights' in event)
if ("translate" in event):
ips["translate"] = event["translate"]
else:
ips["translate"] = ""
return ips
def test_get_s3_bucket_region(self):
bucketRegion = S3Helper.getS3BucketRegion(BUCKET_NAME)
if (REGION == 'us-east-1'):
self.assertEqual(bucketRegion, None)
else:
self.assertEqual(bucketRegion, REGION)
