← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-remakeuser

  • Committer: chadnickbok
  • Date: 2009-01-22 01:21:48 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:1172
Fixed an issue with my previous patch to this file. Now, if
publicmode is not set, users will not be shown tracebacks.

Previously, if publicmode was undefined, users would be shown
tracebacks.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-remakeuser
29
29
import optparse
30
30
import logging
31
31
 
32
 
import ivle.config
33
 
import ivle.database
 
32
import ivle.db
34
33
import ivle.makeuser
35
34
 
36
35
p = optparse.OptionParser()
49
48
                          os.path.basename(sys.argv[0])
50
49
    sys.exit(1)
51
50
 
52
 
store = ivle.database.get_store(ivle.config.Config())
53
 
 
54
 
if options.all:
55
 
    users = store.find(ivle.database.User).order_by(ivle.database.User.login)
56
 
else:
57
 
    if len(arguments) == 0:
58
 
        print >> sys.stderr, "must be run with -a or a username"
59
 
        sys.exit(1)
60
 
    users = [ivle.database.User.get_by_login(store, arguments[0])]
 
51
try:
 
52
    db = ivle.db.DB()
 
53
    if options.all:
 
54
        list = db.get_users()
 
55
    else:
 
56
        if len(arguments) == 0:
 
57
            print >> sys.stderr, "must be run with -a or a username"
 
58
            sys.exit(1)
 
59
        list = [db.get_user(arguments[0])]
 
60
    res = db.get_all('login', ['login', 'unixid'])
 
61
    def repack(flds):
 
62
        return (flds['login'], flds['unixid'])
 
63
    uids = dict(map(repack,res))
 
64
except Exception, message:
 
65
    logging.error(str(message))
 
66
    sys.exit(1)
61
67
 
62
68
logging.info("rebuild started")
63
69
 
64
 
for user in users:
 
70
list.sort(key=lambda user: user.login)
 
71
for user in list:
 
72
    login = user.login
 
73
 
65
74
    try:
66
 
        ivle.makeuser.make_jail(user)
 
75
        # Resolve the user's login into a UID
 
76
        try:
 
77
            uid = uids[login]
 
78
        except KeyError:
 
79
            raise Exception("user %s does not have a unixid in the database"
 
80
                % login)
 
81
        # Remake the user's jail
 
82
        ivle.makeuser.make_jail(login, uid)
67
83
    except Exception, message:
68
84
        logging.warning(str(message))
69
85
        continue
70
86
 
71
 
    logging.debug("recreated user %s's jail." % user.login)
 
87
    logging.debug("recreated user %s's jail." % login)
72
88
    
73
89
logging.info("rebuild completed successfully")