50 строки
1.7 KiB
Python
50 строки
1.7 KiB
Python
|
# Copyright (c) 2011 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.
|
||
|
|
|
||
|
|
|
||
|
|
# Location where chrome reads command line flags from
|
||
|
|
CHROME_COMMAND_FILE = '/data/local/chrome-command-line'
|
||
|
|
|
||
|
|
|
||
|
|
class FlagChanger(object):
|
||
|
|
"""Temporarily changes the flags Chrome runs with."""
|
||
|
|
|
||
|
|
def __init__(self, android_cmd):
|
||
|
|
self._android_cmd = android_cmd
|
||
|
|
self._old_flags = None
|
||
|
|
|
||
|
|
def Set(self, flags, append=False):
|
||
|
|
"""Sets the command line flags used when chrome is started.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
flags: A list of flags to set, eg. ['--single-process'].
|
||
|
|
append: Whether to append to existing flags or overwrite them.
|
||
|
|
"""
|
||
|
|
if flags:
|
||
|
|
assert flags[0] != 'chrome'
|
||
|
|
|
||
|
|
if not self._old_flags:
|
||
|
|
self._old_flags = self._android_cmd.GetFileContents(CHROME_COMMAND_FILE)
|
||
|
|
if self._old_flags:
|
||
|
|
self._old_flags = self._old_flags[0].strip()
|
||
|
|
|
||
|
|
if append and self._old_flags:
|
||
|
|
# Avoid appending flags that are already present.
|
||
|
|
new_flags = filter(lambda flag: self._old_flags.find(flag) == -1, flags)
|
||
|
|
self._android_cmd.SetFileContents(CHROME_COMMAND_FILE,
|
||
|
|
self._old_flags + ' ' +
|
||
|
|
' '.join(new_flags))
|
||
|
|
else:
|
||
|
|
self._android_cmd.SetFileContents(CHROME_COMMAND_FILE,
|
||
|
|
'chrome ' + ' '.join(flags))
|
||
|
|
|
||
|
|
def Restore(self):
|
||
|
|
"""Restores the flags to their original state."""
|
||
|
|
if self._old_flags == None:
|
||
|
|
return # Set() was never called.
|
||
|
|
elif self._old_flags:
|
||
|
|
self._android_cmd.SetFileContents(CHROME_COMMAND_FILE, self._old_flags)
|
||
|
|
else:
|
||
|
|
self._android_cmd.RunShellCommand('rm ' + CHROME_COMMAND_FILE)
|