diff options
| -rwxr-xr-x | 4grab.py | 83 |
1 files changed, 54 insertions, 29 deletions
@@ -50,7 +50,8 @@ def walk_with_wizard(baseurl): thread = inp inp = raw_input("Which category is this thread in? ") print wzrd_msg - t = downloader.get_image_links("%s%s/res/" % (baseurl, inp), [thread]) + t = downloader.get_image_links("%s%s/res/" % (baseurl, inp), + [thread]) else: inp = raw_input("Which category would you like to download? ") config.Configuration().set_category(inp) @@ -86,57 +87,81 @@ parser.add_option("-t", "--thread", dest="thread", metavar="THREAD", - help="Download only THREAD. If THREAD is only an ID, CATEGORY must also be set. Otherwise, no problem :-)") + help="Download only THREAD. If THREAD is only an ID, " + "CATEGORY must also be set. Otherwise, no problem :-)") parser.add_option("-w", "--wizard", action="store_true", dest="wizard", - help="I'll put on my robe and wizard hat and help you get some of those pictures you like") - + help="I'll put on my robe and wizard hat and help you " + "get some of those pictures you like") (options, args) = parser.parse_args() -if options.confval and (options.tempcat or options.thread or options.wizard): +if options.confval and (options.tempcat + or options.thread + or options.wizard): print "Can't configure something and do something else too." exit(1) -if options.wizard and (options.tempcat or options.thread or options.confval): +if options.wizard and (options.tempcat + or options.thread + or options.confval): print "Can't take a walk with the wizard and do something else too." exit(1) if options.confval: if not config.Configuration().option_exists(options.confval[0]): - print "%s: error: %s is not a valid configuration option" % (sys.argv[0], options.confval[0]) + print ("%s: error: %s is not a " + "valid configuration option") % (sys.argv[0], + options.confval[0]) exit(1) print "Setting", options.confval[0], "to", options.confval[1] - config.Configuration().set_option(options.confval[0], options.confval[1]) + config.Configuration().set_option(options.confval[0], + options.confval[1]) config.Configuration().save() exit(0) + elif options.wizard: - walk_with_wizard(base_url) + try: + walk_with_wizard(base_url) + except KeyboardInterrupt: + print + print "Alright, no more wizard hat and robe then. Goodbye" exit(0) + elif options.thread: - if options.thread[:7] == "http://": - t = downloader.get_image_links("", [options.thread]) - elif options.tempcat: - url = "%s%s/res/" % (base_url, options.tempcat) - t = downloader.get_image_links(url, [options.thread]) - else: - print "if THREAD is not an absolute URL, CATEGORY must also be specified" - exit(1) - (skipped, failed, downloaded, total) = downloader.get_images(t) - print "Downloaded: ", downloaded - print "Skipped: ", skipped - print "Failed: ", failed - print "Total: ", total + try: + if options.thread[:7] == "http://": + t = downloader.get_image_links("", [options.thread]) + elif options.tempcat: + url = "%s%s/res/" % (base_url, options.tempcat) + t = downloader.get_image_links(url, [options.thread]) + else: + print ("if THREAD is not an absolute URL, " + "CATEGORY must also be specified") + exit(1) + (skipped, failed, downloaded, total) = downloader.get_images(t) + print "Downloaded: ", downloaded + print "Skipped: ", skipped + print "Failed: ", failed + print "Total: ", total + except KeyboardInterrupt: + print + print "Goodbye" exit(0) + elif options.tempcat: config.Configuration().set_category(options.tempcat) base_url = "%s%s/" % (base_url, config.Configuration().get_category()) -t = downloader.get_thread_links(base_url) -t = downloader.get_image_links(base_url, t) -(skipped, failed, downloaded, total) = downloader.get_images(t) -print "Downloaded: ", downloaded -print "Skipped: ", skipped -print "Failed: ", failed -print "Total: ", total +try: + t = downloader.get_thread_links(base_url) + t = downloader.get_image_links(base_url, t) + (skipped, failed, downloaded, total) = downloader.get_images(t) + print "Downloaded: ", downloaded + print "Skipped: ", skipped + print "Failed: ", failed + print "Total: ", total +except KeyboardInterrupt: + print + print "So you don't want these images? Fine! I'll stop then." |