Merge pull request #435 from fitnr/master

fixed AttributeError with -n argument for utils with -H suppressed

csvkit/cli.py

@@ -267,7 +267,7 @@ def print_column_names(self):
         """
         Pretty-prints the names and indices of all columns to a file-like object (usually sys.stdout).
         """
-        if self.args.no_header_row:
+        if getattr(self.args, 'no_header_row', None):
             raise RequiredHeaderError('You cannot use --no-header-row with the -n or --names options.')
 
         f = self.input_file

tests/test_utilities/test_csvgrep.py

@@ -82,3 +82,16 @@ def test_invalid_column(self):
         utility = CSVGrep(args, output_file)
 
         self.assertRaises(ColumnIdentifierError, utility.main)
+
+    def test_display_column_names(self):
+        args = ['-n', 'examples/realdata/FY09_EDU_Recipients_by_State.csv']
+        output_file = six.StringIO()
+
+        utility = CSVGrep(args, output_file)
+        utility.main()
+
+        input_file = six.StringIO(output_file.getvalue())
+        reader = CSVKitReader(input_file)
+
+        self.assertEqual(next(reader), ['  1: State Name'])
+        self.assertEqual(next(reader), ['  2: State Abbreviate'])