2
2
from browserstack .local_binary import LocalBinary
3
3
from browserstack .bserrors import BrowserStackLocalError
4
4
5
+ try :
6
+ from importlib .metadata import version as package_version , PackageNotFoundError
7
+ except :
8
+ import pkg_resources
9
+
10
+
5
11
class Local :
6
12
def __init__ (self , key = None , binary_path = None , ** kwargs ):
7
13
self .key = os .environ ['BROWSERSTACK_ACCESS_KEY' ] if 'BROWSERSTACK_ACCESS_KEY' in os .environ else key
@@ -18,8 +24,30 @@ def __xstr(self, key, value):
18
24
else :
19
25
return ['-' + key , value ]
20
26
27
+ def get_package_version (self ):
28
+ name = "browserstack-local"
29
+ version = 'None'
30
+ use_fallback = False
31
+ try :
32
+ temp = package_version
33
+ except NameError : # Only catch if package_version is not defined(and not other errors)
34
+ use_fallback = True
35
+
36
+ if use_fallback :
37
+ try :
38
+ version = pkg_resources .get_distribution (name ).version
39
+ except pkg_resources .DistributionNotFound :
40
+ version = 'None'
41
+ else :
42
+ try :
43
+ version = package_version (name )
44
+ except PackageNotFoundError :
45
+ version = 'None'
46
+
47
+ return version
48
+
21
49
def _generate_cmd (self ):
22
- cmd = [self .binary_path , '-d' , 'start' , '-logFile' , self .local_logfile_path , "-k" , self .key ]
50
+ cmd = [self .binary_path , '-d' , 'start' , '-logFile' , self .local_logfile_path , "-k" , self .key , '--source' , 'python:' + self . get_package_version () ]
23
51
for o in self .options .keys ():
24
52
if self .options .get (o ) is not None :
25
53
cmd = cmd + self .__xstr (o , self .options .get (o ))
@@ -51,6 +79,9 @@ def start(self, **kwargs):
51
79
if "onlyCommand" in kwargs and kwargs ["onlyCommand" ]:
52
80
return
53
81
82
+ if 'source' in self .options :
83
+ del self .options ['source' ]
84
+
54
85
self .proc = subprocess .Popen (self ._generate_cmd (), stdout = subprocess .PIPE , stderr = subprocess .PIPE )
55
86
(out , err ) = self .proc .communicate ()
56
87
0 commit comments