add support for 'code' column in datatables (#87)

Author: sergeiQ

.gitignore

@@ -5,4 +5,6 @@
 /quandl.egg-info
 /build
 /dist
-/tmp/
+/tmp/
+.idea/*
+venv

CHANGELOG.md

@@ -1,3 +1,8 @@
+### 3.1.0 - 2017-03-07
+
+* Bug: support column with a name 'code'
+
+
 ### 3.0.1 - 2016-05-25
 
 * Handle unexpected errors

quandl/get_table.py

@@ -6,7 +6,7 @@
 import os
 
 
-def get_table(code, **options):
+def get_table(datatable_code, **options):
     if 'paginate' in options.keys():
         paginate = options.pop('paginate')
     else:
@@ -16,7 +16,7 @@ def get_table(code, **options):
     page_count = 0
     while True:
         next_options = copy.deepcopy(options)
-        next_data = Datatable(code).data(params=next_options)
+        next_data = Datatable(datatable_code).data(params=next_options)
 
         if data is None:
             data = next_data

quandl/version.py

@@ -1 +1 @@
-VERSION = '3.0.1'
+VERSION = '3.1.0'

test/test_connection.py

@@ -67,6 +67,6 @@ def test_build_request(self, mock):
                                  'accept': ('application/json, '
                                             'application/vnd.quandl+json;version=2015-04-09'),
                                  'request-source': 'python',
-                                 'request-source-version': '3.0.1'},
+                                 'request-source-version': '3.1.0'},
                         params={'per_page': 10, 'page': 2})
         self.assertEqual(mock.call_args, expected)

test/test_get_table.py

@@ -37,3 +37,8 @@ def tearDownClass(cls):
     def test_datatable_returns_datatable_object(self, mock):
         df = quandl.get_table('ZACKS/FC', params={})
         self.assertIsInstance(df, pandas.core.frame.DataFrame)
+
+    @patch('quandl.connection.Connection.request')
+    def test_datatable_with_code_returns_datatable_object(self, mock):
+        df = quandl.get_table('AR/MWCF', code="ICEP_WAC_Z2017_S")
+        self.assertIsInstance(df, pandas.core.frame.DataFrame)