summaryrefslogtreecommitdiffstats
path: root/git_taskrepo/sub_commands/cmd_list_tasks.py
blob: 0c311b0e5d5627dc517c1909600ff28ee66deb15 (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

# -*- coding: utf-8 -*-

import xml.dom.minidom
from git_taskrepo.command import Command

class List_Tasks(Command):
    """List Tasks"""
    enabled = True

    def options(self):
        self.parser.usage = "%%prog %s" % self.normalized_name
        self.parser.add_option(
            "--runfor",
            metavar="PACKAGE",
            action="append",
            help="List tasks that should be run for PACKAGE")
        self.parser.add_option(
            "--type",
            metavar="TYPE",
            action="append",
            help="List tasks only of TYPE")
        self.parser.add_option(
            "--job",
            default=False,
            action="store_true",
            help="Generate job xml")

    def run(self, *args, **kwargs):
        self.set_repo(**kwargs)
        self.set_taskrepo(**kwargs)
        conn = self.taskrepo
        with conn:
            cur = conn.cursor()
            cur.execute("SELECT origin FROM config")
            result = cur.fetchone()
            origin = result[0]

            values = []
            joins = []
            where = []
            extra = ""
            if kwargs.get("type"):
                for x in range(0, len(kwargs.get("type"))):
                    joins.append("LEFT JOIN types AS t_%d ON t_%d.task_id = tasks.id" % (x, x))
                    where.append("t_%d.value=?" % (x,))
                    values.append(kwargs.get("type")[x])

            if kwargs.get("runfor"):
                for x in range(0, len(kwargs.get("runfor"))):
                    joins.append("LEFT JOIN runfor AS rf_%d ON rf_%d.task_id = tasks.id" % (x, x))
                    where.append("rf_%d.value=?" % (x,))
                    values.append(kwargs.get("runfor")[x])

            if where:
                extra = ' '.join(joins)
                extra = "%s WHERE %s" % (extra, ' AND '.join(where))

            cur.execute("SELECT name, description, owner FROM tasks %s" % extra, values)
            rows = cur.fetchall()
            if kwargs.get("job"):
                xmldoc = xml.dom.minidom.Document()
                job = xmldoc.createElement('job')
                recipeset = xmldoc.createElement('recipeSet')
                job.appendChild(recipeset)
                recipe = xmldoc.createElement('recipe')
                recipeset.appendChild(recipe)

            for row in rows:
                if kwargs.get("job"):
                    task = xmldoc.createElement('task')
                    recipe.appendChild(task)
                    task.setAttribute('name',"/%s/%s" % (self.repo.working_tree_dir.split('/')[-1],
                                                         row[0]))
                    fetch = xmldoc.createElement('fetch')
                    task.appendChild(fetch)
                    fetch.setAttribute('url', "%s?%s#%s" % (origin, self.repo.active_branch.name, row[0]))
                else:
                    print "%s\n\t[%s, %s]" % (row[0], row[1], row[2])
            if kwargs.get("job"):
                print job.toprettyxml()