code style, name method

KiiCorp · Mar 23, 2016 · 5b663eb · 5b663eb
1 parent 3e8c127
commit 5b663eb
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/query_runner/dynamodb_sql.py b/query_runner/dynamodb_sql.py
@@ -74,7 +74,11 @@ def annotate_query(cls):
 
     @classmethod
     def type(cls):
-        return "ddb_sql"
+        return "dynamodb_sql"
+
+    @classmethod
+    def name(cls):
+        return "DynamoDB (with DQL)"
 
     def __init__(self, configuration):
         super(DynamoDBSQL, self).__init__(configuration)
@@ -99,11 +103,11 @@ def run_query(self, query):
             engine = FragmentEngine()
             connection = engine.connect(**self.configuration.to_dict())
 
-            resDict = engine.execute(query if str(query).endswith(';') else str(query)+';')
+            res_dict = engine.execute(query if str(query).endswith(';') else str(query)+';')
 
             columns = []
             rows = []
-            for item in resDict:
+            for item in res_dict:
 
                 if not columns:
                     for k, v in item.iteritems():