common.py revision bccc576843ee27fefda5094afc88489f4cbff3ff
1#!/usr/bin/python2
2"""Common utility functions."""
3
4DEFAULT_OBJECT_NUMBER = 1238
5DEFAULT_BAD_OBJECT_NUMBER = 23
6OBJECTS_FILE = 'objects.txt'
7WORKING_SET_FILE = 'working_set.txt'
8
9installed = True
10
11
12def ReadWorkingSet():
13  working_set = []
14  f = open(WORKING_SET_FILE, 'r')
15  for l in f:
16    working_set.append(int(l))
17  f.close()
18  return working_set
19
20
21def WriteWorkingSet(working_set):
22  f = open(WORKING_SET_FILE, 'w')
23  for o in working_set:
24    f.write('{0}\n'.format(o))
25  f.close()
26
27
28def ReadObjectsFile():
29  objects_file = []
30  f = open(OBJECTS_FILE, 'r')
31  for l in f:
32    objects_file.append(int(l))
33  f.close()
34  return objects_file
35
36
37def ReadObjectIndex(filename):
38  object_index = []
39  f = open(filename, 'r')
40  for o in f:
41    object_index.append(int(o))
42  f.close()
43  return object_index
44