Python自定义大小截屏

简介: 蝈蝈这两天正忙着收拾家当去公司报道,结果做PHP的发小蛐蛐找到了他,说是想要一个可以截图工具。大致需要做出这样的效果。 虽然已经很久不写Python代码了,但是没办法,盛情难却啊,只好硬着头皮上了。

蝈蝈这两天正忙着收拾家当去公司报道,结果做PHP的发小蛐蛐找到了他,说是想要一个可以截图工具。

大致需要做出这样的效果。
自定义截图实现

虽然已经很久不写Python代码了,但是没办法,盛情难却啊,只好硬着头皮上了。


关于这个需求,蝈蝈想了想,脑海里大概有这么几个实现的方式。

  • 调用QQ的截图工具。
  • 自己写一个。

这第一个嘛,应了那句老话。理想很丰满,现实很骨感。因为被集成的缘故,剖不出来是没办法用的,自认为技术还不到家的蝈蝈很快放弃了这个方法。

那么只能自己写一个了。从谷哥那了解到PIL的ImageGrab可以很方便的截图,默认截图是全屏范围,当然也可以传递一个Bbox元组来实现截图的范围截图。于是思路就很明确了:获取鼠标位置,调用ImageGrab截图


获取鼠标位置
这个嘛,其实还是很简单的。借助pyHook就可以啦。

global old_x, old_y, new_x, new_y, full, hm
    if event.MessageName == "mouse left down":
        old_x, old_y = event.Position
    if event.MessageName == "mouse left up":
        new_x, new_y = event.Position

按下鼠标的那一刻开始记录初始坐标,然后鼠标抬起的那一刻更新结束坐标。这两个坐标的范围就是要截图的范围。这里面需要注意的就是鼠标坐标默认从左上角(0, 0)开始


截图的具体实现

关于具体实现,无非是一个full标记,默认也是截全屏的图,当fullFalse的时候,按照两次鼠标的绝对位置实现范围截图。

# 划屏
    if full:
        image = ImageGrab.grab((0, 0, gsm(0), gsm(1)))
    else:
        image = ImageGrab.grab((old_x, old_y, new_x, new_y))
    image.show()

好啦,核心功能已经做好啦。为了方便蛐蛐进行自定义的拓展,蝈蝈把源码发给了他。

# coding: utf8

# @Author: 郭 璞
# @File: capture.py                                                                 
# @Time: 2017/7/24                                   
# @Contact: 1064319632@qq.com
# @blog: http://blog.csdn.net/marksinoberg
# @Description: 根据鼠标移动进行划屏截图

import pyHook
import pythoncom
import win32gui
from PIL import Image, ImageGrab
from win32api import GetSystemMetrics as gsm

# 提前绑定鼠标位置事件
old_x, old_y = 0, 0
new_x, new_y = 0, 0
def hotkey(key=None):
    """绑定热键,开始进行划屏截图操作"""
    pass


def on_mouse_event(event):
    global old_x, old_y, new_x, new_y, full, hm
    if event.MessageName == "mouse left down":
        old_x, old_y = event.Position
    if event.MessageName == "mouse left up":
        new_x, new_y = event.Position
        # 解除事件绑定
        hm.UnhookMouse()
        hm = None

    # 划屏
    if full:
        image = ImageGrab.grab((0, 0, gsm(0), gsm(1)))
    else:
        image = ImageGrab.grab((old_x, old_y, new_x, new_y))
    image.show()



full = False
hm = None
def capture():

    hm = pyHook.HookManager()
    hm.SubscribeMouseAll(on_mouse_event)
    hm.HookMouse()
    pythoncom.PumpMessages()


capture()

核心功能已经算是完成了,虽然貌似并没有什么太大的用处。

自定义截图实现


因为就要走了,所以蝈蝈没有多少时间来润色,只能这样匆匆交差了。除了代码,蝈蝈特意嘱咐了下面这几句话:

  • 增加保存到本地功能。
  • 绑定系统快捷键,这样打游戏的时候也可以截图。
  • 增加蒙层,截图的时候提供更好的用户体验。

蛐蛐听完之后,貌似也有了自己的想法,然后就自己琢磨去了。其实他不知道的是,蝈蝈对于截到的图的另一层处理。

简易图片相似度分析

# coding: utf8

# @Author: 郭 璞
# @File: similar.py                                                                 
# @Time: 2017/7/23                                   
# @Contact: 1064319632@qq.com
# @blog: http://blog.csdn.net/marksinoberg
# @Description: 两张图片相似度计算实现。

from PIL import Image


def pixel_way(img1, img2):
    image1 = Image.open(img1, 'r')
    image2 = Image.open(img2, 'r')

    return get_pixel_details(image1)==get_pixel_details(image2)


def get_pixel_details(img):
    pixels = img.load()
    r, g, b = 0, 0, 0
    counter = 0
    for x in range(img.size[0]):
        for y in range(img.size[1]):
            counter += 1
            r1, g1, b1 = pixels[x, y]
            r += r1
            g += g1
            b += b1
    return (r/counter, g/counter, b/counter)



if __name__ == '__main__':
    image1 = r'./1.png'
    image2 = r'./1.png'
    img = Image.open(image1, 'r')
    img.resize((256,256)).convert("RGB")
    print(pixel_way(image1, image2))

图片像素直方图

# coding: utf8

# @Author: 郭 璞
# @File: pixel-compare.py                                                                 
# @Time: 2017/7/24                                   
# @Contact: 1064319632@qq.com
# @blog: http://blog.csdn.net/marksinoberg
# @Description: 计算RGB值相关

from PIL import Image
from PIL import ImageDraw

im = Image.open('1.png')
im = im.convert("L")

width, height = im.size
pix = im.load()
a = [0]*256
for w in range(width):
    for h in range(height):
        p = pix[w, h]
        a[p] = a[p] + 1

x = max(a)
print(a, "---", len(a), '-----', x)

image = Image.new('RGB', (256, 256), (255, 255, 255))
draw = ImageDraw.Draw(image)

for k in range(256):
    a[k] = a[k]*200/x
    source = (k, 255)
    target = (k, 255-a[k])
    draw.line([source, target], (100, 100, 100))

image.show()

还有很多更好玩的,但是有时候,话多,不是一件好事,想到这里,蝈蝈又不自觉的回忆起了那段不堪的帮忙的经历,无奈……

                                      the end.
目录
相关文章
|
23天前
|
Python
探索Python中的魔法方法:打造你自己的自定义对象
【8月更文挑战第29天】在Python的世界里,魔法方法如同神秘的咒语,它们赋予了对象超常的能力。本文将带你一探究竟,学习如何通过魔法方法来定制你的对象行为,让你的代码更具魔力。
37 5
|
4月前
|
索引 Python
Python中的异常处理与自定义异常
Python中的异常处理与自定义异常
44 0
|
1月前
|
前端开发 Python
使用Python+openpyxl实现导出自定义样式的Excel文件
本文介绍了如何使用Python的openpyxl库导出具有自定义样式的Excel文件,包括设置字体、对齐方式、行列宽高、边框和填充等样式,并提供了完整的示例代码和运行效果截图。
30 1
使用Python+openpyxl实现导出自定义样式的Excel文件
|
20天前
|
UED Python
探索Python中的魔法方法:打造自定义字符串表示
【8月更文挑战第31天】在Python的世界里,魔法方法是那些以双下划线开头和结尾的特殊方法,它们为类提供了丰富的功能。本文将带你走进这些魔法方法的背后,特别是__str__和__repr__,揭示如何通过它们来定制我们的对象在被打印或转换为字符串时的外观。我们将从基础用法开始,逐步深入到高级技巧,包括继承与重写,最终实现一个优雅的字符串表示方案。准备好了吗?让我们开始这段代码之旅吧!
|
23天前
|
SQL Shell API
python Django教程 之 模型(数据库)、自定义Field、数据表更改、QuerySet API
python Django教程 之 模型(数据库)、自定义Field、数据表更改、QuerySet API
|
1月前
|
Python
python 协程 自定义互斥锁
【8月更文挑战第6天】这段代码展示了如何在Python的异步编程中自定义一个互斥锁(`CustomMutex`类)。该类通过`asyncio.Lock`实现,并提供`acquire`和`release`方法来控制锁的获取与释放。示例还包含了使用此自定义锁的场景:两个任务(`task1`和`task2`)尝试按序获取锁执行操作,直观地演示了互斥锁的作用。这有助于理解Python协程中互斥锁的自定义实现及其基本用法。
|
2月前
|
监控 安全 数据库
逆天改命!用自定义上下文管理器,让你的Python代码效率飙升
【7月更文挑战第7天】Python上下文管理器简化资源管理,通过自定义实现优雅控制。使用with语句自动执行资源获取和释放,确保异常安全。例如,FileContextManager类通过__enter__打开文件,__exit__关闭并处理异常。自定义上下文管理器可封装重复逻辑,增强功能如日志和监控,提升代码效率与质量。利用这一工具,代码更简洁、高效且易于维护。**
27 1
|
2月前
|
程序员 数据库连接 Python
解锁Python新姿势:上下文管理器的自定义技巧,让你的编程之路更顺畅
【7月更文挑战第7天】Python上下文管理器简化资源管理,确保异常时资源正确释放。通过实现`__enter__`和`__exit__`或使用`contextmanager`装饰器自定义管理器。示例展示了类定义和装饰器方法。自定义管理器提升代码可读性,防止资源泄露,是高效编程的关键。**
27 1
|
2月前
|
安全 数据库连接 Python
告别繁琐!自定义Python上下文管理器,让你的资源管理变得如此简单
【7月更文挑战第4天】在Python中,上下文管理器通过`with`语句简化资源管理,确保资源的自动获取与释放,增强程序稳定性。自定义上下文管理器依靠`__enter__`和`__exit__`方法,例如,`CustomFileManager`类展示了如何记录文件操作。自定义管理器能简化代码、保证资源安全释放和提供定制逻辑,从而提升代码的健壮性和可维护性。它是处理文件、连接等资源管理的强大工具。
22 2
|
2月前
|
Python
Python中自定义迭代器
【7月更文挑战第3天】
18 1