From e8275c472b2f114ad435692236b8a6f46e4169b0 Mon Sep 17 00:00:00 2001 From: "Thomas \"Cakeisalie5\" Touhey" Date: Fri, 5 May 2017 16:40:01 +0200 Subject: [PATCH] Better file name detection. --- slideslurp/__init__.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/slideslurp/__init__.py b/slideslurp/__init__.py index d76d8a7..c98e58f 100644 --- a/slideslurp/__init__.py +++ b/slideslurp/__init__.py @@ -3,6 +3,7 @@ import sys import bs4 import requests +from urllib.parse import urlparse from reportlab.pdfgen import canvas @@ -13,11 +14,14 @@ def parse_args(): parser = argparse.ArgumentParser(description=descr) parser.add_argument('url', metavar='url', nargs=1, help='the URL to slurp') - parser.add_argument('--output', '-o', default='out.pdf', + parser.add_argument('--output', '-o', default=None, help='the file to write to (default: out.pdf)') - return parser.parse_args() - + args = parser.parse_args() + if not args.output: + path = urlparse(args.url[0]).path + args.output = '%s.pdf'%path[path.find('/', 1) + 1:] + return args def main(): args = parse_args() @@ -31,7 +35,8 @@ def main(): img_url = img.attrs["data-full"] page_width, page_height = c._pagesize c.setPageRotation(90) - c.drawImage(img_url, 0, 0, page_height, page_width, preserveAspectRatio=True) + c.drawImage(img_url, 0, 0, page_height, page_width, + preserveAspectRatio=True) c.showPage() c.save()