Add bytearray as supported type (fixes #35)

clevercsv/detect_type.py

@@ -141,6 +141,7 @@ def detect_type(self, cell, is_quoted=False):
  ("nan", self.is_nan),
  ("date", self.is_date),
  ("datetime", self.is_datetime),
+ ('bytearray', self.is_bytearray),
  ]
  for name, func in type_tests:
  if func(cell, is_quoted=is_quoted):
@@ -283,6 +284,11 @@ def is_unix_path(self, cell, **kwargs):
  cell = cell.strip(" ")
  return self._run_regex(cell, "unix_path")
 
+ def is_bytearray(self, cell: str, **kwargs) -> bool:
+ if self.strip_whitespace:
+ cell = cell.strip(" ")
+ return cell.startswith("bytearray(b") and cell.endswith(")")
+
 
 def gen_known_type(cells):
  """

tests/test_unit/test_detect_type.py

@@ -332,6 +332,27 @@ def test_unicode_alphanum(self):
  )
  )
 
+ def test_bytearray(self):
+ yes_bytearray = [
+ "bytearray(b'')",
+ "bytearray(b'abc,*&@\"')",
+ "bytearray(b'bytearray(b'')')",
+ ]
+ no_bytearray = [
+ "bytearray(b'abc",
+ "bytearray(b'abc'",
+ "bytearray('abc')",
+ "abc,bytearray(b'def')",
+ ]
+
+ for case in yes_bytearray:
+ with self.subTest(case=case):
+ self.assertTrue(self.td.is_bytearray(case))
+
+ for case in no_bytearray:
+ with self.subTest(case=case):
+ self.assertFalse(self.td.is_bytearray(case))
+
  # Unix path
 
  def test_unix_path(self):