pyAutomated/vbox.py at master · roiff/pyAutomated · GitHub

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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
from ast import While
import remotevbox
from PIL import Image
from io import BytesIO
import queue
import time

from remotevbox.exceptions import (
    MachineCloneError,
    MachineCoredumpError,
    MachineCreateError,
    MachineDisableNetTraceError,
    MachineDiscardError,
    MachineEnableNetTraceError,
    MachineExtraDataError,
    MachineInfoError,
    MachineLaunchError,
    MachineLockError,
    MachinePauseError,
    MachinePowerdownError,
    MachineSaveError,
    MachineSetTraceFileError,
    MachineSnaphotError,
    MachineSnapshotError,
    MachineSnapshotNX,
    MachineUnlockError,
    MachineVrdeInfoError,
    ProgressTimeout,
    WrongLockState,
    WrongMachineState,
)

class Vbox:

    def __init__(self) -> None:
        self.vbox = None
        self.machine = None
        self.queue = queue.Queue()
        self.vs = None
        self.fps = None


    def __del__(self) -> None:
        print("Quit Vbox")
        self.vbox.disconnect()

    def init(self)-> None:
        self.vbox = remotevbox.connect("http://127.0.0.1:18083", "wrench", "ljl767689")
        machines = self.vbox.list_machines()
        self.machine = self.vbox.get_machine(machines[0])
        self.machine.launch()
        self.machine.lock()


    def lock(self):
        self.machine.lock()


    def screenshots(self):
        self.screenshot_data = self.machine.take_screenshot_to_bytes(image_format="PNG")
        return self.screenshot_data

    def screenshots_loop(self):
        while True:
            if not self.queue.empty():
                items = self.queue.get(timeout=0.2)
                self.screenshot_data = self.machine.take_screenshot_to_bytes()
                self.queue.task_done()

    def put_mouse_event_absolute(self,
        x,
        y,
        dz=0,
        dw=0,
        left_pressed=True,
        right_pressed=False,
        middle_pressed=False,
        times = 0.1)->None:
        x=int(x)
        y=int(y)
        self.machine.put_mouse_event_absolute(x,
                    y,
                    dz,
                    dw,
                    left_pressed=left_pressed,
                    right_pressed=right_pressed,
                    middle_pressed=middle_pressed)
        time.sleep(times)
        self.machine.put_mouse_event_absolute(x,
                    y,
                    dz,
                    dw,
                    left_pressed=False,
                    right_pressed=right_pressed,
                    middle_pressed=middle_pressed)
        time.sleep(times)