#!/usr/bin/python3
import yaml
import feedparser
import datetime
import sys
import os
from dateutil.parser import parse
import dateutil.tz as tz
import settings

config=settings.load_settings()

with open('bloggers.yml') as f:
    users = yaml.safe_load(f.read())

if not os.path.exists('out'):
    os.makedirs('out')
try:
    with open('out/report.yml') as f:
        log = yaml.safe_load(f.read())
except IOError:
    log = {}

START = datetime.datetime.strptime(config['start_date'],'%Y/%m/%d')

def parse_published(pub):
    try:
        return parse(pub).astimezone(tz.tzlocal()).replace(tzinfo=None)
    except:
        return parse(pub).replace(tzinfo=None)

def get_date(post):
    for k in ('published', 'created', 'updated'):
        if k in post:
            return post[k]

def get_link(post):
    return post.link

def parse_feeds(weeks, uri):
    feedparser.USER_AGENT = "IronBloggerBot/0.2 +http://ironblogger.de/"
    feed = feedparser.parse(uri)

    if not feed.entries:
        print("WARN: no entries for ", uri, file=sys.stderr)
    for post in feed.entries:
        date = parse_published(get_date(post))

        if date < START:
            continue
        wn = int ( (date - START).days / 7 )

        while len(weeks) <= wn:
            weeks.append([])

        if 'title' in post:
            post = dict(date=date,
                        title=post.title,
                        url=get_link(post))
        if 'title' not in post:
            post = dict(date=date,
                        title="",
                        url=get_link(post))
        if post['url'] not in [p['url'] for p in weeks[wn]]:
            weeks[wn].append(post)

if len(sys.argv) > 1:
    for username in sys.argv[1:]:
        blogs = log.setdefault(username, {})
        for l in users[username]['links']:
            weeks = blogs.setdefault(l[0], [])
            parse_feeds(weeks, l[3])
else:
    for (username, u) in list(users.items()):
        if 'end' in u:
            enddate = datetime.datetime.strptime(u['end'],'%Y/%m/%d')
            if enddate < datetime.datetime.now():
                print("User inactive: ", username)
                continue
        blogs = log.setdefault(username, {})
        for l in u['links']:
            weeks = blogs.setdefault(l[0], [])
            parse_feeds(weeks, l[3])

with open('out/report.yml', 'w') as f:
    yaml.safe_dump(log, f)