9
9
10
10
"""
11
11
import logging
12
-
13
12
from tkinter import ttk
14
13
14
+ from selenium .webdriver import Firefox
15
+
15
16
from openwpm .commands .types import BaseCommand
16
17
from openwpm .config import BrowserParams , ManagerParams
17
18
from openwpm .socket_interface import ClientSocket
18
19
19
- from selenium .webdriver import Firefox
20
-
21
20
22
- def get_screen_resolution (driver ) :
21
+ def get_screen_resolution (driver : Firefox ) -> list [ int ] :
23
22
return driver .execute_script ("return [screen.width, screen.height];" )
24
-
23
+
25
24
26
25
class SetResolution (BaseCommand ):
27
- """ Sets the browser window resolution """
28
- def __init__ (self , width , height ) -> None :
26
+ """Sets the browser window resolution"""
27
+
28
+ def __init__ (self , width : int , height : int ) -> None :
29
29
self .logger = logging .getLogger ("openwpm" )
30
30
self .width = width
31
31
self .height = height
@@ -39,22 +39,22 @@ def execute(
39
39
browser_params : BrowserParams ,
40
40
manager_params : ManagerParams ,
41
41
extension_socket : ClientSocket ,
42
- ):
43
-
42
+ ) -> None :
44
43
self .logger .info (f"Setting window resolution to { self .width } x { self .height } " )
45
44
driver .set_window_size (self .width , self .height )
46
-
45
+
47
46
resolution = get_screen_resolution (driver )
48
47
if resolution [0 ] <= self .width or resolution [1 ] <= self .height :
49
48
self .logger .warn (
50
- f"Browser window resolution ({ self .width } x { self .height } ) exceeds " +
51
- f"screen resolution ({ resolution [0 ]} x { resolution [1 ]} )" )
52
-
53
-
49
+ f"Browser window resolution ({ self .width } x { self .height } ) exceeds "
50
+ + f"screen resolution ({ resolution [0 ]} x { resolution [1 ]} )"
51
+ )
52
+
54
53
55
54
class SetPosition (BaseCommand ):
56
- """ Sets the browser window position """
57
- def __init__ (self , x , y ) -> None :
55
+ """Sets the browser window position"""
56
+
57
+ def __init__ (self , x : int , y : int ) -> None :
58
58
self .logger = logging .getLogger ("openwpm" )
59
59
self .x = x
60
60
self .y = y
@@ -68,6 +68,5 @@ def execute(
68
68
browser_params : BrowserParams ,
69
69
manager_params : ManagerParams ,
70
70
extension_socket : ClientSocket ,
71
- ):
72
-
73
- driver .set_window_position (self .x , self .y , windowHandle = 'current' )
71
+ ) -> None :
72
+ driver .set_window_position (self .x , self .y , windowHandle = "current" )
0 commit comments