mirror of
https://github.com/emilybache/GildedRose-Refactoring-Kata.git
synced 2026-02-09 03:31:28 +00:00
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2015-2016 Yelp
|
|
# Copyright 2019 Yelp
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Utility methods for dealing with counters (not including parsers)."""
|
|
|
|
|
|
def _format_counters(counters, indent='\t', desc='Counters'):
|
|
"""Convert a map from group -> counter name -> amount to a message
|
|
similar to that printed by the Hadoop binary, with no trailing newline.
|
|
"""
|
|
num_counters = sum(len(counter_to_amount)
|
|
for group, counter_to_amount in counters.items())
|
|
message = '%s: %d' % (desc, num_counters)
|
|
|
|
for group, group_counters in sorted(counters.items()):
|
|
if group_counters:
|
|
message += '\n%s%s' % (indent, group)
|
|
for counter, amount in sorted(group_counters.items()):
|
|
message += '\n%s%s%s=%d' % (indent, indent, counter, amount)
|
|
|
|
return message
|
|
|
|
|
|
def _pick_counters(log_interpretation):
|
|
"""Pick counters from a dictionary possibly containing
|
|
step and history interpretations."""
|
|
for log_type in 'step', 'history':
|
|
counters = log_interpretation.get(log_type, {}).get('counters')
|
|
if counters:
|
|
return counters
|
|
else:
|
|
return {}
|
|
|
|
|
|
def _sum_counters(*counters_list):
|
|
"""Combine many maps from group to counter to amount."""
|
|
result = {}
|
|
|
|
for counters in counters_list:
|
|
for group, counter_to_amount in counters.items():
|
|
for counter, amount in counter_to_amount.items():
|
|
result.setdefault(group, {})
|
|
result[group].setdefault(counter, 0)
|
|
result[group][counter] += amount
|
|
|
|
return result
|