forked from browserstack/browserstack-local-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlocal_binary.py
More file actions
123 lines (105 loc) · 4.07 KB
/
local_binary.py
File metadata and controls
123 lines (105 loc) · 4.07 KB
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
import platform, os, sys, stat, tempfile, re, subprocess
from browserstack.bserrors import BrowserStackLocalError
try:
from urllib.request import urlopen
except ImportError:
from urllib2 import urlopen
class LocalBinary:
def __init__(self):
is_64bits = sys.maxsize > 2**32
self.is_windows = False
osname = platform.system()
if osname == 'Darwin':
self.http_path = "https://www.browserstack.com/local-testing/downloads/binaries/BrowserStackLocal-darwin-x64"
elif osname == 'Linux':
if self.is_alpine():
self.http_path = "https://www.browserstack.com/local-testing/downloads/binaries/BrowserStackLocal-alpine"
else:
if is_64bits:
self.http_path = "https://www.browserstack.com/local-testing/downloads/binaries/BrowserStackLocal-linux-x64"
else:
self.http_path = "https://www.browserstack.com/local-testing/downloads/binaries/BrowserStackLocal-linux-ia32"
else:
self.is_windows = True
self.http_path = "https://www.browserstack.com/local-testing/downloads/binaries/BrowserStackLocal.exe"
self.ordered_paths = [
os.path.join(os.path.expanduser('~'), '.browserstack'),
os.getcwd(),
tempfile.gettempdir()
]
self.path_index = 0
def is_alpine(self):
grepOutput = subprocess.run("grep -w NAME /etc/os-release", capture_output=True, shell=True)
if grepOutput.stdout.decode('utf-8').find('Alpine') > -1:
return True
return False
def __make_path(self, dest_path):
try:
if not os.path.exists(dest_path):
os.makedirs(dest_path)
return True
except:
return False
def __available_dir(self):
while self.path_index < len(self.ordered_paths):
if self.__make_path(self.ordered_paths[self.path_index]):
final_path = self.ordered_paths[self.path_index]
self.path_index += 1
return final_path
else:
self.path_index += 1
raise BrowserStackLocalError('Error trying to download BrowserStack Local binary')
def download(self, chunk_size=8192, progress_hook=None):
response = urlopen(self.http_path)
try:
total_size = int(response.info().getheader('Content-Length').strip())
except:
total_size = int(response.info().get_all('Content-Length')[0].strip())
bytes_so_far = 0
dest_parent_dir = self.__available_dir()
dest_binary_name = 'BrowserStackLocal'
if self.is_windows:
dest_binary_name += '.exe'
with open(os.path.join(dest_parent_dir, dest_binary_name), 'wb') as local_file:
while True:
chunk = response.read(chunk_size)
bytes_so_far += len(chunk)
if not chunk:
break
if progress_hook:
progress_hook(bytes_so_far, chunk_size, total_size)
try:
local_file.write(chunk)
except:
return self.download(chunk_size, progress_hook)
final_path = os.path.join(dest_parent_dir, dest_binary_name)
st = os.stat(final_path)
os.chmod(final_path, st.st_mode | stat.S_IXUSR)
return final_path
def __verify_binary(self,path):
try:
binary_response = subprocess.check_output([path,"--version"]).decode("utf-8")
pattern = re.compile("BrowserStack Local version \d+\.\d+")
return bool(pattern.match(binary_response))
except:
return False
def get_binary(self):
dest_parent_dir = os.path.join(os.path.expanduser('~'), '.browserstack')
if not os.path.exists(dest_parent_dir):
os.makedirs(dest_parent_dir)
binary_name = 'BrowserStackLocal.exe' if self.is_windows else 'BrowserStackLocal'
bsfiles = [f for f in os.listdir(dest_parent_dir) if f == binary_name]
if len(bsfiles) == 0:
binary_path = self.download()
else:
binary_path = os.path.join(dest_parent_dir, bsfiles[0])
valid_binary = self.__verify_binary(binary_path)
if valid_binary:
return binary_path
else:
binary_path = self.download()
valid_binary = self.__verify_binary(binary_path)
if valid_binary:
return binary_path
else:
raise BrowserStackLocalError('BrowserStack Local binary is corrupt')