summaryrefslogtreecommitdiffstats
path: root/cobbler/action_litesync.py
blob: 51369eecd2808914cace7327845f94832679548e (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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
"""
Running small pieces of cobbler sync when certain actions are taken,
such that we don't need a time consuming sync when adding new
systems if nothing has changed for systems that have already 
been created.

Copyright 2006, 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 os
import os.path
import shutil
import time
import yaml # Howell-Clark version
import sub_process
import sys

import cexceptions
import utils
import action_sync
import cobbler_msg
import cexceptions
import traceback
import errno

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


class BootLiteSync:
    """
    Handles conversion of internal state to the tftpboot tree layout
    """

    def __init__(self,config):
        """
        Constructor
        """
        self.verbose     = True
        self.config      = config
        self.distros     = config.distros()
        self.profiles    = config.profiles()
        self.systems     = config.systems()
        self.settings    = config.settings()
        self.repos       = config.repos()
        self.sync        = action_sync.BootSync(self.config) 

    def add_single_distro(self, name):
        # get the distro record
        distro = self.distros.find(name)
        if distro is None:
            raise cexceptions.CobblerException("error in distro lookup: %s" % name)
        # generate YAML file in distros/$name in webdir
        self.sync.write_distro_file(distro)
        # copy image files to images/$name in webdir & tftpboot:
        self.sync.copy_single_distro_files(distro)

    def remove_single_distro(self, name):
        # delete distro YAML file in distros/$name in webdir
        self.sync.rmfile(os.path.join(self.settings.webdir, "distros", name))
        # delete contents of images/$name directory in webdir
        self.sync.rmtree(os.path.join(self.settings.webdir, "images", name))
        # delete contents of images/$name in tftpboot
        self.sync.rmtree(os.path.join(self.settings.tftpboot, "images", name))
        # delete potential symlink to tree in webdir/links
        self.sync.rmfile(os.path.join(self.settings.webdir, "links", name)) 

    def add_single_profile(self, name):
        # get the profile object:
        profile = self.profiles.find(name)
        if profile is None:
            raise cexceptions.CobblerException("error in profile lookup")
        # rebuild profile_list YAML file in webdir
        self.sync.write_listings()
        # add profiles/$name YAML file in webdir
        self.sync.write_profile_file(profile)
        # generate kickstart for kickstarts/$name/ks.cfg in webdir
        self.sync.validate_kickstart_for_specific_profile(profile)
    
    def remove_single_profile(self, name):
        # rebuild profile_list YAML file in webdir
        self.sync.write_listings()
        # delete profiles/$name file in webdir
        self.sync.rmfile(os.path.join(self.settings.webdir, "profiles", name))
        # delete contents on kickstarts/$name directory in webdir
        self.sync.rmtree(os.path.join(self.settings.webdir, "kickstarts", name))
    
    def add_single_system(self, name):
        # get the system object:
        system = self.systems.find(name)
        if system is None:
            raise cexceptions.CobblerException("error in system lookup")
        # rebuild system_list file in webdir
        self.sync.regen_ethers() # /etc/ethers, for dnsmasq & rarpd
        self.sync.regen_hosts()  # /var/lib/cobbler/cobbler_hosts, pretty much for dnsmasq
        self.sync.write_listings()
        # write the PXE and YAML files for the system
        self.sync.write_all_system_files(system)
        # per system kickstarts
        self.sync.validate_kickstart_for_specific_system(system)

    def remove_single_system(self, name):
        # rebuild system_list file in webdir
        self.sync.write_listings()
        # delete system YAML file in systems/$name in webdir
        self.sync.rmfile(os.path.join(self.settings.webdir, "systems", name))
        # delete contents of kickstarts_sys/$name in webdir
        filename = self.sync.get_pxe_filename(name)
        self.sync.rmtree(os.path.join(self.settings.webdir, "kickstarts_sys", filename))
        
        # delete PXE Linux configuration file (which might be in one of two places)
        itanic = False
        system_record = self.systems.find(name)
        profile = self.profiles.find(system_record.profile)
        # allow cobbler deletes to still work in the cobbler config is discombobulated
        if profile is not None:
            distro = self.distros.find(profile.distro)
            if distro is not None and distro in [ "ia64", "IA64"]:
                itanic = True
        if not itanic:
            self.sync.rmfile(os.path.join(self.settings.tftpboot, "pxelinux.cfg", filename))
        else:
            self.sync.rmfile(os.path.join(self.settings.tftpboot, filename))