← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/usrmgt-server

  • Committer: dcoles
  • Date: 2008-07-03 04:20:54 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:803
Setup: Modularised setup.py so it is now no longer over 1000 lines. This should 
allow us to get in there and tidy up each module much easier. Also removed 
updatejails since this functionality seems to be duplicated with remakeuser.py 
and remakealluser.py scripts.

Show diffs side-by-side

added added

removed removed

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