buildroot

#!/usr/bin/env python

​

# Copyright (C) 2016 Thomas De Schampheleire <thomas.de.schampheleire@gmail.com>

​

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

​

# TODO (improvements)

# - support K,M,G size suffixes for threshold

# - output CSV file in addition to stdout reporting

​

import csv

import argparse

import sys

​

def read_file_size_csv(inputf, detail=None):

    """Extract package or file sizes from CSV file into size dictionary"""

    sizes = {}

    reader = csv.reader(inputf)

​

    header = next(reader)

    if (header[0] != 'File name' or header[1] != 'Package name' or

        header[2] != 'File size' or header[3] != 'Package size'):

        print(("Input file %s does not contain the expected header. Are you "

               "sure this file corresponds to the file-size-stats.csv "

               "file created by 'make graph-size'?") % inputf.name)

        sys.exit(1)

​

    for row in reader:

        if detail:

            sizes[row[0]] = int(row[2])

        else:

            sizes[row[1]] = int(row[3])

​

    return sizes

​

def compare_sizes(old, new):

    """Return delta/added/removed dictionaries based on two input size

    dictionaries"""

    delta = {}

    oldkeys = set(old.keys())

    newkeys = set(new.keys())

​

    # packages/files in both

    for entry in newkeys.intersection(oldkeys):

        delta[entry] = ('', new[entry] - old[entry])

    # packages/files only in new

    for entry in newkeys.difference(oldkeys):

        delta[entry] = ('added', new[entry])

    # packages/files only in old

    for entry in oldkeys.difference(newkeys):

        delta[entry] = ('removed', -old[entry])