~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to remakeallusers.py

• browse files at revision 804
• compare with another revision
• download diff
• download tarball
• view history from revision 804

• reverse the comparison (804 to 1099.1.54)
• stop comparing with revision 1099.1.54

Show diffs side-by-side

added

removed

remakeallusers.py

 

1

#!/usr/bin/env python

 

2

# IVLE - Informatics Virtual Learning Environment

 

3

# Copyright (C) 2007-2008 The University of Melbourne

 

4

#

 

5

# This program is free software; you can redistribute it and/or modify

 

6

# it under the terms of the GNU General Public License as published by

 

7

# the Free Software Foundation; either version 2 of the License, or

 

8

# (at your option) any later version.

 

9

#

 

10

# This program is distributed in the hope that it will be useful,

 

11

# but WITHOUT ANY WARRANTY; without even the implied warranty of

 

12

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 

13

# GNU General Public License for more details.

 

14

#

 

15

# You should have received a copy of the GNU General Public License

 

16

# along with this program; if not, write to the Free Software

 

17

# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 

18

 

 

19

# Program: RemakeAllUsers

 

20

# Author:  Matt Giuca

 

21

# Date:    3/3/2008

 

22

 

 

23

# Script to rebuild jails for all users on the system.

 

24

# Requires root to run.

 

25

 

 

26

import sys

 

27

import os

 

28

import common.db

 

29

import os.path

 

30

import pwd

 

31

import conf

 

32

import common.makeuser

 

33

import optparse

 

34

import shutil

 

35

import logging

 

36

 

 

37

p = optparse.OptionParser()

 

38

p.add_option('--incremental', '-i', action='store_true')

 

39

p.add_option('--verbose', '-v', action='store_true')

 

40

options, arguments = p.parse_args()

 

41

 

 

42

if options.verbose:

 

43

    logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s',

 

44

                        level=logging.DEBUG)

 

45

 

 

46

if os.getuid() != 0:

 

47

    print >> sys.stderr, "%s must be run as root" % sys.argv[0]

 

48

    sys.exit(1)

 

49

 

 

50

try:

 

51

    db = common.db.DB()

 

52

    list = db.get_users()

 

53

    res = db.get_all('login', ['login', 'unixid'])

 

54

    def repack(flds):

 

55

        return (flds['login'], flds['unixid'])

 

56

    uids = dict(map(repack,res))

 

57

except Exception, message:

 

58

    logging.error(str(message))

 

59

    sys.exit(1)

 

60

 

 

61

# First check that our __templateuser__ directory exits, if not create it

 

62

templateuserdir = os.path.join(conf.jail_base, '__templateuser__')

 

63

stagingdir = templatedir = os.path.join(conf.jail_base, '__staging__')

 

64

if not os.path.isdir(templateuserdir):

 

65

    os.mkdir(templateuserdir)

 

66

 

 

67

# Generate manifest of files that differ between template user and staging

 

68

if options.incremental:

 

69

    logging.info("incremental rebuild started")

 

70

    manifest = common.makeuser.generate_manifest(stagingdir, templateuserdir)

 

71

    logging.debug("generated change manifest")

 

72

    logging.debug("  adding/updating: %s" % manifest[0])

 

73

    logging.debug("  removing: %s" % manifest[1])

 

74

else:

 

75

    # Force a full rebuild

 

76

    logging.info("full rebuild started")

 

77

    manifest = None

 

78

 

 

79

list.sort(key=lambda user: user.login)

 

80

for user in list:

 

81

    login = user.login

 

82

 

 

83

    try:

 

84

        # Resolve the user's login into a UID

 

85

        # Create the user if it does not exist

 

86

        try:

 

87

            uid = uids[login]

 

88

        except KeyError:

 

89

            raise Exception("user %s does not have a unixid in the database"

 

90

                % login)

 

91

        # Remake the user's jail

 

92

        common.makeuser.make_jail(login, uid, manifest=manifest)

 

93

    except Exception, message:

 

94

        logging.warning(str(message))

 

95

        continue

 

96

 

 

97

    logging.debug("recreated user %s's jail." % login)

 

98

    

 

99

# Update the template user directory

 

100

logging.debug("updating templateuser directory")

 

101

shutil.rmtree(templateuserdir)

 

102

#common.makeuser.linktree(stagingdir, templateuserdir)

 

103

common.makeuser.make_jail('__templateuser__', 0, manifest=manifest)

 

104

shutil.rmtree(os.path.join(templateuserdir, 'home'))

 

105

 

 

106

logging.info("rebuild completed successfully")

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1