-
Notifications
You must be signed in to change notification settings - Fork 0
/
dynamodb_details.py
51 lines (40 loc) · 1.13 KB
/
dynamodb_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
49
50
51
from config import conf
class dynamodb_details:
"""
Main class for boto3 client for DynamoDB
"""
def __init__(self,session_client):
self.session = session_client
self.connection = self.session.client('dynamodb')
def get_instances(self):
"""
Getting all instance
"""
response = self.connection.list_tables()
return response['TableNames']
def get_arn(self,table):
"""
Getting table arn for given table
"""
res = self.connection.describe_table(TableName=table)
return res['Table']['TableArn']
def get_tags(self,arn):
"""
Fetching tags for table
"""
response = self.connection.list_tags_of_resource(ResourceArn=arn)
if 'Tags' in response.keys():
return response['Tags']
else:
return []
class table_details:
"""
Table details class
"""
def __init__(self,table):
self.table = table
def assign_tags(self,tag):
self.tag = ''
for t in tag:
if conf['tag'] in t.values():
self.tag = t['Value']