tic/parser/view_parser.py

   1 import logging
   2 from operator import itemgetter
   3 from tic.utils.rpmmisc import meetRequireVersion
   4
   5
   6 def make_view_data(pkg_group):
   7     logger = logging.getLogger(__name__)
   8
   9     def _select_provide_rpm(capability, require):
  10         provide_list = []
  11         if require.get('ver') is not None:
  12             for provide in capability:
  13                 ver_data = provide['data']
  14                 if not ver_data.get('ver'):
  15                     ver_data = pkg_dict.get(provide['name']).get('version')
  16                 if meetRequireVersion(require, ver_data):
  17                     provide_list.append(provide)
  18         else:
  19             provide_list = capability
  20
  21         result = []
  22         for provide in provide_list:
  23             result.append(pkg_dict[provide['name']])
  24
  25         return result
  26
  27     def set_meta_require(meta_info):
  28         meta_nodes = meta_info.get('nodes')
  29         for child_meta in meta_nodes:
  30             set_meta_require(child_meta)
  31
  32         duplicate = set()
  33         pkg_info = pkg_dict[meta_info['metaname']]
  34         for dep_tag in ['requires', 'recommends', 'suggests']:
  35             if pkg_info.get(dep_tag):
  36                 for req in pkg_info.get(dep_tag):
  37                     targets = []
  38                     if req['name'] in provides:
  39                         targets = _select_provide_rpm(provides[req['name']], req)
  40                     elif req['name'] in files:
  41                         for fname in files[req['name']]:
  42                             targets.append(pkg_dict.get(fname))
  43                     elif req['name'] in pkg_dict:
  44                         targets.append(pkg_dict.get(req['name']))
  45                     else:
  46                         # the dependent package does not exist.
  47                         pass
  48
  49                     for pkg in targets:
  50                         # The meta-pkg of the other group are excluded.
  51                         if not pkg.get('meta') and pkg['name'] not in duplicate:
  52                             refer_count[pkg['id']] += 1
  53                             duplicate.add(pkg['name'])
  54                             meta_nodes.append(make_node(pkg))
  55         meta_nodes.sort()
  56     def make_node(pkg_info):
  57         #return dict(id=pkg_info['id'], text=pkg_info['name'], nodes=[])
  58         return dict(text=pkg_info['name'], nodes=[])
  59     def make_meta_node(pkgname, viewtext):
  60         return dict(text=viewtext, metaname=pkgname, nodes=[])
  61
  62     # view_data for tree view on web-ui
  63     view_data = []
  64     # temporary dictionary for referencing meta
  65     view_ref = {}
  66
  67     if not pkg_group:
  68         return view_data
  69
  70     pkg_dict = pkg_group['pkg_dict']
  71     provides = pkg_group['provides']
  72     files = pkg_group['files']
  73     refer_count = [0] * len(pkg_dict)
  74
  75     # sort meta-pkg by pkt_name in ascending order
  76     meta_info = pkg_group.get('meta_info')
  77     meta_info['root'] = sorted(meta_info['root'], key=itemgetter(0))
  78     meta_info['sub1'] = sorted(meta_info['sub1'], key=itemgetter(0))
  79     meta_info['sub2'] = sorted(meta_info['sub2'], key=itemgetter(0))
  80
  81     # make tree of meta
  82     for root in meta_info['root']:
  83         root_node = make_meta_node(root[0], root[1])
  84         view_ref[root[0]] = root_node
  85         view_data.append(root_node)
  86
  87     for sub1 in meta_info['sub1']:
  88         sub1_node = make_meta_node(sub1[0], sub1[2])
  89         view_ref[sub1[0]] = sub1_node
  90         # search root
  91         if sub1[1] in view_ref:
  92             # add to root node
  93             view_ref[sub1[1]]['nodes'].append(sub1_node)
  94         else:
  95             # If root-meta does not exist, sub1 is added to top-level
  96             view_data.append(sub1_node)
  97
  98     for sub2 in meta_info['sub2']:
  99         sub2_node = make_meta_node(sub2[0], sub2[3])
 100         view_ref[sub2[0]] = sub2_node
 101         # search sub1
 102         if sub2[2] in view_ref:
 103             view_ref[sub2[2]]['nodes'].append(sub2_node)
 104         # search root
 105         elif sub2[1] in view_ref:
 106             # If sub1-meta does not exist, sub2 is added to child of root-meta
 107             view_ref[sub2[1]]['nodes'].append(sub2_node)
 108         else:
 109             # If root-meta/sub1-meta does not exist, sub1 is added to top-level
 110             view_data.append(sub2_node)
 111
 112     # configure meta dependency tree (requires)
 113     for meta_pkg in view_data:
 114         set_meta_require(meta_pkg)
 115
 116     #The remaining rpms are grouped into a MISC tree
 117     misc_info = {}
 118     misc_info['text'] = 'ETC'
 119     misc_info['nodes'] = []
 120     for k, v in pkg_dict.iteritems():
 121         # Pkg is not referenced from Meta-pkg
 122         if not v.get('meta') and refer_count[v['id']] == 0:
 123             misc_info['nodes'].append(make_node(v))
 124
 125     view_data.append(misc_info)
 126     logger.info('meta: %d, misc: %d', len(view_ref), len(misc_info['nodes']))
 127     return view_data