基于Dx实现的截图程序示例代码,Python实现的网页截图功能

本文所述的次第示例为VC++图象特效的截图示例,须求DirectX
3.0上述版,代码中的GetScreen函数是本截图程序的显要。运转这一个顺序可用Esc键停止。代码中须要ddutil.h与ddutil.cpp文件,请自行下载增添。关于InitDDraw(卡塔尔(قطر‎函数,功效是开头化DirectDraw处境,成立换页链(主页面,二个后台缓冲区),以至开创三个机械漏刻。

正文实例陈说了Python完毕的网页截图功用。分享给大家供我们参照他事他说加以考查,具体如下:

实际的功能代码如下:

方法一、使用PyQt4的QtWebKit组件

#include <windows.h>
#include <windowsx.h>
#include <stdio.h>
#include <ddraw.h>
#include <math.h>
#include "ddutil.h"
#define TITLE "截图示例" //窗口标题
#define CLASSNAME "GetScreen" //窗口类名
#define COLORS 16
HINSTANCE hInst; //应用程序实例句柄
HWND hWndMain; //主窗口句柄
LPDIRECTDRAW      lpDD;      // DirectDraw对象
LPDIRECTDRAWSURFACE   lpDDSPrimary;  // 主页面
LPDIRECTDRAWSURFACE   lpDDSBack;   // 后台缓冲区
LPDIRECTDRAWSURFACE   lpDDSPic1;   // 离屏页面1
LPDIRECTDRAWPALETTE   lpDDPal;    // 调色板
BOOL          bActive;    // 应用程序是否活跃?
HBITMAP hbm;
RECT rect;
//函数声明
void FreeObjects( void );
BOOL InitDDraw(void);
BOOL InitSurfaces(void);
void UpdateFrame(void);
void MakeRect(RECT *rect, long left, long top, long right, long bottom);
///////////////////////////////////////////////
// 截图函数,是本程序的关键
// 函数名:GetScreen
// 参数:指向屏幕矩形的指针
// 返值:Bitmap设备
//////////////////////////////////////////////
HANDLE GetScreen(LPRECT lpRect)
{
 //定义屏幕的DC和内存DC
 HDC hScrDC,hMemDC;
 //定义Bitmap设备
 HANDLE hBitmap,hOldBitmap;
 //定义屏幕坐标变量
 unsigned int nX,nY,nX2,nY2;
 unsigned int nWidth,nHeight;
 //定义屏幕分辩率变量
 unsigned int xScrn,yScrn;
 //确保屏幕矩形不为空
 if(IsRectEmpty(lpRect))
 return NULL;
 //创建屏幕的DC
 hScrDC=CreateDC("DISPLAY",NULL,NULL,NULL);
 //创建的内存DC
 hMemDC=CreateCompatibleDC(hScrDC);
 //将屏幕矩形坐标付给坐标变量
 nX=lpRect->left;
 nY=lpRect->top ;
 nX2=lpRect->right ;
 nY2=lpRect->bottom ;
 //获得屏幕分辩率
 xScrn=GetDeviceCaps(hScrDC,HORZRES);
 yScrn=GetDeviceCaps(hScrDC,VERTRES);
 if(nX<0)
 nX=0;
 if(nY<0)
 nY=0;
 if(nX2>xScrn)
 nX2=xScrn;
 if(nY2>yScrn)
 nY2=yScrn;
 //获得屏幕宽度与长度
 nWidth=nX2-nX;
 nHeight=nY2-nY;
 //获得屏幕图像并付给一个Bitmap设备
 hBitmap=CreateCompatibleBitmap(hScrDC,nWidth,nHeight);
 hOldBitmap=(HBITMAP)SelectObject(hMemDC,hBitmap);
 BitBlt(hMemDC,0,0,nWidth,nHeight,hScrDC,nX,nY,SRCCOPY);
 hBitmap=(HBITMAP)SelectObject(hMemDC,hOldBitmap);
 //清除设备
 DeleteDC(hScrDC);
 DeleteDC(hMemDC);
 return hBitmap;
}
//*******************************************************************
//函数:FreeObject
//功能:释放所有DirectDraw对象
//*******************************************************************
void FreeObjects( void )
{
 //释放hbm位图对象
  DeleteObject(hbm);

  if( lpDD != NULL )//释放DirectDraw对象
  {
    if( lpDDSPrimary != NULL )//释放主页面
    {
      lpDDSPrimary->Release();
      lpDDSPrimary = NULL;
    }
    if( lpDDSPic1 != NULL )//释放离屏页面1
    {
      lpDDSPic1->Release();
      lpDDSPic1 = NULL;
    }
    if( lpDDPal != NULL )//释放调色板
    {
      lpDDPal->Release();
      lpDDPal = NULL;
    }
    lpDD->Release();
    lpDD = NULL;
  }
} 
//*******************************************************************
//函数:RestoreAll
//功能:页面丢失后,恢复页面内存
//*******************************************************************
HRESULT RestoreAll( void )
{
  HRESULT   ddrval;
 //恢复主页面,这也将恢复换页链中的所有页面
  ddrval = lpDDSPrimary->Restore();
 //恢复离屏页面
  ddrval = lpDDSPic1->Restore();
 //重新绘制页面图象
 InitSurfaces();
  return ddrval;
}
//*******************************************************************
//函数:WindowProc
//功能:主窗口的消息处理过程
//*******************************************************************
LRESULT CALLBACK WinProc( HWND hWnd, UINT message, 
              WPARAM wParam, LPARAM lParam )
{
  switch( message )
  {
  case WM_SETCURSOR:
 SetCursor(LoadCursor( NULL, IDC_ARROW ));
 return TRUE;
  case WM_ACTIVATEAPP://应用程序激活消息
    bActive = wParam;
    break;
  case WM_KEYDOWN://击键消息
    switch( wParam )
    {
    case VK_ESCAPE:
      PostMessage(hWnd, WM_CLOSE, 0, 0);
      break;
 }
    break;
  case WM_DESTROY://销毁窗口消息
    FreeObjects();
    PostQuitMessage(0);
    break;
  }
 //调用缺省的过程处理过程
  return DefWindowProc(hWnd, message, wParam, lParam);
}
//******************************************************************
//函数:InitWindow()
//功能:创建主窗口。
//******************************************************************
BOOL InitWindow( HINSTANCE hInstance, int nCmdShow )
{
  WNDCLASS wc; //窗口类结构
 //填充窗口类结构
  wc.style = 0;
  wc.lpfnWndProc = WinProc;
  wc.cbClsExtra = 0;
  wc.cbWndExtra = 0;
  wc.hInstance = hInstance;
  wc.hIcon = LoadIcon( hInstance, IDI_APPLICATION );
  wc.hCursor = LoadCursor( NULL, IDC_ARROW );
  wc.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);//选择黑色刷做为窗口背景
  wc.lpszMenuName = NULL;
 wc.lpszClassName = CLASSNAME;
 //注册窗口类
  RegisterClass( &wc );
 //创建主窗口
  hWndMain= CreateWindowEx(
 0,
 CLASSNAME, //窗口的类名称,必须与上面的wc.lpszClassName一致
 TITLE, //窗口的标题名
 WS_POPUP,
 0,
 0,
 GetSystemMetrics( SM_CXSCREEN ),
 GetSystemMetrics( SM_CYSCREEN ),
 NULL,
 NULL,
 hInstance,
 NULL );
  if( !hWndMain ) 
 return FALSE;
 //显示并更新窗口
  ShowWindow( hWndMain, nCmdShow );
 return TRUE;
}

//******************************************************************
//函数:InitDDraw()
//功能:初始化DirectDraw环境,创建换页链(主页面,一个后台缓冲区)
// 以及创建一个定时器。
//******************************************************************
BOOL InitDDraw(void)
{
  DDSURFACEDESC    ddsd;
  DDSCAPS       ddscaps;
  HRESULT       ddrval;
  //创建DirectDraw对象
 ddrval = DirectDrawCreate( NULL, &lpDD, NULL );
  if( ddrval != DD_OK )
    return FALSE;
  //取得全屏独占模式
  ddrval = lpDD->SetCooperativeLevel( hWndMain, DDSCL_EXCLUSIVE | DDSCL_FULLSCREEN );
  if( ddrval != DD_OK )
    return FALSE;
  //设置显示器显示模式为当前屏幕的分辩率,16位增强显示模式
  ddrval = lpDD->SetDisplayMode(GetSystemMetrics(SM_CXSCREEN), GetSystemMetrics(SM_CYSCREEN), COLORS);
  if( ddrval != DD_OK )
    return FALSE;
  //填充换页链结构
  ddsd.dwSize = sizeof( ddsd );
  ddsd.dwFlags = DDSD_CAPS | DDSD_BACKBUFFERCOUNT;
  ddsd.ddsCaps.dwCaps = DDSCAPS_PRIMARYSURFACE |
             DDSCAPS_FLIP |
             DDSCAPS_COMPLEX;
 //后台缓冲区数量为2
  ddsd.dwBackBufferCount = 2;
 //创建换页链,包括了主页面及其后台缓冲区
  ddrval = lpDD->CreateSurface( &ddsd, &lpDDSPrimary, NULL );
  if( ddrval != DD_OK )
    return FALSE;
  //获得指向后台缓冲区的页面指针
 ddscaps.dwCaps = DDSCAPS_BACKBUFFER;
  ddrval = lpDDSPrimary->GetAttachedSurface(&ddscaps, &lpDDSBack);
  if( ddrval != DD_OK )
    return FALSE;
 //创建离屏页面
 ZeroMemory(&ddsd, sizeof(ddsd));
  ddsd.dwSize = sizeof(ddsd);
  ddsd.dwFlags = DDSD_CAPS | DDSD_HEIGHT |DDSD_WIDTH;
  ddsd.ddsCaps.dwCaps = DDSCAPS_OFFSCREENPLAIN;
  ddsd.dwWidth =GetSystemMetrics(SM_CXSCREEN);
  ddsd.dwHeight = GetSystemMetrics(SM_CYSCREEN);
  if (lpDD->CreateSurface(&ddsd, &lpDDSPic1, NULL) != DD_OK)
 return FALSE;
 //调用页面初始化函数
 if( !InitSurfaces() )
    return FALSE;
  return TRUE;
}

//******************************************************************
//函数:WinMain()
//功能:应用程序入口
//******************************************************************
int PASCAL WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance,LPSTR lpCmdLine, int nCmdShow)
{
  MSG  msg;
 hInst=hInstance;
 //获得屏幕分辩率
 rect.left=rect.top=0;
 rect.right=GetSystemMetrics(SM_CXSCREEN);
 rect.bottom=GetSystemMetrics(SM_CYSCREEN);
 //调用截屏函数
 hbm=(HBITMAP)GetScreen(&rect);
 //初始化主窗口
 if (!InitWindow( hInstance, nCmdShow))
  return FALSE;
 //初始化DirectDraw环境
 if (!InitDDraw())
 {
 MessageBox(hWndMain, "初始化DirectDraw过程中出错!", "Error", MB_OK);
 FreeObjects();
 DestroyWindow(hWndMain);
 return FALSE;
 }

 //进入消息循环
 while(1)
 {
 if(PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE))
 {
  if(!GetMessage(&msg, NULL, 0, 0 ))
  return msg.wParam;
  TranslateMessage(&msg); 
  DispatchMessage(&msg);
 }
 else if(bActive)
 {
  UpdateFrame();
 }
 else WaitMessage();
 }
  return msg.wParam;
} 
//******************************************************************
//函数:InitSurfaces()
//功能:初始化页面图象
//******************************************************************
BOOL InitSurfaces( void )
{
 if (hbm == NULL)
 return FALSE;
 DDCopyBitmap(lpDDSPic1, hbm, 0, 0, GetSystemMetrics(SM_CXSCREEN), GetSystemMetrics(SM_CYSCREEN));

  return TRUE;
}
//更新屏幕
void UpdateFrame( void )
{
 HRESULT ddrval;
 //清屏后台缓冲区
 DDBLTFX ddBltFx;
 ddBltFx.dwSize = sizeof(DDBLTFX);
 ddBltFx.dwFillColor = DDColorMatch(lpDDSBack, RGB(0,0,0));
 lpDDSBack->Blt(NULL, NULL, NULL, DDBLT_WAIT | DDBLT_COLORFILL, &ddBltFx);
 //调用灰度的实现函数
 RECT srect, drect;
 MakeRect(&srect, 0, 0, rect.right ,rect.bottom);
 MakeRect(&drect, 0, 0, rect.right , rect.bottom);
 //将背景图象Blit到后台缓冲区
 lpDDSBack->Blt(&drect, lpDDSPic1, &srect, DDBLT_WAIT, NULL);
  // 换页
  while( 1 )
  {
    ddrval = lpDDSPrimary->Flip( NULL, DDFLIP_WAIT );//调用换页函数
    if( ddrval == DD_OK )//成功则退出while循环
      break;
    else if( ddrval == DDERR_SURFACELOST )//如果页面丢失,则恢复页面,再继续while循环
      RestoreAll();
    else
  break;
 }
}
void MakeRect(RECT *rect, long left, long top, long right, long bottom)
{
 rect->left=left;
 rect->top=top;
 rect->right=right;
 rect->bottom=bottom;
}
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys
import os.path
from PyQt4 import QtGui,QtCore,QtWebKit
class PageShotter(QtGui.QWidget):
  def __init__(self,url,filename,parent=None):
    QtGui.QWidget.__init__(self,parent)
    self.url = url
    self.filename = filename
    self.webpage = None
  def shot(self):
    webview = QtWebKit.QWebView(self)
    webview.load(QtCore.QUrl(self.url))
    self.webpage = webview.page()
    self.connect(webview,QtCore.SIGNAL("loadFinished(bool)"),self.save_page)
  def save_page(self,finished):
    #print finished
    if finished:
      print u"开始截图!"
      size = self.webpage.mainFrame().contentsSize()
      print u"页面宽:%d,页面高:%d" % (size.width(),size.height())
      self.webpage.setViewportSize(QtCore.QSize(size.width()+16,size.height()))
      img = QtGui.QImage(size, QtGui.QImage.Format_ARGB32)
      painter = QtGui.QPainter(img)
      self.webpage.mainFrame().render(painter)
      painter.end()
      filename= self.filename;
      if img.save(filename):
        filepath = os.path.join(os.path.dirname(__file__), filename)
        print u"截图完毕:%s" % filepath
      else:
        print u"截图失败";
    else:
      print u"网页加载失败!"
    self.close()
if __name__=="__main__":
  app = QtGui.QApplication(sys.argv)
  shotter = PageShotter("https://www.jb51.net/", 'shot.png')
  shotter.shot()
  sys.exit(app.exec_())

你只怕感兴趣的篇章:

  • C++达成显示屏截图功用
  • C++完成显示器截图

运维后输出:

QFont::setPixelSize: Pixel size <= 0 (0)
发轫截图!
页面宽:1058,页面高:9819
截图完成:C:pyjb51PyDemosrcDemoshot.png

注:

Python2的叁十一个人操作系统安装包可时至今日下载:
陆十二人操作系统下对应PyQt安装包可现今下载:)

Python3可直接利用pip命令安装PyQt5库,如:

pip3 install PyQt5-sip

方法二、使用selenium

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import time
from selenium import webdriver
browser = webdriver.Firefox()
browser.set_window_size(1055, 800)
browser.get("https://www.jb51.net/")
browser.find_element_by_id("idClose").click()
time.sleep(5)
browser.save_screenshot("shot.png")
browser.quit()

PS:selenium库相符能够利用pip命令安装:

pip install selenium

其余,使用selenium时还亟需下载geckodriver来驱动第三方浏览器,对于selenium3.x版本都会接纳geckodriver来驱动firefox,所以必要下载geckodriver.exe,下载地址:

下载后将geckodriver.exe放在C:Python27就能够(查看情状变量path中是还是不是加多C:Python27该路径)

越来越多关于Python相关内容感兴趣的读者可查看本站专项论题:《Python图片操作技巧总括》、《Python数据布局与算法教程》、《Python函数使用工夫总计》、《Python字符串操作本领汇总》、《Python入门与进级卓绝教程》及《Python文件与目录操作手艺汇总》

愿意本文所述对大家Python程序设计具有助于。

你恐怕感兴趣的作品:

  • Python中接收 Selenium
    达成网页截图实例
  • 必赢365net手机版,利用Python保存网页上的图片可能封存页面为截图
  • Python
    完成网页自动截图的亲自过问讲授
  • Python编写显示器截图程序方法
  • python操作摄像头截图完成远程监察和控制的例证
  • Python落成荧屏截图的代码及函数详明
  • Python完成荧屏截图的两种方法
  • 利用Python编写三个在Linux下降成截图分享的本子的教程
  • Python达成对PPT文件进行截图操作的点子
  • Python+selenium完结截图图片并保留截取的图样
  • python调用系统ffmpeg完成录制截图、http发送
  • python搜集博客中上传的QQ截图像和文字件

Leave a Comment.