Civil Time Unit conversions compatibility between pybind11_abseil & Google-internal Clif_PyObjAs/From implementations.

This is to support the PyCLIF-pybind11 integration.

PiperOrigin-RevId: 525462353

Author: rwgk

pybind11_abseil/absl_casters.h

@@ -235,10 +235,10 @@ struct type_caster<absl::Time> {
   }
 };
 
-template <typename CivilTimeType>
-struct absl_civil_time_caster {
+template <typename CivilTimeUnitType>
+struct absl_civil_datetime_caster {
  public:
-  PYBIND11_TYPE_CASTER(CivilTimeType, const_name("CivilTimeType"));
+  PYBIND11_TYPE_CASTER(CivilTimeUnitType, const_name("CivilDateTime"));
 
   bool load(handle src, bool convert) {
     if (!convert || !hasattr(src, "year") || !hasattr(src, "month") ||
@@ -252,42 +252,68 @@ struct absl_civil_time_caster {
       minute = GetInt64Attr(src, "minute");
       second = GetInt64Attr(src, "second");
     }
-    value = CivilTimeType(GetInt64Attr(src, "year"), GetInt64Attr(src, "month"),
+    value =
+        CivilTimeUnitType(GetInt64Attr(src, "year"), GetInt64Attr(src, "month"),
                           GetInt64Attr(src, "day"), hour, minute, second);
     return true;
   }
 
-  static handle cast(const CivilTimeType& src, return_value_policy, handle) {
+  static handle cast(const CivilTimeUnitType& src, return_value_policy,
+                     handle) {
     auto py_datetime_t = module::import("datetime").attr("datetime");
     auto py_datetime = py_datetime_t(src.year(), src.month(), src.day(),
                                      src.hour(), src.minute(), src.second());
     return py_datetime.release();
   }
 };
 
+template <typename CivilTimeUnitType>
+struct absl_civil_date_caster {
+ public:
+  PYBIND11_TYPE_CASTER(CivilTimeUnitType, const_name("CivilDate"));
+
+  bool load(handle src, bool convert) {
+    if (!convert || !hasattr(src, "year") || !hasattr(src, "month") ||
+        !hasattr(src, "day")) {
+      return false;
+    }
+    value =
+        CivilTimeUnitType(GetInt64Attr(src, "year"), GetInt64Attr(src, "month"),
+                          GetInt64Attr(src, "day"));
+    return true;
+  }
+
+  static handle cast(const CivilTimeUnitType& src, return_value_policy,
+                     handle) {
+    auto py_datetime_t = module::import("datetime").attr("date");
+    auto py_datetime = py_datetime_t(src.year(), src.month(), src.day());
+    return py_datetime.release();
+  }
+};
+
 template <>
 struct type_caster<absl::CivilSecond>
-    : public absl_civil_time_caster<absl::CivilSecond> {};
+    : public absl_civil_datetime_caster<absl::CivilSecond> {};
 
 template <>
 struct type_caster<absl::CivilMinute>
-    : public absl_civil_time_caster<absl::CivilMinute> {};
+    : public absl_civil_datetime_caster<absl::CivilMinute> {};
 
 template <>
 struct type_caster<absl::CivilHour>
-    : public absl_civil_time_caster<absl::CivilHour> {};
+    : public absl_civil_datetime_caster<absl::CivilHour> {};
 
 template <>
 struct type_caster<absl::CivilDay>
-    : public absl_civil_time_caster<absl::CivilDay> {};
+    : public absl_civil_date_caster<absl::CivilDay> {};
 
 template <>
 struct type_caster<absl::CivilMonth>
-    : public absl_civil_time_caster<absl::CivilMonth> {};
+    : public absl_civil_date_caster<absl::CivilMonth> {};
 
 template <>
 struct type_caster<absl::CivilYear>
-    : public absl_civil_time_caster<absl::CivilYear> {};
+    : public absl_civil_date_caster<absl::CivilYear> {};
 
 // Returns {true, a span referencing the data contained by src} without copying
 // or converting the data if possible. Otherwise returns {false, an empty span}.

pybind11_abseil/tests/absl_test.py

@@ -194,8 +194,9 @@ def test_return_civilday(self):
     truncated = self.TEST_DATETIME.replace(
         hour=0, minute=0, second=0, microsecond=0)
     self.assertEqual(
-        truncated,
-        absl_example.make_civilday(self.TEST_DATETIME_UTC.timestamp()))
+        truncated.date(),
+        absl_example.make_civilday(self.TEST_DATETIME_UTC.timestamp()),
+    )
 
   def test_pass_datetime_as_civilday(self):
     truncated = self.TEST_DATETIME_UTC.replace(
@@ -207,8 +208,9 @@ def test_return_civilmonth(self):
     truncated = self.TEST_DATETIME.replace(
         day=1, hour=0, minute=0, second=0, microsecond=0)
     self.assertEqual(
-        truncated,
-        absl_example.make_civilmonth(self.TEST_DATETIME_UTC.timestamp()))
+        truncated.date(),
+        absl_example.make_civilmonth(self.TEST_DATETIME_UTC.timestamp()),
+    )
 
   def test_pass_datetime_as_civilmonth(self):
     truncated = self.TEST_DATETIME_UTC.replace(
@@ -221,8 +223,9 @@ def test_return_civilyear(self):
     truncated = self.TEST_DATETIME.replace(
         month=1, day=1, hour=0, minute=0, second=0, microsecond=0)
     self.assertEqual(
-        truncated,
-        absl_example.make_civilyear(self.TEST_DATETIME_UTC.timestamp()))
+        truncated.date(),
+        absl_example.make_civilyear(self.TEST_DATETIME_UTC.timestamp()),
+    )
 
   def test_pass_datetime_as_civilyear(self):
     truncated = self.TEST_DATETIME_UTC.replace(