log_source.py
changeset 77 4287fb77e312
parent 76 cc3ab2c39ded
child 78 85345abbd46a
--- a/log_source.py	Tue Feb 10 01:02:26 2009 +0200
+++ b/log_source.py	Tue Feb 10 01:05:35 2009 +0200
@@ -2,7 +2,7 @@
     A source of IRC log files
 """
 
-import datetime, calendar, itertools, functools
+import datetime, calendar, itertools, functools, math
 import os, errno
 import pytz
 
@@ -42,24 +42,34 @@
         else :
             skip = None
 
+        # count the full number of lines
+        line_count = 0
+
         # collect lines
         lines = []
 
         # iterate using get_date
         for line in self.get_date(dt) :
+            # count them
+            line_count += 1
+
             # skip?
             if skip :
                 skip -= 1
                 continue
-            
+ 
+            # already full?
+            if len(lines) > count :
+                continue
+           
             # store line
             lines.append(line)
-
-            # count?
-            if len(lines) >= count :
-                break
-
-        return (page, 0, lines)
+        
+        # calculate max_pages
+        max_pages = math.ceil(float(line_count) / count)
+        
+        # return
+        return (page, max_pages, lines)
 
     def get_month_days (self, dt) :
         """