← Back to branch summary

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

« back to all changes in this revision

Viewing changes to scripts/usrmgt-server

  • Committer: William Grant
  • Date: 2010-07-03 01:38:41 UTC
  • Revision ID: grantw@unimelb.edu.au-20100703013841-6ifl1wsn4xj4w72k
External media directories with None as the path whitelist now permit all paths. This lets us include complicated dependencies more easily.

Show diffs side-by-side

added added

removed removed

Lines of Context:
scripts/usrmgt-server
1
 
#!/usr/bin/python
2
 
 
3
 
import os
4
 
import sys
5
 
import logging
6
 
 
7
 
import conf
8
 
import common.db
9
 
import common.chat
10
 
import common.makeuser
11
 
import common.studpath
12
 
 
13
 
# usage:
14
 
#   usrmgt-server <port> <magic>
15
 
 
16
 
# User management operations:
17
 
#   - Create local user
18
 
#   - [Re]Create jail for a user
19
 
#       - Create a svn repository for a user
20
 
#           - create repository
21
 
#           - svn config
22
 
#           - svn auth
23
 
#       - /etc/passwd entry
24
 
#   - Disable a user's account
25
 
#   - Enable a user's account
26
 
#   - Remove a user
27
 
#   - Rebuild svn config
28
 
#   - Rebuild svn auth file
29
 
#   - Rebuild passwd + push to nodes.
30
 
 
31
 
def activate_user(props):
32
 
    """Create the on-disk stuff for the given user.
33
 
       Sets the state of the user in the db from pending to enabled.
34
 
       Expected properties:
35
 
        login       - the user name for the jail
36
 
                      STRING REQUIRED
37
 
       Return Value: None
38
 
    """
39
 
 
40
 
    os.umask(0022) # Bad, but start_server sets it worse.
41
 
 
42
 
    login = props['login']
43
 
 
44
 
    db = common.db.DB()
45
 
 
46
 
    try:
47
 
 
48
 
        # FIXME: check we're pending
49
 
 
50
 
        details = db.get_user(login)
51
 
 
52
 
        # make svn config/auth
53
 
 
54
 
        repopath = os.path.join(conf.svn_repo_path, 'users', login)
55
 
        logging.debug("Creating user's Subversion repository")
56
 
        common.makeuser.make_svn_repo(repopath, throw_on_error=False)
57
 
 
58
 
        rebuild_svn_config(props)
59
 
 
60
 
        logging.debug("Adding Subversion authentication")
61
 
        passwd = common.makeuser.make_svn_auth(login, throw_on_error=False)
62
 
        logging.debug("passwd: %s" % passwd)
63
 
 
64
 
        logging.debug("Creating jail")
65
 
        common.makeuser.make_jail(login, details.unixid, svn_pass=passwd)
66
 
 
67
 
        logging.info("Enabling user")
68
 
        db.update_user(login, state='enabled')
69
 
 
70
 
        return {"response": "okay"}
71
 
 
72
 
    finally:
73
 
        db.close()
74
 
 
75
 
def rebuild_svn_config(props):
76
 
    """Rebuilds the svn config file
77
 
    Return value:
78
 
        response (okay, failure)
79
 
    """
80
 
    try:
81
 
        common.makeuser.rebuild_svn_config()
82
 
    except Exception, e:
83
 
        logging.warning('Rebuild of Subversion authorization config failed!')
84
 
        return{'response': 'failure', 'msg': repr(e)}
85
 
 
86
 
    return {'response': 'okay'}
87
 
 
88
 
def rebuild_svn_group_config(props):
89
 
    """Rebuilds the svn group config file
90
 
    Return value:
91
 
        response (okay, failure)
92
 
    """
93
 
    try:
94
 
        common.makeuser.rebuild_svn_group_config()
95
 
    except Exception, e:
96
 
        logging.warning(
97
 
            'Rebuild of Subversion group authorization config failed!')
98
 
        return{'response': 'failure', 'msg': repr(e)}
99
 
 
100
 
    return {'response': 'okay'}
101
 
 
102
 
def create_group_repository(props):
103
 
    """Creates on disk repository for the given group
104
 
    Expected properties:
105
 
        subj_short_name, year, semester, groupnm
106
 
    Return value:
107
 
        response (okay, failure)
108
 
    """
109
 
 
110
 
    subj_short_name = props['subj_short_name']
111
 
    year = props['year']
112
 
    semester = props['semester']
113
 
    groupnm = props['groupnm']
114
 
 
115
 
    namespace = "_".join([subj_short_name, year, semester, groupnm])
116
 
    repopath = os.path.join(conf.svn_repo_path, 'groups', namespace)
117
 
    logging.debug("Creating Subversion repository %s"%repopath)
118
 
    try:
119
 
        common.makeuser.make_svn_repo(repopath)
120
 
    except Exception, e:
121
 
        logging.error("Failed to create Subversion repository %s: %s"%
122
 
            (repopath,repr(e)))
123
 
        return {'response': 'failure', 'msg': repr(e)}
124
 
 
125
 
    return {'response': 'okay'}
126
 
 
127
 
actions = {
128
 
        'activate_user':activate_user,
129
 
        'create_group_repository':create_group_repository,
130
 
        'rebuild_svn_config':rebuild_svn_config,
131
 
        'rebuild_svn_group_config':rebuild_svn_group_config,
132
 
    }
133
 
 
134
 
def initializer():
135
 
    try:
136
 
        pidfile = open('/var/run/usrmgt-server.pid', 'w')
137
 
        pidfile.write('%d\n' % os.getpid())
138
 
        pidfile.close()
139
 
    except IOError, (errno, strerror):
140
 
        print "Couldn't write PID file. IO error(%s): %s" % (errno, strerror)
141
 
        sys.exit(1)
142
 
 
143
 
def dispatch(props):
144
 
    logging.debug(repr(props))
145
 
    action = props.keys()[0]
146
 
    return actions[action](props[action])
147
 
 
148
 
if __name__ == "__main__":
149
 
    pid = os.getpid()
150
 
 
151
 
    logging.basicConfig(filename="/var/log/usrmgt.log", level=logging.INFO)
152
 
    logging.info("Starting usrmgt server on port %d (pid = %d)" %
153
 
                 (conf.usrmgt_port, pid))
154
 
 
155
 
    common.chat.start_server(conf.usrmgt_port, conf.usrmgt_magic, True, dispatch, initializer)