Added more supporting formats

Author: jprakash-db

src/databricks/sql/utils.py

@@ -629,7 +629,13 @@ def convert_decimals_in_arrow_table(table, description) -> "pyarrow.Table":
 
 
 def datetime_parser(datetime_string):
-    formats = ["%Y-%m-%d %H:%M:%S", "%Y-%m-%d"]
+    formats = [
+        "%Y-%m-%d %H:%M:%S.%f",  # MySQL and PostgreSQL format
+        "%Y-%m-%d %H:%M:%S",  # Common SQL format
+        "%Y-%m-%d",  # MySQL Date only
+        "%m/%d/%Y %I:%M:%S %p",  # US format MSSQL Server
+        "%d-%b-%Y %H:%M:%S",  # Oracle format
+    ]
 
     for fmt in formats:
         try:

tests/unit/test_util.py

@@ -20,6 +20,9 @@ def get_column_table_and_description(self):
             ("date_column", "date", None, None, None, None, None),
             ("timestamp_column", "timestamp", None, None, None, None, None),
             ("timestamp_ntz_column", "timestamp", None, None, None, None, None),
+            ("timestamp_column_2", "timestamp", None, None, None, None, None),
+            ("timestamp_column_3", "timestamp", None, None, None, None, None),
+            ("timestamp_column_4", "timestamp", None, None, None, None, None),
             ("binary_column", "binary", None, None, None, None, None),
             ("array_column", "array", None, None, None, None, None),
             ("map_column", "map", None, None, None, None, None),
@@ -41,6 +44,9 @@ def get_column_table_and_description(self):
             ("2023-12-31",),
             ("2023-12-31 12:30:00",),
             ("2023-12-31 12:30:00",),
+            ("2021-09-30 11:27:35.123",),
+            ("03/08/2024 02:30:15 PM",),
+            ("08-Mar-2024 14:30:15",),
             (b"\xde\xad\xbe\xef",),
             ('["item1","item2"]',),
             ('{"key1":"value1","key2":"value2"}',),
@@ -71,6 +77,9 @@ def test_convert_to_assigned_datatypes_in_column_table(self):
             (datetime.date(2023, 12, 31), datetime.date),
             (datetime.datetime(2023, 12, 31, 12, 30, 0), datetime.datetime),
             (datetime.datetime(2023, 12, 31, 12, 30, 0), datetime.datetime),
+            (datetime.datetime(2021, 9, 30, 11, 27, 35, 123000), datetime.datetime),
+            (datetime.datetime(2024, 3, 8, 14, 30, 15), datetime.datetime),
+            (datetime.datetime(2024, 3, 8, 14, 30, 15), datetime.datetime),
             (b"\xde\xad\xbe\xef", bytes),
             ('["item1","item2"]', str),
             ('{"key1":"value1","key2":"value2"}', str),