Adds keep_time keyword argument to next() and previous()

Author: sdispater

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 [Unreleased]
 
+### Added
+
+- Added the `keep_time` keyword argument to `next()`/`previous()` methods to keep time information.
+
 ### Changed
 
 - Improved `diff_for_humans()` method to display more intuitive strings on edge cases.

pendulum/pendulum.py

@@ -1421,7 +1421,7 @@ def _end_of_week(self):
 
         return dt.end_of('day')
 
-    def next(self, day_of_week=None):
+    def next(self, day_of_week=None, keep_time=False):
         """
         Modify to the next occurrence of a given day of the week.
         If no day_of_week is provided, modify to the next occurrence
@@ -1431,18 +1431,26 @@ def next(self, day_of_week=None):
         :param day_of_week: The next day of week to reset to.
         :type day_of_week: int or None
 
+        :param keep_time: Whether to keep the time information or not.
+        :type keep_time: bool
+
         :rtype: Pendulum
         """
         if day_of_week is None:
             day_of_week = self.day_of_week
 
-        dt = self.start_of('day').add(days=1)
+        if keep_time:
+            dt = self
+        else:
+            dt = self.start_of('day')
+
+        dt = dt.add(days=1)
         while dt.day_of_week != day_of_week:
             dt = dt.add(days=1)
 
         return dt
 
-    def previous(self, day_of_week=None):
+    def previous(self, day_of_week=None, keep_time=False):
         """
         Modify to the previous occurrence of a given day of the week.
         If no day_of_week is provided, modify to the previous occurrence
@@ -1452,12 +1460,20 @@ def previous(self, day_of_week=None):
         :param day_of_week: The previous day of week to reset to.
         :type day_of_week: int or None
 
+        :param keep_time: Whether to keep the time information or not.
+        :type keep_time: bool
+
         :rtype: Pendulum
         """
         if day_of_week is None:
             day_of_week = self.day_of_week
 
-        dt = self.start_of('day').subtract(days=1)
+        if keep_time:
+            dt = self
+        else:
+            dt = self.start_of('day')
+
+        dt = dt.subtract(days=1)
         while dt.day_of_week != day_of_week:
             dt = dt.subtract(days=1)
 

tests/pendulum_tests/test_day_of_week_modifiers.py

@@ -76,6 +76,13 @@ def test_next_saturday(self):
         d = Pendulum.create(1975, 5, 21).next(6)
         self.assertPendulum(d, 1975, 5, 24, 0, 0, 0)
 
+    def test_next_keep_time(self):
+        d = Pendulum.create(1975, 5, 21, 12).next()
+        self.assertPendulum(d, 1975, 5, 28, 0, 0, 0)
+
+        d = Pendulum.create(1975, 5, 21, 12).next(keep_time=True)
+        self.assertPendulum(d, 1975, 5, 28, 12, 0, 0)
+
     def test_previous(self):
         d = Pendulum.create(1975, 5, 21).previous()
         self.assertPendulum(d, 1975, 5, 14, 0, 0, 0)
@@ -88,6 +95,13 @@ def test_previous_saturday(self):
         d = Pendulum.create(1975, 5, 21).previous(6)
         self.assertPendulum(d, 1975, 5, 17, 0, 0, 0)
 
+    def test_previous_keep_time(self):
+        d = Pendulum.create(1975, 5, 21, 12).previous()
+        self.assertPendulum(d, 1975, 5, 14, 0, 0, 0)
+
+        d = Pendulum.create(1975, 5, 21, 12).previous(keep_time=True)
+        self.assertPendulum(d, 1975, 5, 14, 12, 0, 0)
+
     def test_first_day_of_month(self):
         d = Pendulum.create(1975, 11, 21).first_of('month', )
         self.assertPendulum(d, 1975, 11, 1, 0, 0, 0)