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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#!/usr/bin/env python3
# Copyright 2018 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Prints all histogram names."""
from __future__ import print_function
import argparse
import os
import subprocess
import sys
import tempfile
import io
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'common'))
import path_util
import extract_histograms
import histogram_paths
import merge_xml
def get_names(xml_files):
"""Returns all histogram names generated from a list of xml files.
Args:
xml_files: A list of open file objects containing histogram definitions.
Returns:
A tuple of (names, obsolete names), where the obsolete names is a subset of
all names.
"""
doc = merge_xml.MergeFiles(files=xml_files)
histograms, had_errors = extract_histograms.ExtractHistogramsFromDom(doc)
if had_errors:
raise ValueError("Error parsing inputs.")
names = set(extract_histograms.ExtractNames(histograms))
obsolete_names = set(extract_histograms.ExtractObsoleteNames(histograms))
return (names, obsolete_names)
def histogram_xml_files():
return [open(f, encoding="utf-8") for f in histogram_paths.ALL_XMLS]
def get_diff(revision):
"""Returns the added / removed histogram names relative to git revision
Args:
revision: A git revision as described in
https://git-scm.com/docs/gitrevisions
Returns:
A tuple of (added names, removed names, obsoleted names), where each entry
is sorted in ascending order.
"""
def get_file_at_revision(path):
"""Returns a file-like object containing |path|'s content at |revision|"""
obj = "%s:%s" % (revision, path)
contents = subprocess.check_output(
("git", "cat-file", "--textconv", obj)).decode()
# Just store the contents in memory. histograms.xml is big, but it isn't
# _that_ big.
return io.StringIO(contents)
prev_files = []
for p in histogram_paths.ALL_XMLS_RELATIVE:
try:
prev_files.append(get_file_at_revision(p))
except subprocess.CalledProcessError:
# Paths might not exist in the provided revision.
continue
current_histogram_names, current_obsolete_names = get_names(
histogram_xml_files())
prev_histogram_names, prev_obsolete_names = get_names(prev_files)
added_names = sorted(list(current_histogram_names - prev_histogram_names))
removed_names = sorted(list(prev_histogram_names - current_histogram_names))
obsoleted_names = sorted(list(current_obsolete_names - prev_obsolete_names))
return (added_names, removed_names, obsoleted_names)
def print_diff_names(revision):
added_names, removed_names, obsoleted_names = get_diff(revision)
print("%d histograms added:" % len(added_names))
for name in added_names:
print(name)
print("%d histograms removed:" % len(removed_names))
for name in removed_names:
print(name)
print("%d histograms obsoleted:" % len(obsoleted_names))
for name in obsoleted_names:
print(name)
def main(argv):
parser = argparse.ArgumentParser(description='Print histogram names.')
parser.add_argument('--diff',
type=str,
help='Git revision to diff against (e.g. HEAD~)')
args = parser.parse_args(argv[1:])
if args.diff is not None:
print_diff_names(args.diff)
else:
name_set, _ = get_names(histogram_xml_files())
for name in sorted(list(name_set)):
print(name)
if __name__ == '__main__':
main(sys.argv)
|
