2013-05-14 2 views
-1

У меня есть код html в виде строковых данных. Теперь мне нужно программно отобразить код, сделать скриншот с размером определенного элемента!Как сделать снимок экрана из html в Python

Технически, я хочу выполнить веб-браузер в программе. Возможно ли это?

Я ищу в Python среде

+2

Может быть [Селен] (https://pypi.python.org/pypi/selenium) – root

+0

Да Селен наилучшим решением, он имеет интерфейс Python. – specialscope

+0

@specialscope Как вы думаете, селен может работать на облачных платформах? скажем, google-app-engine, open shift и т. д. (любой PaaS) – Iamcool

ответ

3

Использование PySide или PyQT, это довольно несколько строк кода:

UPDATE: фиксированный код:

from PySide.QtCore import QUrl, QTimer 
from PySide.QtGui import QApplication, QImage, QPainter 
from PySide.QtWebKit import QWebView 


class Browser(QWebView): 

    def __init__(self, app): 
     QWebView.__init__(self) 
     self.parent_app = app 
     self.loadFinished.connect(self._load_finished) 
     self.wait = 5 * 1000 # 5 secs 

    def _load_finished(self, ok): 
     if self.wait: 
      QTimer.singleShot(self.wait, lambda: self._load_finished(ok)) 
      self.wait = None 
      return 

     frame = self.page().mainFrame() 
     self.page().setViewportSize(frame.contentsSize()) 
     image = QImage(self.page().viewportSize(), QImage.Format_ARGB32) 
     painter = QPainter(image) 
     frame.render(painter) 
     painter.end() 
     image.save('test.png') 

     self.close() 
     self.parent_app.quit() 

    def open(self, url): 
     self.load(QUrl(url)) 


if __name__ == '__main__': 
    app = QApplication([]) 
    html = """ 
    <html> 
     <head> 
      <script type="text/javascript"> 
       setTimeout(function() { 
        var e = document.getElementById("later"); 
        e.innerHTML = "arrived"; 
       }, 2500); 
      </script> 
     </head> 
     <body> 
      <div id="later"></div> 
      <div style="margin: 0 auto; width: 500px;"> 
       <img src="http://www.caminodesantiago.me/wp-content/uploads/water-bottle.jpg" /> 
      </div> 
     </body> 
    </html> 
    """ 
    browser = Browser(app) 
    browser.setHtml(html) 
    app.exec_() 
+0

Это похоже на работу .. но у меня есть проблема. В моем коде JS определяет некоторый дизайн страницы. и он должен быть загружен до того, как скрипт примет его изображение. – Iamcool

+0

добавление 'self.wait' не работает – Iamcool

+0

, кроме того, этот скрипт не заканчивается – Iamcool

1

Как уже отмечалось, Selenium WebDriver Python Привязки могут помочь. Ваш код может выглядеть следующим образом:

from selenium import webdriver 
from selenium.common.exceptions import WebDriverException 

browserHandler = webdriver.Firefox() 
browserHandler.get(yourUrl) 
try: 
    browserHandler.get_screenshot_as_file(yourPathToNewImage) 
except WebDriverException: 
    print("WebDriverException caught while trying to get a screenshot") 
Смежные вопросы