oe-pkgdata-browser 9.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253
  1. #! /usr/bin/env python3
  2. import os, sys, enum, ast
  3. scripts_path = os.path.dirname(os.path.realpath(__file__))
  4. lib_path = scripts_path + '/lib'
  5. sys.path = sys.path + [lib_path]
  6. import scriptpath
  7. bitbakepath = scriptpath.add_bitbake_lib_path()
  8. if not bitbakepath:
  9. print("Unable to find bitbake by searching parent directory of this script or PATH")
  10. sys.exit(1)
  11. import bb
  12. import gi
  13. gi.require_version('Gtk', '3.0')
  14. from gi.repository import Gtk, Gdk, GObject
  15. RecipeColumns = enum.IntEnum("RecipeColumns", {"Recipe": 0})
  16. PackageColumns = enum.IntEnum("PackageColumns", {"Package": 0, "Size": 1})
  17. FileColumns = enum.IntEnum("FileColumns", {"Filename": 0, "Size": 1})
  18. import time
  19. def timeit(f):
  20. def timed(*args, **kw):
  21. ts = time.time()
  22. print ("func:%r calling" % f.__name__)
  23. result = f(*args, **kw)
  24. te = time.time()
  25. print ('func:%r args:[%r, %r] took: %2.4f sec' % \
  26. (f.__name__, args, kw, te-ts))
  27. return result
  28. return timed
  29. def human_size(nbytes):
  30. import math
  31. suffixes = ['B', 'kB', 'MB', 'GB', 'TB', 'PB']
  32. human = nbytes
  33. rank = 0
  34. if nbytes != 0:
  35. rank = int((math.log10(nbytes)) / 3)
  36. rank = min(rank, len(suffixes) - 1)
  37. human = nbytes / (1000.0 ** rank)
  38. f = ('%.2f' % human).rstrip('0').rstrip('.')
  39. return '%s %s' % (f, suffixes[rank])
  40. def load(filename, suffix=None):
  41. from configparser import ConfigParser
  42. from itertools import chain
  43. parser = ConfigParser()
  44. if suffix:
  45. parser.optionxform = lambda option: option.replace("_" + suffix, "")
  46. with open(filename) as lines:
  47. lines = chain(("[fake]",), lines)
  48. parser.read_file(lines)
  49. # TODO extract the data and put it into a real dict so we can transform some
  50. # values to ints?
  51. return parser["fake"]
  52. def find_pkgdata():
  53. import subprocess
  54. output = subprocess.check_output(("bitbake", "-e"), universal_newlines=True)
  55. for line in output.splitlines():
  56. if line.startswith("PKGDATA_DIR="):
  57. return line.split("=", 1)[1].strip("\'\"")
  58. # TODO exception or something
  59. return None
  60. def packages_in_recipe(pkgdata, recipe):
  61. """
  62. Load the recipe pkgdata to determine the list of runtime packages.
  63. """
  64. data = load(os.path.join(pkgdata, recipe))
  65. packages = data["PACKAGES"].split()
  66. return packages
  67. def load_runtime_package(pkgdata, package):
  68. return load(os.path.join(pkgdata, "runtime", package), suffix=package)
  69. def recipe_from_package(pkgdata, package):
  70. data = load(os.path.join(pkgdata, "runtime", package), suffix=package)
  71. return data["PN"]
  72. def summary(data):
  73. s = ""
  74. s += "{0[PKG]} {0[PKGV]}-{0[PKGR]}\n{0[LICENSE]}\n{0[SUMMARY]}\n".format(data)
  75. return s
  76. class PkgUi():
  77. def __init__(self, pkgdata):
  78. self.pkgdata = pkgdata
  79. self.current_recipe = None
  80. self.recipe_iters = {}
  81. self.package_iters = {}
  82. builder = Gtk.Builder()
  83. builder.add_from_file(os.path.join(os.path.dirname(__file__), "oe-pkgdata-browser.glade"))
  84. self.window = builder.get_object("window")
  85. self.window.connect("delete-event", Gtk.main_quit)
  86. self.recipe_store = builder.get_object("recipe_store")
  87. self.recipe_view = builder.get_object("recipe_view")
  88. self.package_store = builder.get_object("package_store")
  89. self.package_view = builder.get_object("package_view")
  90. # Somehow resizable does not get set via builder xml
  91. package_name_column = builder.get_object("package_name_column")
  92. package_name_column.set_resizable(True)
  93. file_name_column = builder.get_object("file_name_column")
  94. file_name_column.set_resizable(True)
  95. self.recipe_view.get_selection().connect("changed", self.on_recipe_changed)
  96. self.package_view.get_selection().connect("changed", self.on_package_changed)
  97. self.package_store.set_sort_column_id(PackageColumns.Package, Gtk.SortType.ASCENDING)
  98. builder.get_object("package_size_column").set_cell_data_func(builder.get_object("package_size_cell"), lambda column, cell, model, iter, data: cell.set_property("text", human_size(model[iter][PackageColumns.Size])))
  99. self.label = builder.get_object("label1")
  100. self.depends_label = builder.get_object("depends_label")
  101. self.recommends_label = builder.get_object("recommends_label")
  102. self.suggests_label = builder.get_object("suggests_label")
  103. self.provides_label = builder.get_object("provides_label")
  104. self.depends_label.connect("activate-link", self.on_link_activate)
  105. self.recommends_label.connect("activate-link", self.on_link_activate)
  106. self.suggests_label.connect("activate-link", self.on_link_activate)
  107. self.file_store = builder.get_object("file_store")
  108. self.file_store.set_sort_column_id(FileColumns.Filename, Gtk.SortType.ASCENDING)
  109. builder.get_object("file_size_column").set_cell_data_func(builder.get_object("file_size_cell"), lambda column, cell, model, iter, data: cell.set_property("text", human_size(model[iter][FileColumns.Size])))
  110. self.files_view = builder.get_object("files_scrollview")
  111. self.files_label = builder.get_object("files_label")
  112. self.load_recipes()
  113. self.recipe_view.set_cursor(Gtk.TreePath.new_first())
  114. self.window.show()
  115. def on_link_activate(self, label, url_string):
  116. from urllib.parse import urlparse
  117. url = urlparse(url_string)
  118. if url.scheme == "package":
  119. package = url.path
  120. recipe = recipe_from_package(self.pkgdata, package)
  121. it = self.recipe_iters[recipe]
  122. path = self.recipe_store.get_path(it)
  123. self.recipe_view.set_cursor(path)
  124. self.recipe_view.scroll_to_cell(path)
  125. self.on_recipe_changed(self.recipe_view.get_selection())
  126. it = self.package_iters[package]
  127. path = self.package_store.get_path(it)
  128. self.package_view.set_cursor(path)
  129. self.package_view.scroll_to_cell(path)
  130. return True
  131. else:
  132. return False
  133. def on_recipe_changed(self, selection):
  134. self.package_store.clear()
  135. self.package_iters = {}
  136. (model, it) = selection.get_selected()
  137. if not it:
  138. return
  139. recipe = model[it][RecipeColumns.Recipe]
  140. packages = packages_in_recipe(self.pkgdata, recipe)
  141. for package in packages:
  142. # TODO also show PKG after debian-renaming?
  143. data = load_runtime_package(self.pkgdata, package)
  144. # TODO stash data to avoid reading in on_package_changed
  145. self.package_iters[package] = self.package_store.append([package, int(data["PKGSIZE"])])
  146. package = recipe if recipe in packages else sorted(packages)[0]
  147. path = self.package_store.get_path(self.package_iters[package])
  148. self.package_view.set_cursor(path)
  149. self.package_view.scroll_to_cell(path)
  150. def on_package_changed(self, selection):
  151. self.label.set_text("")
  152. self.file_store.clear()
  153. self.depends_label.hide()
  154. self.recommends_label.hide()
  155. self.suggests_label.hide()
  156. self.provides_label.hide()
  157. self.files_view.hide()
  158. self.files_label.hide()
  159. (model, it) = selection.get_selected()
  160. if it is None:
  161. return
  162. package = model[it][PackageColumns.Package]
  163. data = load_runtime_package(self.pkgdata, package)
  164. self.label.set_text(summary(data))
  165. files = ast.literal_eval(data["FILES_INFO"])
  166. if files:
  167. self.files_label.set_text("{0} files take {1}.".format(len(files), human_size(int(data["PKGSIZE"]))))
  168. self.files_view.show()
  169. for filename, size in files.items():
  170. self.file_store.append([filename, size])
  171. else:
  172. self.files_view.hide()
  173. self.files_label.set_text("This package has no files.")
  174. self.files_label.show()
  175. def update_deps(field, prefix, label, clickable=True):
  176. if field in data:
  177. l = []
  178. for name, version in bb.utils.explode_dep_versions2(data[field]).items():
  179. if clickable:
  180. l.append("<a href='package:{0}'>{0}</a> {1}".format(name, " ".join(version)).strip())
  181. else:
  182. l.append("{0} {1}".format(name, " ".join(version)).strip())
  183. label.set_markup(prefix + ", ".join(l))
  184. label.show()
  185. else:
  186. label.hide()
  187. update_deps("RDEPENDS", "Depends: ", self.depends_label)
  188. update_deps("RRECOMMENDS", "Recommends: ", self.recommends_label)
  189. update_deps("RSUGGESTS", "Suggests: ", self.suggests_label)
  190. update_deps("RPROVIDES", "Provides: ", self.provides_label, clickable=False)
  191. def load_recipes(self):
  192. for recipe in sorted(os.listdir(pkgdata)):
  193. if os.path.isfile(os.path.join(pkgdata, recipe)):
  194. self.recipe_iters[recipe] = self.recipe_store.append([recipe])
  195. if __name__ == "__main__":
  196. import argparse
  197. parser = argparse.ArgumentParser(description='pkgdata browser')
  198. parser.add_argument('-p', '--pkgdata', help="Optional location of pkgdata")
  199. args = parser.parse_args()
  200. pkgdata = args.pkgdata if args.pkgdata else find_pkgdata()
  201. # TODO assert pkgdata is a directory
  202. window = PkgUi(pkgdata)
  203. Gtk.main()