Skip to content

Commit

Permalink
Add 'section' from package meta to graph JSON
Browse files Browse the repository at this point in the history
  • Loading branch information
remram44 committed Aug 16, 2017
1 parent 28efd61 commit 347dad2
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions reprounzip/reprounzip/unpackers/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -187,11 +187,12 @@ def json(self, process_map):
class Package(object):
"""Structure representing a system package.
"""
def __init__(self, name, version=None):
def __init__(self, name, version=None, section=None):
self.id = None
self.name = name
self.version = version
self.files = set()
self.section = section

def dot(self, fp, level_pkgs):
assert self.id is not None
Expand Down Expand Up @@ -237,7 +238,7 @@ def json(self, level_pkgs):
else:
assert False
return {'name': self.name, 'version': self.version or None,
'files': files}
'section': self.section, 'files': files}


def parse_levels(level_pkgs, level_processes, level_other_files):
Expand Down Expand Up @@ -546,7 +547,8 @@ def filefilter(path):
if pkg is not None:
package = packages.get(pkg.name)
if package is None:
package = Package(pkg.name, pkg.version)
package = Package(pkg.name, pkg.version,
pkg.meta.get('section', None))
packages[pkg.name] = package
package.files.add(fi)
package_map[fi] = package
Expand Down Expand Up @@ -698,7 +700,6 @@ def graph_json(target, runs, packages, other_files, package_map, edges,
endp_file = unicode_(fi)
if mode is None:
endp_prog['reads'].append(endp_file)
# TODO: argv?
elif mode & FILE_WRITE:
endp_prog['writes'].append(endp_file)
elif mode & FILE_READ:
Expand Down

0 comments on commit 347dad2

Please sign in to comment.