import calendar
from zope import interface, component
from DateTime import DateTime
from quintagroup.blog.star.interfaces import IQGBlogEntryRetriever
from Products.CMFCore.utils import getToolByName
from OFS.interfaces import IFolder
from Products.ATContentTypes.interface import IATTopic

class FolderEntryGetter:
    """Gets blog entries in any sort of folder"""
    
    interface.implements(IQGBlogEntryRetriever)
    component.adapts(IFolder)
    
    def __init__(self, context):
        self.context = context
    
    def _base_query(self):
        portal_properties = getToolByName(self.context, 'portal_properties', None)
        site_properties = getattr(portal_properties, 'site_properties', None)
        portal_types = site_properties.getProperty('blog_types', None)
        if portal_types is None:
            portal_types = ('Document', 'News Item', 'File')

        path = '/'.join(self.context.getPhysicalPath())
        
        return dict(path={'query': path, 'depth':1},
                    portal_type=portal_types,
                    sort_on='effective', sort_order='reverse')
    
    def get_entries(self, year=None, month=None, **kw):
        catalog = getToolByName(self.context, 'portal_catalog')
        query = self._base_query()
        if year:
            if month:
                lastday = calendar.monthrange(year, month)[1]
                startdate = DateTime(year, month, 1, 0, 0)
                enddate = DateTime(year, month, lastday, 23, 59, 59)
            else:
                startdate = DateTime(year, 1, 1, 0, 0)
                enddate = DateTime(year, 12, 31, 0, 0)
            query['effective'] = dict(query=(startdate, enddate),
                                      range='minmax')
        #import pdb;pdb.set_trace()
        query.update(dict([(k,v) for k,v in kw.items() if v]))

        return catalog.searchResults(**query)

    
class TopicEntryGetter(FolderEntryGetter):
    """Gets blog entries for collections"""
    
    interface.implements(IQGBlogEntryRetriever)
    component.adapts(IATTopic)
    
    def __init__(self, context):
        self.context = context
        
    def _base_query(self):
        return self.context.buildQuery()