Daily-Dose/main.py

294 lines
7.3 KiB
Python

import configparser
from datetime import date
import feedparser
from mdutils import MdUtils
from os import path
from shutil import copy
#################
# Configuration #
#################
markdown = True
pdf = False
html = True
animations = True
html_stylesheet = "styles/simple.css"
title_animation = "fade-down"
heading_animation = "fade-right"
list_animation = "fade-left"
ConvertToHTML = True
title = date.today().strftime('%d %B, %Y')
og_title = title
feed_configuration = ['feeds.ini','covid.ini']
archive_configuration = {"feeds.ini":{"id":"daily-dose-list","folder":"archive-daily-dose",'title':"Daily-Dose","hide_overflow":True},"covid.ini":{"id":"covid-19-list","folder":"archive-covid-19","title":"Covid-19 Sentry","hide_overflow":False}}
update_archive = True
#############
# Functinos #
#############
def GetPosts(feed):
errored = False
Posts = {}
ToIgnore = []
ReadSummary = True
URL = None
ShowLink = False
ToRead = 5
try:
URL = feeds[feed]["URL"]
except KeyError:
print("Improper configuration for %s. No URL Specified")
exit(1)
try:
ToIgnore = eval(feeds[feed]["IGNORE"])
except KeyError:
None
try:
ReadSummary = eval(feeds[feed]["SUMMARY"])
except KeyError:
ReadSummary = True
try:
ToRead = eval(feeds[feed]["ToRead"])
except:
None
try:
ShowLink = eval(feeds[feed]["LINK"])
except KeyError:
None
try:
rss = feedparser.parse(URL)
except:
print("Could not fetch data for feed: %s" % feed)
errored = True
if errored:
return None
maximum = len(rss.entries)
count = 0
if ToRead == -1 or ToRead > maximum:
ToRead = maximum
posts = []
summaries = []
links = []
added = 0
while count != maximum and added != ToRead:
Skip = False
Title = str(rss.entries[count].title)
Summary = rss.entries[count].summary
for words in ToIgnore:
if (words.lower() in Title.lower()) or (words.lower() in Summary.lower()):
Skip = True
break
if not Skip:
added += 1
posts.append(Title)
if ReadSummary:
summaries.append(Summary)
if ShowLink:
links.append(rss.entries[count].link)
count += 1
for idx in range(len(posts)):
Posts[idx] = {}
Posts[idx]["post"] = {}
Posts[idx]["post"]["title"] = posts[idx].encode('utf-8')
if ReadSummary:
Posts[idx]["post"]["summary"] = summaries[idx].encode('utf-8')
else:
Posts[idx]["summary"] = None
if ShowLink:
Posts[idx]["post"]["link"] = links[idx].encode('utf-8')
else:
Posts[idx]["post"]["link"] = None
Posts["NoOfPosts"] = len(posts)
Posts["Summary"] = ReadSummary
Posts["ShowLink"] = ShowLink
return Posts
def add_today(HomeFile,TagID,folder):
title = og_title
if path.exists(title+".html") or path.exists(str("./%s/"%folder)+title+".html"):
if path.exists(str("./%s/"%folder)+title+".html"):
None
else:
copy(str(title+".html"),folder)
else:
print("File does not exist!")
exit(1)
TagToFind = 'ul'
TagID = TagID
soup = None
with open(HomeFile) as fp:
soup = BeautifulSoup(fp,'html.parser')
ToUpdate = soup.find(TagToFind,{"id":TagID})
FilePath = str("./%s/"%folder) + str(str(title)+".html")
NewTag = soup.new_tag('li')
anchor = soup.new_tag('a',href=str('./%s/'%folder+title+".html"))
anchor.string = title
NewTag.append(anchor)
ToUpdate.insert(0,NewTag)
with open(HomeFile,'w') as fp:
fp.write(str(soup))
################
# Main Program #
###############
if (pdf and not markdown) or (html and not markdown):
print("Markdown should be True to convert to pdf/html")
exit(1)
if (update_archive and not html):
print("HTML is required to update archive")
exit(1)
for config in feed_configuration:
title = og_title
feeds = configparser.ConfigParser()
if feeds.read(config) == []:
print("%s does not exist!"%config)
exit(1)
else:
print("Reading %s"%config)
feeds.read(config)
rss_feeds = [x for x in feeds.keys()]
rss_feeds.pop(0)
print("Read %s feeds from the configuration file" % str(len(rss_feeds)))
print(rss_feeds)
posts = {}
for feed in rss_feeds:
sauce = GetPosts(feed)
if sauce == None:
rss_feeds.remove(feed)
else:
posts[feed] = sauce
#print(posts)
if markdown:
mdfile = None
mdfile = MdUtils(file_name=title,title=archive_configuration[config]['title'])
for feed in posts:
mdfile.new_header(level=1,title="From %s" % feed)
for idx in range(posts[feed]["NoOfPosts"]):
mdfile.write(str("* **"+posts[feed][idx]["post"]["title"].decode('utf-8'))+"**")
if posts[feed]["Summary"]:
mdfile.write(" - ")
mdfile.write(posts[feed][idx]["post"]["summary"].decode('utf-8'))
if posts[feed]["ShowLink"]:
mdfile.write(" - [link](%s)"%posts[feed][idx]["post"]["link"].decode('utf-8'))
mdfile.write("\n\n")
elif posts[feed]["Summary"]:
mdfile.write("\n\n")
else:
mdfile.write("\n")
mdfile.new_line("\n")
mdfile.new_table_of_contents(table_title='Contents', depth=2)
mdfile.create_md_file()
import pypandoc
if pdf:
ifname = str(title.strip()) + ".md"
ofname = str(title.strip()) + ".pdf"
args = ['-V', 'geometry:margin=1.5cm']
convert = pypandoc.convert_file(ifname, 'pdf', outputfile=ofname,extra_args=args)
if html:
from bs4 import BeautifulSoup
ifname = str(title.strip()) + ".md"
ofname = str(title.strip()) + ".html"
convert = pypandoc.convert_file(ifname,'html',outputfile=ofname,extra_args=['-s'])
assert(convert) == ''
soup = None
with open(ofname) as fp:
soup = BeautifulSoup(fp,'html5lib')
title = soup.new_tag('title')
title.string = archive_configuration[config]['title']
soup.head.append(title)
viewport = soup.new_tag("meta",content="width=device-width, initial-scale=1.0")
viewport.attrs["name"] = "viewport"
soup.head.append(viewport)
custom_css = soup.new_tag('link',href=html_stylesheet,rel='stylesheet')
soup.head.append(custom_css)
custom_css = soup.new_tag('link',href=str("../"+html_stylesheet),rel='stylesheet')
soup.head.append(custom_css)
if archive_configuration[config]["hide_overflow"]:
a = soup.new_tag('style')
a.string = "*{overflow-x:hidden;}"
soup.head.append(a)
if animations:
aos_css = soup.new_tag('link',href='https://unpkg.com/aos@2.3.1/dist/aos.css',rel='stylesheet')
soup.head.append(aos_css)
aos_js = soup.new_tag('script',src="https://unpkg.com/aos@2.3.1/dist/aos.js")
soup.head.append(aos_js)
aos_script = soup.new_tag('script')
aos_script.string = "AOS.init();"
soup.body.append(aos_script)
for feed in rss_feeds:
ToFindID = str("from-"+str(feed.strip().replace(":","").replace(" ","-").lower()))
ToEdit = soup.find("h1", {"id": ToFindID})
ToEdit['data-aos'] = 'fade-right'
soup.find_all("h1")[0]['data-aos'] = title_animation
soup.find("h1",{"id":"contents"})['data-aos'] = heading_animation
soup.find("h1",{"id":"contents"})['data-aos-anchor-placement'] = "top-bottom"
paragraphs = soup.find_all("p")
for paras in paragraphs:
paras['data-aos'] = list_animation
paras['data-aos-anchor-placement'] = "bottom-bottom"
lis = soup.find_all("li")
for li in lis:
if li.a == None:
li['data-aos'] = list_animation
li['data-aos-anchor-placement'] = "bottom-bottom"
with open(ofname, "w") as outf:
outf.write(str(soup))
if update_archive:
add_today('index.html',archive_configuration[config]["id"],folder=archive_configuration[config]["folder"])