diff -r cc3ab2c39ded -r 4287fb77e312 log_source.py --- 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) : """