Add implementation for load_with_datetime in Python package.

python/Cargo.toml

@@ -21,6 +21,7 @@ env_logger = "0"
 reqwest = { version = "*", features = ["native-tls-vendored"] }
 serde_json = "1"
 arrow  = { version = "5" }
+chrono = "0"
 
 [dependencies.pyo3]
 version = "0.14"

python/deltalake/table.py

@@ -152,6 +152,14 @@ def load_version(self, version: int) -> None:
         """
         self._table.load_version(version)
 
+    def load_with_datetime(self, ds: str) -> None:
+        """
+        Time travel Delta table to latest version that's created at or before provided `ds` argument.
+
+        :param ds: the identifier of the datetime point of the DeltaTable to load
+        """
+        self._table.load_with_datetime(ds)
+
     def schema(self) -> Schema:
         """
         Get the current schema of the DeltaTable.

python/src/lib.rs

@@ -4,6 +4,7 @@ extern crate arrow;
 extern crate pyo3;
 
 use arrow::datatypes::Schema as ArrowSchema;
+use chrono::{DateTime, FixedOffset, Utc};
 use deltalake::partitions::PartitionFilter;
 use pyo3::create_exception;
 use pyo3::exceptions::PyException;
@@ -25,6 +26,10 @@ impl PyDeltaTableError {
     fn from_tokio(err: tokio::io::Error) -> pyo3::PyErr {
         PyDeltaTableError::new_err(err.to_string())
     }
+
+    fn from_chrono(err: chrono::ParseError) -> pyo3::PyErr {
+        PyDeltaTableError::new_err(err.to_string())
+    }
 }
 
 #[inline]
@@ -100,6 +105,16 @@ impl RawDeltaTable {
             .map_err(PyDeltaTableError::from_raw)
     }
 
+    pub fn load_with_datetime(&mut self, ds: &str) -> PyResult<()> {
+        let datetime = DateTime::<Utc>::from(
+            DateTime::<FixedOffset>::parse_from_rfc3339(ds)
+                .map_err(PyDeltaTableError::from_chrono)?,
+        );
+        rt()?
+            .block_on(self._table.load_with_datetime(datetime))
+            .map_err(PyDeltaTableError::from_raw)
+    }
+
     pub fn files_by_partitions(
         &self,
         partitions_filters: Vec<(&str, &str, PartitionFilterValue)>,

python/tests/test_table_read.py

@@ -18,6 +18,17 @@ def test_read_simple_table_by_version_to_dict():
     assert dt.to_pyarrow_dataset().to_table().to_pydict() == {"value": [1, 2, 3]}
 
 
+def test_load_with_datetime():
+    table_path = "../rust/tests/data/simple_table"
+    dt = DeltaTable(table_path)
+    dt.load_with_datetime("2020-05-01T00:47:31-07:00")
+    assert dt.version() == 0
+    dt.load_with_datetime("2020-05-02T22:47:31-07:00")
+    assert dt.version() == 1
+    dt.load_with_datetime("2020-05-25T22:47:31-07:00")
+    assert dt.version() == 4
+
+
 def test_read_simple_table_update_incremental():
     table_path = "../rust/tests/data/simple_table"
     dt = DeltaTable(table_path, version=0)