81 lines
2.1 KiB
Python
Executable File
81 lines
2.1 KiB
Python
Executable File
#!/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()):
|
|
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)
|