DynamoDB Test

1
$ cat dynamodbTest.py
2
import boto3
3

4
#dynamodb = boto3.resource('dynamodb')
5
# Hard coding strings as credentials, not recommended. Use configs or env variables AWS_ACCESS_KEY, AWS_SECRET_KEY
6
dynamodb = boto3.resource(
7
    'dynamodb',
8
    aws_access_key_id='KEY_ID_REMOVED',
9
    aws_secret_access_key='ACCESS_KEY_REMOVED',
10
    region_name = 'us-east-1'
11
)
12

13
def create_table(tableName):
14
  table = dynamodb.create_table(
15
    TableName=tableName,
16
    KeySchema=[
17
        {
18
            'AttributeName': 'username',
19
            'KeyType': 'HASH'
20
        },
21
        {
22
            'AttributeName': 'last_name',
23
            'KeyType': 'RANGE'
24
        }
25
    ],
26
    AttributeDefinitions=[
27
        {
28
            'AttributeName': 'username',
29
            'AttributeType': 'S'
30
        },
31
        {
32
            'AttributeName': 'last_name',
33
            'AttributeType': 'S'
34
        },
35
    ],
36
    ProvisionedThroughput={
37
        'ReadCapacityUnits': 1,
38
        'WriteCapacityUnits': 1
39
    }
40
  )
41

42
  table.meta.client.get_waiter('table_exists').wait(TableName=tableName)
43
  print 'Table item count: {}'.format(table.item_count)
44

45
def delete_table(tableName):
46
  table = dynamodb.Table(tableName)
47
  table.delete()
48

49
def put_item(tableName):
50
  table = dynamodb.Table(tableName)
51

52
  response = table.put_item(
53
   Item={
54
        'username': 'jdoe',
55
        'first_name': 'jane',
56
        'last_name': 'doe',
57
        'age': 20,
58
        'account_type': 'librarian',
59
    }
60
  )
61

62
  print response
63

64
def get_item(tableName):
65
  table = dynamodb.Table(tableName)
66

67
  response = table.get_item(
68
   Key={
69
        'username': 'jdoe',
70
        'last_name': 'doe'
71
    }
72
  )
73

74
  item = response['Item']
75
  name = item['first_name']
76

77
  print(item)
78
  print("Hello, {}" .format(name))
79

80
def update_item(tableName):
81
  table = dynamodb.Table(tableName)
82

83
  table.update_item(
84
    Key={
85
        'username': 'jdoe',
86
        'last_name': 'doe'
87
    },
88
    UpdateExpression='SET age = :val1',
89
    ExpressionAttributeValues={
90
        ':val1': 23
91
    }
92
  )
93

94
def delete_item(tableName):
95
  table = dynamodb.Table(tableName)
96

97
  table.delete_item(
98
    Key={
99
        'username': 'jdoe',
100
        'last_name': 'doe'
101
    }
102
  )
103

104
def batch_write(tableName):
105
  table = dynamodb.Table(tableName)
106

107
  with table.batch_writer() as batch:
108
    batch.put_item(
109
        Item={
110
            'account_type': 'end_user',
111
            'username': 'bbob',
112
            'first_name': 'billy',
113
            'last_name': 'bob',
114
            'age': 20,
115
            'address': {
116
                'road': '1 fake street',
117
                'city': 'Houston',
118
                'state': 'TX',
119
                'country': 'USA'
120
            }
121
        }
122
    )
123
    batch.put_item(
124
        Item={
125
            'account_type': 'librarian',
126
            'username': 'user1',
127
            'first_name': 'user1 first name',
128
            'last_name': 'user1 last name',
129
            'age': 20,
130
            'address': {
131
                'road': '10 fake street',
132
                'city': 'Dallas',
133
                'state': 'TX',
134
                'country': 'USA'
135
            }
136
        }
137
    )
138
    batch.put_item(
139
        Item={
140
            'account_type': 'end_user',
141
            'username': 'user2',
142
            'first_name': 'user2 first name',
143
            'last_name': 'user2 last name',
144
            'age': 23,
145
            'address': {
146
                'road': '12 fake street',
147
                'city': 'Austin',
148
                'province': 'TX',
149
                'state': 'USA'
150
            }
151
        }
152
    )
153

154
def create_multiple_items(tableName,itemCount):
155

156
  table = dynamodb.Table(tableName)
157

158
  with table.batch_writer() as batch:
159
    for i in range(itemCount):
160
        batch.put_item(
161
            Item={
162
                'account_type': 'anonymous',
163
                'username': 'user-' + str(i),
164
                'first_name': 'unknown',
165
                'last_name': 'unknown'
166
            }
167
        )
168

169

170
def query(tableName):
171
  from boto3.dynamodb.conditions import Key, Attr
172
  table = dynamodb.Table(tableName)
173

174
  response = table.query(
175
    KeyConditionExpression=Key('username').eq('user2')
176
  )
177

178
  items = response['Items']
179
  print(items)
180

181
def scan(tableName):
182
  from boto3.dynamodb.conditions import Key, Attr
183

184
  table = dynamodb.Table(tableName)
185

186
  response = table.scan(
187
    FilterExpression=Attr('age').gt(23)
188
  )
189

190
  items = response['Items']
191
  print(items)
192

193
  len(items)
194
  for x in range(len(items)):
195
    items[x]['username']
196

197
def query_filter(tableName):
198
  from boto3.dynamodb.conditions import Key, Attr
199

200
  table = dynamodb.Table(tableName)
201

202
  response = table.scan(
203
    FilterExpression=Attr('first_name').begins_with('r') & Attr('account_type').eq('librarian')
204
  )
205

206
  items = response['Items']
207
  print(items)
208

209

210
# Comment/uncomment below to play with the different functions
211
#create_table('staff')
212

213
#put_item('staff')
214
#get_item('staff')
215
#update_item('staff')
216
#delete_item('staff')
217

218
#batch_write('staff')
219

220
#create_multiple_items('staff', 100)
221

222
#query('staff')
223
#scan('staff')
224
#query_filter('staff')
225

226
#delete_table('staff')