-
Notifications
You must be signed in to change notification settings - Fork 0
/
emr_details.py
48 lines (39 loc) · 1.11 KB
/
emr_details.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from config import conf
class emr_details:
"""
Main class for boto3 client for emr
"""
def __init__(self,session_client):
self.session = session_client
self.connection = self.session.client('emr')
def get_instances(self):
"""
Fetching all instances
"""
response = self.connection.list_clusters()
return response['Clusters']
def get_tags(self,id):
"""
Fetching tags from each given cluster id
"""
response = self.connection.describe_cluster(ClusterId=id)
if 'Tags' in response['Cluster'].keys():
return response['Cluster']['Tags']
else:
return []
class emr_cluster_details:
"""
Emr details class
"""
def __init__(self,cluster):
self.cluster = cluster
self.id = self.cluster['Id']
self.name = self.cluster['Name']
self.tag = ''
def assign_tags(self,tag):
"""
Assign tags for cluster details
"""
for t in tag:
if conf['tag'] in t.values():
self.tag = t['Value']