--- 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) :
"""