22from browserstack .local_binary import LocalBinary
33from browserstack .bserrors import BrowserStackLocalError
44
5+ try :
6+ from importlib .metadata import version as package_version , PackageNotFoundError
7+ except :
8+ import pkg_resources
9+
10+
511class Local :
612 def __init__ (self , key = None , binary_path = None , ** kwargs ):
713 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):
1824 else :
1925 return ['-' + key , value ]
2026
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+
2149 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 () ]
2351 for o in self .options .keys ():
2452 if self .options .get (o ) is not None :
2553 cmd = cmd + self .__xstr (o , self .options .get (o ))
@@ -51,6 +79,9 @@ def start(self, **kwargs):
5179 if "onlyCommand" in kwargs and kwargs ["onlyCommand" ]:
5280 return
5381
82+ if 'source' in self .options :
83+ del self .options ['source' ]
84+
5485 self .proc = subprocess .Popen (self ._generate_cmd (), stdout = subprocess .PIPE , stderr = subprocess .PIPE )
5586 (out , err ) = self .proc .communicate ()
5687
0 commit comments