catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add glue.batch_get_partition endpoint

Browse source
This commit is contained in:
Don Kuntz 2019-07-17 14:07:19 -05:00
commit c8d8aa4dd0
No known key found for this signature in database
GPG key ID: 4599624BD234F373
2 changed files with 74 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

17
moto/glue/responses.py
View file

@ -141,6 +141,23 @@ class GlueResponse(BaseResponse):
return json.dumps({'Partition': p.as_dict()})
def batch_get_partition(self):
database_name = self.parameters.get('DatabaseName')
table_name = self.parameters.get('TableName')
partitions_to_get = self.parameters.get('PartitionsToGet')
table = self.glue_backend.get_table(database_name, table_name)
partitions = []
for values in partitions_to_get:
try:
p = table.get_partition(values=values["Values"])
partitions.append(p.as_dict())
except PartitionNotFoundException:
continue
return json.dumps({'Partitions': partitions})
def create_partition(self):
database_name = self.parameters.get('DatabaseName')
table_name = self.parameters.get('TableName')