* Fix code formatting

pub-solar · Dec 8, 2024 · 64c7eed · 64c7eed
1 parent a80b5c9
commit 64c7eed
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/watson/frames.py b/watson/frames.py
@@ -8,17 +8,25 @@
 
 
 hour_shift = 0
+
+
 def set_hour_shift(hour):
     global hour_shift
     hour_shift = hour
 
 
 def shifted_floor(time, timeframe):
-    return time.shift(hours=-hour_shift).floor(timeframe).shift(hours=hour_shift)
+    time = time.shift(hours=-hour_shift)
+    time = time.floor(timeframe)
+    time = time.shift(hours=hour_shift)
+    return time
 
 
 def shifted_ceil(time, timeframe):
-    return time.shift(hours=-hour_shift).ceil(timeframe).shift(hours=hour_shift)
+    time = time.shift(hours=-hour_shift)
+    time = time.ceil(timeframe)
+    time = time.shift(hours=hour_shift)
+    return time
 
 
 def shifted_from(from_):
@@ -83,7 +91,7 @@ def __gte__(self, other):
 
 
 class Span(object):
-    def __init__(self, start, stop, timeframe = 'day'):
+    def __init__(self, start, stop, timeframe='day'):
         self.timeframe = timeframe
         self.start = shifted_floor(start, self.timeframe)
         self.stop = shifted_ceil(stop, self.timeframe)

diff --git a/watson/watson.py b/watson/watson.py
@@ -556,7 +556,7 @@ def report(self, from_, to, current=None, projects=None, tags=None,
                             cur['tags'], id="current")
 
         span = self.frames.span(from_, to)
-        
+
         frames_by_project = sorted_groupby(
             self.frames.filter(
                 projects=projects or None, tags=tags or None,