From de228518ff1a0744fd86b7d970c1d6c3b17f9dc3 Mon Sep 17 00:00:00 2001
From: Willy Hoang <wh@knewton.com>
Date: Fri, 2 Oct 2015 12:35:27 -0400
Subject: [PATCH] Add support for retrieving dynamodb resources

Added a new namespace option to gather dynamodb resources (table names).
---
 plumbum.py                       |  7 +++++++
 sample_templates/dynamodb.yml.j2 | 22 ++++++++++++++++++++++
 test_plumbum.py                  |  6 ++++++
 3 files changed, 35 insertions(+)
 create mode 100644 sample_templates/dynamodb.yml.j2

diff --git a/plumbum.py b/plumbum.py
index 8fc445d..16d4835 100644
--- a/plumbum.py
+++ b/plumbum.py
@@ -170,6 +170,12 @@ def list_kinesis_applications(region, filter_by_kwargs):
         kinesis_streams[stream_name] = shard_ids
     return kinesis_streams
 
+def list_dynamodb(region, filter_by_kwargs):
+    """List all DynamoDB tables."""
+    conn = boto.dynamodb.connect_to_region(region)
+    tables = conn.list_tables()
+    return lookup(tables, filter_by=filter_by_kwargs)
+
 list_resources = {
     'ec2': list_ec2,
     'elb': list_elb,
@@ -178,6 +184,7 @@ def list_kinesis_applications(region, filter_by_kwargs):
     'asg': list_autoscaling_group,
     'sqs': list_sqs,
     'kinesisapp': list_kinesis_applications,
+    'dynamodb': list_dynamodb
 }
 
 
diff --git a/sample_templates/dynamodb.yml.j2 b/sample_templates/dynamodb.yml.j2
new file mode 100644
index 0000000..c14f966
--- /dev/null
+++ b/sample_templates/dynamodb.yml.j2
@@ -0,0 +1,22 @@
+{%- set metrics = {'ConsumedReadCapacityUnits': {'stat': ['Minimum', 'Maximum', 'Sum', 'Average'], 'unit': 'Count'},
+                   'ConsumedWriteCapacityUnits': {'stat': ['Minimum', 'Maximum', 'Sum', 'Average'], 'unit': 'Count'},
+                   } -%}
+
+# If connecting to a different region other than default, set region
+Auth:
+  region: "{{ region }}"
+Metrics:
+{%- for table_name in resources %}
+  {%- for metric in metrics %}
+- Namespace: "AWS/DynamoDB"
+  MetricName: "{{ metric }}"
+  Statistics: "{{ metrics[metric]['stat'] }}"
+  Unit: "{{ metrics[metric]['unit'] }}"
+  Dimensions:
+    TableName: "{{ table_name }}"
+  Options:
+    Formatter: 'cloudwatch.%(Namespace)s.{{ stream_name}}.%(MetricName)s.%(statistic)s.%(Unit)s'
+    Period: 1
+  {% endfor %}
+{% endfor %}
+
diff --git a/test_plumbum.py b/test_plumbum.py
index c3b97c1..6f45fcb 100644
--- a/test_plumbum.py
+++ b/test_plumbum.py
@@ -71,6 +71,12 @@ def test_filters_and_extras_with_region_specified(self):
         self.assertEqual(len(filter_by), 1)
         self.assertEqual(len(extras), 2)
 
+    def test_dynamodb_namespace(self):
+        argv = ['foo.j2', 'dynamodb', 'us-east-1']
+        templ, ns, region, filter_by, extras = plumbum.interpret_options(argv)
+        self.assertEqual(templ, 'foo.j2')
+        self.assertEqual(ns, 'dynamodb')
+        self.assertEqual(region, 'us-east-1')
 
 if __name__ == '__main__':
     unittest.main()