← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/usrmgt-server

  • Committer: mattgiuca
  • Date: 2008-07-15 07:19:34 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:875
Added "migrations" directory, which contains incremental database update
    scripts.
Updated users.sql, uniqueness key on offering table.
Added migration matching this update to the migrations directory. Mm handy!

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
 
    os.umask(0022) # Bad, but start_server sets it worse.
43
 
 
44
 
    login = props['login']
45
 
 
46
 
    # FIXME: check we're pending
47
 
 
48
 
    # Get the full User object from the db associated with this
49
 
    user = ivle.database.User.get_by_login(store, login)
50
 
 
51
 
    # make svn config/auth
52
 
    repopath = os.path.join(config['paths']['svn']['repo_path'],
53
 
                            'users', login)
54
 
    logging.debug("Creating user's Subversion repository")
55
 
    ivle.makeuser.make_svn_repo(repopath, throw_on_error=True)
56
 
 
57
 
    rebuild_svn_config(store, props, config)
58
 
 
59
 
    logging.debug("Adding Subversion authentication")
60
 
    passwd = ivle.makeuser.make_svn_auth(store, login, config,
61
 
                                         throw_on_error=True)
62
 
 
63
 
    logging.debug("Creating jail")
64
 
    ivle.makeuser.make_jail(user, config)
65
 
 
66
 
    logging.info("Enabling user")
67
 
    user.state = u'enabled'
68
 
 
69
 
    return {"response": "okay"}
70
 
 
71
 
def rebuild_svn_config(store, props, config):
72
 
    """Rebuilds the svn config file
73
 
    @param config: An ivle.config.Config object.
74
 
    @return: response (okay, failure)
75
 
    """
76
 
    try:
77
 
        ivle.makeuser.rebuild_svn_config(store, config)
78
 
    except Exception, e:
79
 
        logging.warning('Rebuild of Subversion authorization config failed!')
80
 
        return{'response': 'failure', 'msg': repr(e)}
81
 
 
82
 
    return {'response': 'okay'}
83
 
 
84
 
def rebuild_svn_group_config(store, props, config):
85
 
    """Rebuilds the svn group config file
86
 
    @param config: An ivle.config.Config object.
87
 
    @return: response (okay, failure)
88
 
    """
89
 
    try:
90
 
        ivle.makeuser.rebuild_svn_group_config(store, config)
91
 
    except Exception, e:
92
 
        logging.warning(
93
 
            'Rebuild of Subversion group authorization config failed!')
94
 
        return{'response': 'failure', 'msg': repr(e)}
95
 
 
96
 
    return {'response': 'okay'}
97
 
 
98
 
def create_group_repository(store, props, config):
99
 
    """Creates on disk repository for the given group
100
 
    @param config: An ivle.config.Config object.
101
 
    Expected properties:
102
 
        subj_short_name, year, semester, groupnm
103
 
    @return: response (okay, failure)
104
 
    """
105
 
 
106
 
    subj_short_name = props['subj_short_name']
107
 
    year = props['year']
108
 
    semester = props['semester']
109
 
    groupnm = props['groupnm']
110
 
 
111
 
    namespace = "_".join([subj_short_name, year, semester, groupnm])
112
 
    repopath = os.path.join(config['paths']['svn']['repo_path'],
113
 
                            'groups', namespace)
114
 
    logging.debug("Creating Subversion repository %s"%repopath)
115
 
    try:
116
 
        ivle.makeuser.make_svn_repo(repopath)
117
 
    except Exception, e:
118
 
        logging.error("Failed to create Subversion repository %s: %s"%
119
 
            (repopath,repr(e)))
120
 
        return {'response': 'failure', 'msg': repr(e)}
121
 
 
122
 
    return {'response': 'okay'}
123
 
 
124
 
actions = {
125
 
        'activate_user':activate_user,
126
 
        'create_group_repository':create_group_repository,
127
 
        'rebuild_svn_config':rebuild_svn_config,
128
 
        'rebuild_svn_group_config':rebuild_svn_group_config,
129
 
    }
130
 
 
131
 
def initializer():
132
 
    logging.basicConfig(filename="/var/log/usrmgt.log", level=logging.INFO)
133
 
    logging.info("Starting usrmgt server on port %d (pid = %d)" %
134
 
                 (config['usrmgt']['port'], pid))
135
 
 
136
 
    try:
137
 
        pidfile = open('/var/run/usrmgt-server.pid', 'w')
138
 
        pidfile.write('%d\n' % os.getpid())
139
 
        pidfile.close()
140
 
    except IOError, (errno, strerror):
141
 
        print "Couldn't write PID file. IO error(%s): %s" % (errno, strerror)
142
 
        sys.exit(1)
143
 
 
144
 
def dispatch(props):
145
 
    logging.debug(repr(props))
146
 
 
147
 
    store = ivle.database.get_store(config)
148
 
    action = props.keys()[0]
149
 
    res = actions[action](store, props[action], config)
150
 
 
151
 
    if res['response'] == 'okay':
152
 
        store.commit()
153
 
    else:
154
 
        store.rollback()
155
 
    store.close()
156
 
    return res
157
 
 
158
 
if __name__ == "__main__":
159
 
    pid = os.getpid()
160
 
 
161
 
    ivle.chat.start_server(config['usrmgt']['port'],config['usrmgt']['magic'],
162
 
                           True, dispatch, initializer)