Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2282)

Side by Side Diff: webrtc/tools/testing/utils.py

Issue 2965593002: Move webrtc/{tools => rtc_tools} (Closed)
Patch Set: Adding back root changes Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « webrtc/tools/testing/setup_apprtc.py ('k') | webrtc/tools/video_analysis.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 #!/usr/bin/env python
2 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
3 #
4 # Use of this source code is governed by a BSD-style license
5 # that can be found in the LICENSE file in the root of the source
6 # tree. An additional intellectual property rights grant can be found
7 # in the file PATENTS. All contributing project authors may
8 # be found in the AUTHORS file in the root of the source tree.
9
10 """Utilities for all our deps-management stuff."""
11
12 import os
13 import shutil
14 import sys
15 import subprocess
16 import tarfile
17 import time
18 import zipfile
19
20
21 def RunSubprocessWithRetry(cmd):
22 """Invokes the subprocess and backs off exponentially on fail."""
23 for i in range(5):
24 try:
25 subprocess.check_call(cmd)
26 return
27 except subprocess.CalledProcessError as exception:
28 backoff = pow(2, i)
29 print 'Got %s, retrying in %d seconds...' % (exception, backoff)
30 time.sleep(backoff)
31
32 print 'Giving up.'
33 raise exception
34
35
36 def DownloadFilesFromGoogleStorage(path, auto_platform=True):
37 print 'Downloading files in %s...' % path
38
39 extension = 'bat' if 'win32' in sys.platform else 'py'
40 cmd = ['download_from_google_storage.%s' % extension,
41 '--bucket=chromium-webrtc-resources',
42 '--directory', path]
43 if auto_platform:
44 cmd += ['--auto_platform', '--recursive']
45 subprocess.check_call(cmd)
46
47
48 # Code partially copied from
49 # https://cs.chromium.org#chromium/build/scripts/common/chromium_utils.py
50 def RemoveDirectory(*path):
51 """Recursively removes a directory, even if it's marked read-only.
52
53 Remove the directory located at *path, if it exists.
54
55 shutil.rmtree() doesn't work on Windows if any of the files or directories
56 are read-only, which svn repositories and some .svn files are. We need to
57 be able to force the files to be writable (i.e., deletable) as we traverse
58 the tree.
59
60 Even with all this, Windows still sometimes fails to delete a file, citing
61 a permission error (maybe something to do with antivirus scans or disk
62 indexing). The best suggestion any of the user forums had was to wait a
63 bit and try again, so we do that too. It's hand-waving, but sometimes it
64 works. :/
65 """
66 file_path = os.path.join(*path)
67 print 'Deleting `{}`.'.format(file_path)
68 if not os.path.exists(file_path):
69 print '`{}` does not exist.'.format(file_path)
70 return
71
72 if sys.platform == 'win32':
73 # Give up and use cmd.exe's rd command.
74 file_path = os.path.normcase(file_path)
75 for _ in xrange(3):
76 print 'RemoveDirectory running %s' % (' '.join(
77 ['cmd.exe', '/c', 'rd', '/q', '/s', file_path]))
78 if not subprocess.call(['cmd.exe', '/c', 'rd', '/q', '/s', file_path]):
79 break
80 print ' Failed'
81 time.sleep(3)
82 return
83 else:
84 shutil.rmtree(file_path, ignore_errors=True)
85
86
87 def UnpackArchiveTo(archive_path, output_dir):
88 extension = os.path.splitext(archive_path)[1]
89 if extension == '.zip':
90 _UnzipArchiveTo(archive_path, output_dir)
91 else:
92 _UntarArchiveTo(archive_path, output_dir)
93
94
95 def _UnzipArchiveTo(archive_path, output_dir):
96 print 'Unzipping {} in {}.'.format(archive_path, output_dir)
97 zip_file = zipfile.ZipFile(archive_path)
98 try:
99 zip_file.extractall(output_dir)
100 finally:
101 zip_file.close()
102
103
104 def _UntarArchiveTo(archive_path, output_dir):
105 print 'Untarring {} in {}.'.format(archive_path, output_dir)
106 tar_file = tarfile.open(archive_path, 'r:gz')
107 try:
108 tar_file.extractall(output_dir)
109 finally:
110 tar_file.close()
111
112
113 def GetPlatform():
114 if sys.platform.startswith('win'):
115 return 'win'
116 if sys.platform.startswith('linux'):
117 return 'linux'
118 if sys.platform.startswith('darwin'):
119 return 'mac'
120 raise Exception("Can't run on platform %s." % sys.platform)
OLDNEW
« no previous file with comments | « webrtc/tools/testing/setup_apprtc.py ('k') | webrtc/tools/video_analysis.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698