特殊情况下需要分享屏幕? 基于 flask 的微型屏幕共享服务器（可实现屏幕共享）

最新推荐文章于 2024-09-18 14:38:50 发布

Bing@DBA

最新推荐文章于 2024-09-18 14:38:50 发布

阅读量2.1k

点赞数 3

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/qq_42768234/article/details/104301603

版权

python 专栏收录该内容

26 篇文章 9 订阅

订阅专栏

在这里插入图片描述

前言:

今天给大家分享一个有趣的小项目,话说在很久以前我需要做一个报告,原来的屏幕共享软件出了点问题,于是临时写了这个程序来应急,效果还不错,今天分享给大家,如果遇到某些特殊情况,可以用来应急

一、使用环境

Pillow 6.0.0
Opencv-Python 4.1.1.26
flask 1.1.1

二、项目介绍

其实也不是很难,之前我使用 UDP广播的共享屏幕, 现在我们用 Python 有一个库加 Pillow 可以获取屏幕, 可以实现截图效果, 将图片转换成字节流, 但是 Pillow 这个库有一个缺点,如果屏幕使用了缩放他是按比例截取的,等于说你获取的屏幕图片不是完整的图片,接下来我介绍一个如何使用 Python 截取屏幕并转换为字节流 :

Pillow

from PIL import ImageGrab

img = ImageGrab.grab()
img.save("a.png")

pillow 在屏幕缩放情况下无法截取全屏

pyautogui

import pyautogui

img = pyautogui.screenshot()
img.save("a.png")

pyautogui 在屏幕缩放的情况下也可以获取全屏

图片转换为字节流

import pyautogui
from io import BytesIO

img = pyautogui.screenshot()
output_buffer = BytesIO()  # 创建二进制对象
img.save(output_buffer, format='JPEG', quality=100)  # quality提升图片分辨率
frame = output_buffer.getvalue()  # 获取二进制数据
print(frame)

二、动手实践

创建一个新的文件夹 (跟我一起做吧)
在文件夹里再创建一个 templates 文件夹
在 templates 创建一个 index.html
创建 app.py camera_pil.py base_camera.py 三个空的Python 文件
创建完成后将如下代码粘贴到 index.html

<html>
<head>
    <title>屏幕共享</title>
</head>
<body>
<h4>屏幕共享</h4>
<img src="{{ url_for('video_feed') }}">
</body>
</html>

然后再将如下代码粘贴到: base_camera.py

import time
import threading
try:
    from greenlet import getcurrent as get_ident
except ImportError:
    try:
        from thread import get_ident
    except ImportError:
        from _thread import get_ident


class CameraEvent(object):
    def __init__(self):
        self.events = {}

    def wait(self):
        ident = get_ident()
        if ident not in self.events:
            self.events[ident] = [threading.Event(), time.time()]
        return self.events[ident][0].wait()

    def set(self):
        now = time.time()
        remove = None
        for ident, event in self.events.items():
            if not event[0].isSet():
                event[0].set()
                event[1] = now
            else:
                if now - event[1] > 5:
                    remove = ident
        if remove:
            del self.events[remove]

    def clear(self):
        self.events[get_ident()][0].clear()


class BaseCamera(object):
    thread = None
    frame = None
    last_access = 0
    event = CameraEvent()

    def __init__(self):
        if BaseCamera.thread is None:
            BaseCamera.last_access = time.time()

            BaseCamera.thread = threading.Thread(target=self._thread)
            BaseCamera.thread.start()

            while self.get_frame() is None:
                time.sleep(0)

    def get_frame(self):
        BaseCamera.last_access = time.time()

        BaseCamera.event.wait()
        BaseCamera.event.clear()

        return BaseCamera.frame

    @staticmethod
    def frames():
        raise RuntimeError('Must be implemented by subclasses.')

    @classmethod
    def _thread(cls):
        print('Starting camera thread.')
        frames_iterator = cls.frames()
        for frame in frames_iterator:
            BaseCamera.frame = frame
            BaseCamera.event.set()
            time.sleep(0)
            if time.time() - BaseCamera.last_access > 10:
                frames_iterator.close()
                print('Stopping camera thread due to inactivity.')
                break
        BaseCamera.thread = None

再将如下代码粘贴到: camera_pil.py

from io import BytesIO
import cv2
from PIL import ImageGrab, Image
from base_camera import BaseCamera


class Camera(BaseCamera):
    video_source = 0

    @staticmethod
    def set_video_source(source):
        Camera.video_source = source

    @staticmethod
    def frames():
        camera = cv2.VideoCapture(Camera.video_source)
        if not camera.isOpened():
            raise RuntimeError('Error')

        while True:
            image = ImageGrab.grab()  # 获取屏幕数据
            # w, h = image.size
            image = image.resize((1366, 750), Image.ANTIALIAS)  # 图片缩放
            output_buffer = BytesIO()  # 创建二进制对象
            image.save(output_buffer, format='JPEG', quality=100)  # quality提升图片分辨率
            frame = output_buffer.getvalue()  # 获取二进制数据
            yield frame  # 生成器返回一张图片的二进制数据

再将如下代码粘贴到: apps.py

import os
from importlib import import_module
from flask import Flask, render_template, Response

if os.environ.get('CAMERA'):
    Camera = import_module('camera_' + os.environ['CAMERA']).Camera
else:
    from camera_pil import Camera

app = Flask(__name__)


@app.route('/')
def index():
    """
    视图函数
    :return:
    """
    return render_template('index.html')


def gen(camera):
    """
    流媒体发生器
    """
    while True:
        frame = camera.get_frame()

        yield (b'--frame\r\n'
               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')


@app.route('/video_feed')
def video_feed():
    """流媒体数据"""
    return Response(gen(Camera()),
                    mimetype='multipart/x-mixed-replace; boundary=frame')


if __name__ == '__main__':
    app.run(threaded=True, host="127.0.0.1", port=80)

运行 app.py ,然后打开你的浏览器输入: http://127.0.0.1/ 试试看吧!
如果要使用的话, 请把 host=“127.0.0.1” 改为 host=“0.0.0.0” 然后运行屏幕共享服务器, 查看自己以太网的 ip 告诉给接收者,让他们使用浏览器访问你的 ip 地址,就可以实现屏幕共享了,你会发现你的CPU会狂转（也没那么夸张）毕竟是应急的嘛 ~ 嘿嘿嘿