1#!/usr/bin/env python3 2# -*- coding: utf-8 -*- 3# Copyright 2020 The Chromium OS Authors. All rights reserved. 4# Use of this source code is governed by a BSD-style license that can be 5# found in the LICENSE file. 6 7"""Common utility functions.""" 8 9DEFAULT_OBJECT_NUMBER = 1238 10DEFAULT_BAD_OBJECT_NUMBER = 23 11OBJECTS_FILE = 'objects.txt' 12WORKING_SET_FILE = 'working_set.txt' 13 14 15def ReadWorkingSet(): 16 working_set = [] 17 with open(WORKING_SET_FILE, 'r', encoding='utf-8') as f: 18 for l in f: 19 working_set.append(int(l)) 20 return working_set 21 22 23def WriteWorkingSet(working_set): 24 with open(WORKING_SET_FILE, 'w', encoding='utf-8') as f: 25 for o in working_set: 26 f.write('{0}\n'.format(o)) 27 28 29def ReadObjectsFile(): 30 objects_file = [] 31 with open(OBJECTS_FILE, 'r', encoding='utf-8') as f: 32 for l in f: 33 objects_file.append(int(l)) 34 return objects_file 35 36 37def ReadObjectIndex(filename): 38 object_index = [] 39 with open(filename, 'r', encoding='utf-8') as f: 40 for o in f: 41 object_index.append(int(o)) 42 return object_index 43