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

Unified Diff: build/android/pylib/device/device_blacklist.py

Issue 1281923003: [Android] Add --blacklist-file as a command-line option. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: fix telemetry_unittests Created 5 years, 4 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « build/android/provision_devices.py ('k') | build/android/pylib/device/device_utils.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: build/android/pylib/device/device_blacklist.py
diff --git a/build/android/pylib/device/device_blacklist.py b/build/android/pylib/device/device_blacklist.py
index a141d62b81b828f17a243dbb70adbe8772ddba30..0724654be7b9b5575eb551ce9eda7cb82ca5e095 100644
--- a/build/android/pylib/device/device_blacklist.py
+++ b/build/android/pylib/device/device_blacklist.py
@@ -7,55 +7,76 @@ import os
import threading
from pylib import constants
-_BLACKLIST_JSON = os.path.join(
+
+# TODO(jbudorick): Remove this once the blacklist is optional.
+BLACKLIST_JSON = os.path.join(
constants.DIR_SOURCE_ROOT,
os.environ.get('CHROMIUM_OUT_DIR', 'out'),
'bad_devices.json')
-# Note that this only protects against concurrent accesses to the blacklist
-# within a process.
-_blacklist_lock = threading.RLock()
+class Blacklist(object):
-def ReadBlacklist():
- """Reads the blacklist from the _BLACKLIST_JSON file.
+ def __init__(self, path):
+ self._blacklist_lock = threading.RLock()
+ self._path = path
- Returns:
- A list containing bad devices.
- """
- with _blacklist_lock:
- if not os.path.exists(_BLACKLIST_JSON):
- return []
+ def Read(self):
+ """Reads the blacklist from the blacklist file.
- with open(_BLACKLIST_JSON, 'r') as f:
- return json.load(f)
+ Returns:
+ A list containing bad devices.
+ """
+ with self._blacklist_lock:
+ if not os.path.exists(self._path):
+ return []
+ with open(self._path, 'r') as f:
+ return json.load(f)
-def WriteBlacklist(blacklist):
- """Writes the provided blacklist to the _BLACKLIST_JSON file.
+ def Write(self, blacklist):
+ """Writes the provided blacklist to the blacklist file.
- Args:
- blacklist: list of bad devices to write to the _BLACKLIST_JSON file.
- """
- with _blacklist_lock:
- with open(_BLACKLIST_JSON, 'w') as f:
- json.dump(list(set(blacklist)), f)
+ Args:
+ blacklist: list of bad devices to write to the blacklist file.
+ """
+ with self._blacklist_lock:
+ with open(self._path, 'w') as f:
+ json.dump(list(set(blacklist)), f)
+ def Extend(self, devices):
+ """Adds devices to blacklist file.
-def ExtendBlacklist(devices):
- """Adds devices to _BLACKLIST_JSON file.
+ Args:
+ devices: list of bad devices to be added to the blacklist file.
+ """
+ with self._blacklist_lock:
+ blacklist = ReadBlacklist()
+ blacklist.extend(devices)
+ WriteBlacklist(blacklist)
- Args:
- devices: list of bad devices to be added to the _BLACKLIST_JSON file.
- """
- with _blacklist_lock:
- blacklist = ReadBlacklist()
- blacklist.extend(devices)
- WriteBlacklist(blacklist)
+ def Reset(self):
+ """Erases the blacklist file if it exists."""
+ with self._blacklist_lock:
+ if os.path.exists(self._path):
+ os.remove(self._path)
+
+
+def ReadBlacklist():
+ # TODO(jbudorick): Phase out once all clients have migrated.
+ return Blacklist(BLACKLIST_JSON).Read()
+
+
+def WriteBlacklist(blacklist):
+ # TODO(jbudorick): Phase out once all clients have migrated.
+ Blacklist(BLACKLIST_JSON).Write(blacklist)
+
+
+def ExtendBlacklist(devices):
+ # TODO(jbudorick): Phase out once all clients have migrated.
+ Blacklist(BLACKLIST_JSON).Extend(devices)
def ResetBlacklist():
- """Erases the _BLACKLIST_JSON file if it exists."""
- with _blacklist_lock:
- if os.path.exists(_BLACKLIST_JSON):
- os.remove(_BLACKLIST_JSON)
+ # TODO(jbudorick): Phase out once all clients have migrated.
+ Blacklist(BLACKLIST_JSON).Reset()
« no previous file with comments | « build/android/provision_devices.py ('k') | build/android/pylib/device/device_utils.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698