Skip to content

Commit d57d2ee

Browse files
committed
On CSV sniffing error use extension for delimiter
1 parent 0374b82 commit d57d2ee

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

src/rich_cli/__main__.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -719,8 +719,15 @@ def render_csv(
719719
try:
720720
dialect = sniffer.sniff(csv_data[:1024], delimiters=",\t|;")
721721
has_header = sniffer.has_header(csv_data[:1024])
722-
except Exception as error:
723-
on_error(str(error))
722+
except csv.Error as error:
723+
if resource.endswith(".csv"):
724+
dialect = csv.get_dialect("excel")
725+
has_header = True
726+
elif resource.endswith(".tsv"):
727+
dialect = csv.get_dialect("excel-tab")
728+
has_header = True
729+
else:
730+
on_error(str(error))
724731

725732
csv_file = io.StringIO(csv_data)
726733
reader = csv.reader(csv_file, dialect=dialect)

0 commit comments

Comments
 (0)