Adding argparse

test1.py

@@ -1,44 +0,0 @@
-#!/usr/bin/python3
-import sys
-from PySide import QtWebKit, QtCore, QtGui
-
-#page = QtWebKit.QWebPage()
-#page.mainFrame().load(url)
-
-app = QtGui.QApplication(sys.argv)
-
-printer = QtGui.QPrinter()
-printer.setOutputFileName('out.pdf')
-printer.setOutputFormat(QtGui.QPrinter.PdfFormat)
-printer.setPageSize(QtGui.QPrinter.A4)
-printer.setOrientation(QtGui.QPrinter.Landscape)        
-printer.setFullPage(True)
-
-image = QtGui.QImage(1024,768, QtGui.QImage.Format_ARGB32)
-
-view = QtWebKit.QWebView()
-
-def out(ok):
-    print('out')
-    view.page().setViewportSize(view.page().currentFrame().contentsSize())
-    image = QtGui.QImage(view.page().viewportSize(),QtGui.QImage.Format_ARGB32)
-    painter = QtGui.QPainter(image)
-    view.page().mainFrame().render(painter)
-    painter.end()
-    image.save('out.png')
-    view.page().mainFrame().print_(printer)
-    print('adios')
-    app.exit()
-
-view.loadFinished.connect(out)
-view.load(QtCore.QUrl('http://www.gnu.org'))
-print("load")
-
-#image = QtGui.QImage(1024,768, QtGui.QImage.Format_ARGB32)
-#view.render(image)
-#image.save('out.png')
-view.page().mainFrame().print_(printer)
-#view.render(printer)
-
-#view.show()
-sys.exit(app.exec_())

webpage2img.py

@@ -1,6 +1,6 @@
 #!/usr/bin/python3
 import sys
-import utils
+import argparse
 from PySide import QtWebKit, QtCore, QtGui
 
 #self._app = QtGui.QApplication(sys.argv)
@@ -39,8 +39,7 @@ class WebPage2Img(object):
         self.wxview.page().mainFrame().render(painter)
         painter.end()
         image.save(self._out_fname)
-        #self.wxview.page().mainFrame().print_(printer)
-        self.__exit()
+        self.exit()
 
     def _pdf(self, ok):
         print('_pdf')
@@ -51,10 +50,10 @@ class WebPage2Img(object):
         printer.setOrientation(QtGui.QPrinter.Landscape)        
         #printer.setFullPage(True)
         self.wxview.page().mainFrame().print_(printer)
-        self.__exit()
+        self.exit()
 
         
-    def __exit(self, retcode = 0):
+    def exit(self, retcode = 0):
         print('Exiting.')
         self._app.exit(retcode)
         exit(retcode)
@@ -66,6 +65,24 @@ class WebPage2Img(object):
         self.wxview.load(QtCore.QUrl(self.uri))
         sys.exit(self._app.exec_())
 
-wp = WebPage2Img(sys.argv[1])
-#wp.png('out.png')
-wp.pdf('out.pdf')
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Convert a webpage to an \
+image.')
+    parser.add_argument('-f', '--format',  dest='fmt', metavar='pdf|png', type=str,
+        default='pdf', help='Choose output format')
+    parser.add_argument('-o', '--output', metavar='FILENAME', type=str,
+        default='out.pdf', help='Output file')
+    parser.add_argument('uri', metavar='URI', type=str)
+    args = parser.parse_args()
+
+    wp = WebPage2Img(args.uri)
+    if args.fmt.lower() == 'png':
+        wp.png(args.output)
+    elif args.fmt.lower() == 'pdf':
+        wp.pdf(args.output)
+    else:
+        print('Unknown format "%s"\n' % args.fmt)
+        parser.print_help()
+        wp.exit(1)
+