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