def download(args):
"""
Implement the 'greg download' command
"""
session = c.Session(args)
issues = aux.parse_for_download(args)
if issues == ['']:
sys.exit("You need to give a list of issues, of the form "
"a, b-c, d..."
"")
dumpfilename = os.path.join(session.data_dir, 'feeddump')
if not os.path.isfile(dumpfilename):
sys.exit(("You need to run "
"greg check"
"<feed>"
" before using "
"greg download"
"."))
with open(dumpfilename, mode='rb') as dumpfile:
dump = pickle.load(dumpfile)
try:
feed = c.Feed(session, dump[0], dump[1])
except Exception:
sys.exit(("... something went wrong."
"Are you sure your last "
"greg check"
" went well?"))
for number in issues:
entry = dump[1].entries[eval(number)]
feed.info = []
feed.entrylinks = []
feed.fix_linkdate(entry)
feed.download_entry(entry)
def check(args):
"""
Implement the 'greg check' command
"""
session = c.Session(args)
if str(args["url"]) != 'None':
url = args["url"]
name = "DEFAULT"
else:
try:
url = session.feeds[args["feed"]]["url"]
name = args["feed"]
except KeyError:
sys.exit("You don't appear to have a feed with that name.")
podcast = aux.parse_podcast(url)
for entry in enumerate(podcast.entries):
listentry = list(entry)
print(listentry[0], end=": ")
try:
print(listentry[1]["title"], end=" (")
except:
print(listentry[1]["link"], end=" (")
try:
print(listentry[1]["updated"], end=")")
except:
print("", end=")")
print()
dumpfilename = os.path.join(session.data_dir, 'feeddump')
with open(dumpfilename, mode='wb') as dumpfile:
dump = [name, podcast]
pickle.dump(dump, dumpfile)
def sync(args):
"""
Implement the 'greg sync' command
"""
import operator
session = c.Session(args)
if "all" in args["names"]:
allfeeds = session.list_feeds()
# We want to check if a feed should be synced without actually
# parsing it and all. The "bare" option in the Feed class is
# a hacky solution that avoids a major rewrite.
targetfeeds = []
for target in allfeeds:
testfeed = c.Feed(session, target)
if testfeed.shouldsync:
targetfeeds.append(target)
else:
print("Skipping", target, end=".\n")
else:
targetfeeds = []
for name in args["names"]:
if name not in session.feeds:
print("You don't have a feed called {}.".format(name),
file=sys.stderr,
flush=True)
else:
targetfeeds.append(name)
for target in targetfeeds:
feed = c.Feed(session, target, None)
if not feed.wentwrong:
try:
title = feed.podcast.target.title
except AttributeError:
title = target
print("Checking", title, end="...\n")
currentdate, stop = feed.how_many()
entrycounter = 0
entries_to_download = feed.podcast.entries
for entry in entries_to_download:
feed.fix_linkdate(entry)
# Sort entries_to_download, but only if you want to download as
# many as there are
if stop >= len(entries_to_download):
entries_to_download.sort(key=operator.attrgetter("linkdate"),
reverse=False)
for entry in entries_to_download:
if entry.linkdate > currentdate:
downloaded = feed.download_entry(entry)
entrycounter += downloaded
if entrycounter >= stop:
break
print("Done")
else:
msg = ''.join([
"I cannot sync ", feed,
" just now. Are you connected to the internet?"
])
print(msg, file=sys.stderr, flush=True)
def info(args):
"""
Provide information of a number of feeds
"""
session = c.Session(args)
if "all" in args["names"]:
feeds = session.list_feeds()
else:
feeds = args["names"]
for feed in feeds:
aux.pretty_print(session, feed)
def edit(args): # Edits the information associated with a certain feed
session = c.Session(args)
feed_info = os.path.join(session.data_dir, args["name"])
if not args["name"] in session.feeds:
sys.exit("You don't have a feed with that name.")
for key, value in args.items():
if value is not None and key == "url":
session.feeds[args["name"]][key] = str(value)
with open(session.data_filename, 'w') as configfile:
session.feeds.write(configfile)
if value is not None and key == "downloadfrom":
try:
dateinfo = (session.feeds[args["name"]]["date_info"] ==
"not available")
except KeyError:
session.feeds[args["name"]]["date_info"] = "available"
# provisionally!
with open(session.data_filename, 'w') as configfile:
session.feeds.write(configfile)
dateinfo = False # provisionally
if dateinfo:
print(("{} has no date information that I can use."
"Using --downloadfrom might not have the"
"results that you expect.").format(args["name"]),
file=sys.stderr,
flush=True)
# A dummy entry with the new downloadfrom date.
try:
# Remove from the feed file all entries
# after or equal to downloadfrom, then append line
with open(feed_info, 'r') as previous:
previouslist = previous.readlines()
current = [
aline for aline in previouslist
if value > aux.get_date(aline)
]
with open(feed_info, 'w') as currentfile:
currentfile.writelines(current)
with open(feed_info, 'a') as currentfile:
json.dump(
{
'entrylink': "added by the edit command",
'linkdate': value
}, currentfile)
currentfile.write('\n')
except FileNotFoundError:
# Write the file with a dummy entry with the right date
with open(feed_info, 'w') as currentfile:
json.dump(
{
'entrylink': "added by the edit command",
'linkdate': value
}, currentfile)
currentfile.write('\n')
def sync(args):
"""
Implement the 'greg sync' command
"""
import operator
session = c.Session(args)
if "all" in args["names"]:
targetfeeds = session.list_feeds()
else:
targetfeeds = []
for name in args["names"]:
if name not in session.feeds:
print("You don't have a feed called {}.".format(name),
file=sys.stderr,
flush=True)
else:
targetfeeds.append(name)
for target in targetfeeds:
feed = c.Feed(session, target, None)
if not feed.wentwrong:
try:
title = feed.podcast.target.title
except AttributeError:
title = target
print("Checking", title, end="...\n")
currentdate, stop = feed.how_many()
entrycounter = 0
entries_to_download = feed.podcast.entries
for entry in entries_to_download:
feed.fix_linkdate(entry)
# Sort entries_to_download, but only if you want to download as
# many as there are
if stop >= len(entries_to_download):
entries_to_download.sort(key=operator.attrgetter("linkdate"),
reverse=False)
for entry in entries_to_download:
if entry.linkdate > currentdate:
downloaded = feed.download_entry(entry)
entrycounter += downloaded
if entrycounter >= stop:
break
print("Done")
else:
msg = ''.join(
["I cannot sync ", feed, " just now: ", feed.wentwrong])
print(msg, file=sys.stderr, flush=True)
def add(args):
"""
Add a new feed
"""
session = c.Session(args)
if args["name"] in session.feeds.sections():
sys.exit("You already have a feed with that name.")
if args["name"] in ["all", "DEFAULT"]:
sys.exit(
("greg uses "
"{}"
" for a special purpose."
"Please choose another name for your feed.").format(args["name"]))
entry = {}
for key, value in args.items():
if value is not None and key != "func" and key != "name":
entry[key] = value
session.feeds[args["name"]] = entry
with open(session.data_filename, 'w') as configfile:
session.feeds.write(configfile)
def remove(args):
"""
Remove the feed given in <args>
"""
session = c.Session(args)
if not args["name"] in session.feeds:
sys.exit("You don't have a feed with that name.")
inputtext = ("Are you sure you want to remove the {} "
" feed? (y/N) ").format(args["name"])
reply = input(inputtext)
if reply != "y" and reply != "Y":
return 0
else:
session.feeds.remove_section(args["name"])
with open(session.data_filename, 'w') as configfile:
session.feeds.write(configfile)
try:
os.remove(os.path.join(session.data_dir, args["name"]))
except FileNotFoundError:
pass
def list_for_user(args):
session = c.Session(args)
for feed in session.list_feeds():
print(feed)
print()
