OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2015 The Swarming Authors. All rights reserved. | 2 # Copyright 2015 The Swarming Authors. All rights reserved. |
3 # Use of this source code is governed by the Apache v2.0 license that can be | 3 # Use of this source code is governed by the Apache v2.0 license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 """Starts local Swarming and Isolate servers.""" | 6 """Starts local Swarming and Isolate servers.""" |
7 | 7 |
8 import argparse | 8 import argparse |
9 import os | 9 import os |
10 import shutil | 10 import shutil |
11 import sys | 11 import sys |
12 import tempfile | 12 import tempfile |
13 | 13 |
14 | 14 |
15 APP_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | 15 APP_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
| 16 CLIENT_DIR = os.path.join(os.path.dirname(os.path.dirname(APP_DIR)), 'client') |
16 sys.path.insert(0, APP_DIR) | 17 sys.path.insert(0, APP_DIR) |
17 | 18 |
| 19 sys.path.insert(0, CLIENT_DIR) |
| 20 from third_party.depot_tools import fix_encoding |
| 21 sys.path.pop(0) |
| 22 |
18 import test_env | 23 import test_env |
19 test_env.setup_test_env() | 24 test_env.setup_test_env() |
20 | 25 |
21 from tool_support import local_app | 26 from tool_support import local_app |
22 | 27 |
23 | 28 |
24 class LocalServers(object): | 29 class LocalServers(object): |
25 """Local Swarming and Isolate servers.""" | 30 """Local Swarming and Isolate servers.""" |
26 def __init__(self, listen_all): | 31 def __init__(self, listen_all): |
27 self._isolate_server = None | 32 self._isolate_server = None |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
81 self._swarming_server.wait() | 86 self._swarming_server.wait() |
82 | 87 |
83 def dump_log(self): | 88 def dump_log(self): |
84 if self._isolate_server: | 89 if self._isolate_server: |
85 self._isolate_server.dump_log() | 90 self._isolate_server.dump_log() |
86 if self._swarming_server: | 91 if self._swarming_server: |
87 self._swarming_server.dump_log() | 92 self._swarming_server.dump_log() |
88 | 93 |
89 | 94 |
90 def main(): | 95 def main(): |
| 96 fix_encoding.fix_encoding() |
91 parser = argparse.ArgumentParser(description=sys.modules[__name__].__doc__) | 97 parser = argparse.ArgumentParser(description=sys.modules[__name__].__doc__) |
92 parser.add_argument('-a', '--all', action='store_true') | 98 parser.add_argument('-a', '--all', action='store_true') |
93 args = parser.parse_args() | 99 args = parser.parse_args() |
94 servers = LocalServers(args.all) | 100 servers = LocalServers(args.all) |
95 try: | 101 try: |
96 servers.start() | 102 servers.start() |
97 print('Swarming: %s' % servers.swarming_server.url) | 103 print('Swarming: %s' % servers.swarming_server.url) |
98 print('Isolate : %s' % servers.isolate_server.url) | 104 print('Isolate : %s' % servers.isolate_server.url) |
99 servers.wait() | 105 servers.wait() |
100 servers.dump_log() | 106 servers.dump_log() |
101 except KeyboardInterrupt: | 107 except KeyboardInterrupt: |
102 print >> sys.stderr, '<Ctrl-C> received; stopping servers' | 108 print >> sys.stderr, '<Ctrl-C> received; stopping servers' |
103 finally: | 109 finally: |
104 exit_code = servers.stop(False) | 110 exit_code = servers.stop(False) |
105 return exit_code | 111 return exit_code |
106 | 112 |
107 | 113 |
108 if __name__ == '__main__': | 114 if __name__ == '__main__': |
109 sys.exit(main()) | 115 sys.exit(main()) |
OLD | NEW |