aboutsummaryrefslogtreecommitdiff
blob: bc892ca85e02d9f34c0e7c36ee47c1b28f24bb1f (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
96
97
98
99
100
101
102
103
104
105
106
107
108
#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
    requirements.py
    ~~~~~~~~~~~~~~~
    
    a simple script that creates a JSON file with the list of dependencies
    that are not from octave-forge package (SystemRequirements and
    BuildRequires). It writes the JSON content to the stdout.
    
    :copyright: (c) 2010 by Rafael Goncalves Martins
    :license: GPL-2, see LICENSE for more details.
"""

import json
import sys
import os
import portage

from _emerge.actions import load_emerge_config
from _emerge.search import search

current_dir = os.path.dirname(os.path.realpath(__file__))
if os.path.exists(os.path.join(current_dir, '..', 'g_octave')):
    sys.path.insert(0, os.path.join(current_dir, '..'))

from g_octave import description, description_tree, exception

def main(argv):
    
    if len(argv) <= 1:
        print >> sys.stderr, 'one argument required: the json file.'
        return 1
    
    # init portage stuff
    settings, trees, mtimedb = load_emerge_config()
    root_config = trees[settings['ROOT']]['root_config']
    s = search(root_config, False, False, False, False, False)
    
    desc_tree = description_tree.DescriptionTree(parse_sysreq = False)
    
    dependencies = []
    
    for pkg in desc_tree.packages():
        try:
            desc = desc_tree[pkg]
        except exception.DescriptionTreeException, err:
            print >> sys.stderr, 'DescriptionTree error: %s' % err
            return 1
        
        deps = []
        
        if desc.systemrequirements is not None:
            deps += [i.strip() for i in desc.systemrequirements.split(',')]
        
        if desc.buildrequires is not None:
            deps += [i.strip() for i in desc.buildrequires.split(',')]
        
        dep_aux = []
        for dep in deps:
            match = description.re_depends.match(dep)
            if match is not None:
                my_match = match.group(1).split('-')[0]
                if my_match not in dep_aux:
                    dependencies.append((match.group(1), my_match))
                    dep_aux.append(my_match)
    
    json_dict = dict(dependencies=dict())
    
    try:
        with open(argv[1], 'r') as fp:
            json_dict = json.load(fp)
    except:
        pass
    
    for dep_name, dep in dependencies:
        s.execute(dep)
        print dep_name
        temp = []
        for i in range(len(s.matches['pkg'])):
            print '    %i: %s' % (i, s.matches['pkg'][i][0])
            temp.append(s.matches['pkg'][i][0])
        
        if dep_name in json_dict['dependencies']:
            select = raw_input('Select a package [%s]: ' % \
                json_dict['dependencies'][dep_name])
        else:
            select = raw_input('Select a package: ')
        try:
            json_dict['dependencies'][dep_name] = temp[int(select)]
        except:
            if select != '' or dep_name not in json_dict['dependencies']:
                json_dict['dependencies'][dep_name] = select
        print 'Selected: %s' % json_dict['dependencies'][dep_name]
        print 
    
    try:
        with open(argv[1], 'w') as fp:
            json.dump(json_dict, fp, sort_keys=True, indent=4)
    except:
        print >> sys.stderr, 'failed to save the json file.'
        return 1
    
    return 0

if __name__ == '__main__':
    sys.exit(main(sys.argv))