summaryrefslogtreecommitdiffstats
path: root/cobbler/modules/cli_repo.py
blob: e8ec04a42963c50aa9e217aa38aacb55962b950d (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
"""
Repo CLI module.

Copyright 2007, Red Hat, Inc
Michael DeHaan <mdehaan@redhat.com>

This software may be freely redistributed under the terms of the GNU
general public license.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
"""

import distutils.sysconfig
import sys

plib = distutils.sysconfig.get_python_lib()
mod_path="%s/cobbler" % plib
sys.path.insert(0, mod_path)

from rhpl.translate import _, N_, textdomain, utf8
import commands
import cexceptions


class RepoFunction(commands.CobblerFunction):

    def help_me(self):
        return commands.HELP_FORMAT % ("cobbler repo","<add|edit|copy|list|rename|remove|report> [ARGS|--help]")

    def command_name(self):
        return "repo"

    def subcommands(self):
        return [ "add", "edit", "copy", "rename", "remove", "list", "report" ]

    def add_options(self, p, args):


        if not self.matches_args(args,["remove","report","list"]):

            p.add_option("--arch",             dest="arch",             help="overrides repo arch if required")
        if self.matches_args(args,["add"]):
            p.add_option("--clobber", dest="clobber", help="allow add to overwrite existing objects", action="store_true")
        if not self.matches_args(args,["remove","report","list"]):
            p.add_option("--createrepo-flags", dest="createrepo_flags", help="additional flags for createrepo")
            p.add_option("--keep-updated",     dest="keep_updated",     help="update on each reposync, yes/no")

        p.add_option("--name",                 dest="name",             help="ex: 'Fedora-8-updates-i386' (REQUIRED)")
        
        if not self.matches_args(args,["remove","report","list"]):
            p.add_option("--mirror",           dest="mirror",           help="source to mirror (REQUIRED)")
            p.add_option("--priority",         dest="priority",         help="set priority") 
            p.add_option("--rpm-list",         dest="rpm_list",         help="just mirror these rpms")
            p.add_option("--yumopts",          dest="yumopts",          help="ex: pluginvar=abcd")

        if self.matches_args(args,["copy","rename"]):

            p.add_option("--newname",          dest="newname",          help="used for copy/edit")

        if not self.matches_args(args,["remove","report","list"]):
            p.add_option("--no-sync",     action="store_true", dest="nosync", help="suppress sync for speed")
        if not self.matches_args(args,["report","list"]):
            p.add_option("--no-triggers", action="store_true", dest="notriggers", help="suppress trigger execution")
        if not self.matches_args(args,["remove","report","list"]):
            p.add_option("--owners", dest="owners", help="specify owners for authz_ownership module")


    def run(self):

        obj = self.object_manipulator_start(self.api.new_repo,self.api.repos)
        if obj is None:
            return True

        if self.options.arch:             obj.set_arch(self.options.arch)
        if self.options.createrepo_flags: obj.set_createrepo_flags(self.options.createrepo_flags)
        if self.options.rpm_list:         obj.set_rpm_list(self.options.rpm_list)
        if self.options.keep_updated:     obj.set_keep_updated(self.options.keep_updated)
        if self.options.priority:         obj.set_priority(self.options.priority)
        if self.options.mirror:           obj.set_mirror(self.options.mirror)
        if self.options.yumopts:          obj.set_yumopts(self.options.yumopts)

        if self.options.owners:
            obj.set_owners(self.options.owners)

        return self.object_manipulator_finish(obj, self.api.repos, self.options)



########################################################
# MODULE HOOKS

def register():
    """
    The mandatory cobbler module registration hook.
    """
    return "cli"

def cli_functions(api):
    return [
       RepoFunction(api)
    ]
    return []