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
|
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import ctypes
import json
class ChromeDriver(object):
"""Starts and controls a single Chrome instance on this machine."""
def __init__(self, lib_path):
self._lib = ctypes.CDLL(lib_path)
self._session_id = self._ExecuteCommand('newSession')
def _ExecuteCommand(self, name, params={}, session_id=''):
cmd = {
'name': name,
'parameters': params,
'sessionId': session_id
}
cmd_json = json.dumps(cmd)
response_data = ctypes.c_char_p()
response_size = ctypes.c_uint()
self._lib.ExecuteCommand(
ctypes.c_char_p(cmd_json),
ctypes.c_uint(len(cmd_json)),
ctypes.byref(response_data),
ctypes.byref(response_size))
response_json = ctypes.string_at(response_data, response_size.value)
self._lib.Free(response_data)
return json.loads(response_json)['value']
def _ExecuteSessionCommand(self, name, params={}):
return self._ExecuteCommand(name, params, self._session_id)
def Quit(self):
"""Quits the browser and ends the session."""
self._ExecuteSessionCommand('quit')
|
