summaryrefslogtreecommitdiffstats
path: root/scripts/dump_cache.py
blob: 3aaa780b5807e9e13f6862400d8f8a867c36ed61 (plain)
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
#!/usr/bin/env python3
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
#
# Copyright (C) 2012, 2018 Wind River Systems, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# 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.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

#
# Used for dumping the bb_cache.dat
#
import os
import sys
import argparse

# External modules
scripts_path = os.path.dirname(os.path.realpath(__file__))
lib_path = scripts_path + '/lib'
sys.path.insert(0, lib_path)

import scriptpath
# Figure out where is the bitbake/lib/bb since we need bb.siggen and bb.process
bitbakepath = scriptpath.add_bitbake_lib_path()
if not bitbakepath:
    sys.stderr.write("Unable to find bitbake by searching parent directory of this script or PATH\n")
    sys.exit(1)

from bb.cache import CoreRecipeInfo

import pickle

class DumpCache(object):
    def __init__(self):
        parser = argparse.ArgumentParser(
            description="bb_cache.dat's dumper",
            epilog="Use %(prog)s --help to get help")
        parser.add_argument("-r", "--recipe",
            help="specify the recipe, default: all recipes", action="store")
        parser.add_argument("-m", "--members",
            help = "specify the member, use comma as separator for multiple ones, default: all members", action="store", default="")
        parser.add_argument("-s", "--skip",
            help = "skip skipped recipes", action="store_true")
        parser.add_argument("cachefile",
            help = "specify bb_cache.dat", nargs = 1, action="store", default="")

        self.args = parser.parse_args()

    def main(self):
        with open(self.args.cachefile[0], "rb") as cachefile:
            pickled = pickle.Unpickler(cachefile)
            while True:
                try:
                    key = pickled.load()
                    val = pickled.load()
                except Exception:
                    break
                if isinstance(val, CoreRecipeInfo):
                    pn = val.pn

                    if self.args.recipe and self.args.recipe != pn:
                        continue

                    if self.args.skip and val.skipped:
                        continue

                    if self.args.members:
                        out = key
                        for member in self.args.members.split(','):
                            out += ": %s" % val.__dict__.get(member)
                        print("%s" % out)
                    else:
                        print("%s: %s" % (key, val.__dict__))
                elif not self.args.recipe:
                    print("%s %s" % (key, val))

if __name__ == "__main__":
    try:
        dump = DumpCache()
        ret = dump.main()
    except Exception as esc:
        ret = 1
        import traceback
        traceback.print_exc()
    sys.exit(ret)