mirror of
https://github.com/JimmXinu/FanFicFare.git
synced 2025-12-06 08:52:55 +01:00
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
logging.basicConfig(level=logging.DEBUG,format="%(levelname)s:%(filename)s(%(lineno)d):%(message)s")
|
|
|
|
import sys, os
|
|
from optparse import OptionParser
|
|
import getpass
|
|
|
|
from fanficdownloader import adapters,writers,exceptions
|
|
|
|
import ConfigParser
|
|
|
|
def writeStory(config,adapter,writeformat):
|
|
writer = writers.getWriter(writeformat,config,adapter.getStory())
|
|
writer.writeStory()
|
|
del writer
|
|
|
|
def main():
|
|
|
|
# read in args, anything starting with -- will be treated as --<varible>=<value>
|
|
usage = "usage: %prog [options] storyurl"
|
|
parser = OptionParser(usage)
|
|
parser.add_option("-f", "--format", dest="format", default='epub',
|
|
help="write story as FORMAT, epub(default), text or html", metavar="FORMAT")
|
|
parser.add_option("-o", "--option",
|
|
action="append", dest="options",
|
|
help="set an option NAME=VALUE", metavar="NAME=VALUE")
|
|
parser.add_option("-m", "--meta-only",
|
|
action="store_true", dest="metaonly",
|
|
help="Retrieve metadata and stop",)
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
if len(args) != 1:
|
|
parser.error("incorrect number of arguments")
|
|
|
|
config = ConfigParser.SafeConfigParser()
|
|
|
|
logging.debug('reading defaults.ini config file, if present')
|
|
config.read('defaults.ini')
|
|
logging.debug('reading personal.ini config file, if present')
|
|
config.read('personal.ini')
|
|
|
|
try:
|
|
config.add_section("overrides")
|
|
except ConfigParser.DuplicateSectionError:
|
|
pass
|
|
if options.options:
|
|
for opt in options.options:
|
|
(var,val) = opt.split('=')
|
|
config.set("overrides",var,val)
|
|
|
|
try:
|
|
adapter = adapters.getAdapter(config,args[0])
|
|
|
|
try:
|
|
adapter.getStoryMetadataOnly()
|
|
except exceptions.FailedToLogin, ftl:
|
|
print "Login Failed, Need Username/Password."
|
|
sys.stdout.write("Username: ")
|
|
adapter.username = sys.stdin.readline().strip()
|
|
adapter.password = getpass.getpass(prompt='Password: ')
|
|
#print("Login: `%s`, Password: `%s`" % (adapter.username, adapter.password))
|
|
adapter.getStoryMetadataOnly()
|
|
|
|
if options.metaonly:
|
|
print adapter.getStoryMetadataOnly()
|
|
return
|
|
## XXX Use format.
|
|
## XXX Doing all three formats actually causes some interesting
|
|
## XXX config issues with format-specific sections.
|
|
print "format: %s" % options.format
|
|
writeStory(config,adapter,"epub")
|
|
writeStory(config,adapter,"html")
|
|
writeStory(config,adapter,"txt")
|
|
del adapter
|
|
|
|
except exceptions.InvalidStoryURL, isu:
|
|
print isu
|
|
except exceptions.StoryDoesNotExist, dne:
|
|
print dne
|
|
except exceptions.UnknownSite, us:
|
|
print us
|
|
|
|
if __name__ == "__main__":
|
|
main()
|