← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/usrmgt-server

  • Committer: William Grant
  • Date: 2009-05-26 02:10:08 UTC
  • Revision ID: grantw@unimelb.edu.au-20090526021008-hnka598yacu5w6da
Always set isdir = False in the file browser template filling.

Previously we used os.path.isdir outside the jail, which was terribly
unreliable and prone to crashing. We should do it properly once we can get
server-side access to the jails.

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)