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

428 by mattgiuca
common.db: Added get_users function to get all users.
1
#!/usr/bin/env python
2
# IVLE - Informatics Virtual Learning Environment
3
# Copyright (C) 2007-2008 The University of Melbourne
4
#
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
#
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
#
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
19
# Program: ListUsers
20
# Author:  Matt Giuca
21
# Date:    6/2/2008
22
23
# Script to list all users on the system.
24
# Requires root to run.
25
26
import sys
27
import os
28
import common.db
754 by mattgiuca
listusers.py: Added options processing (now accepts -h and -n).
29
import getopt
30
31
# Options processing
32
def usage():
33
    print "listusers.py [OPTIONS]"
34
    print "     Gets a list of all users in the IVLE database."
35
    print "     Must be run as root."
36
    print "     OPTIONS"
37
    print "         -h --help       This message."
38
    print "         -n --names      Just print usernames."
39
40
justnames = False
41
try:
42
    opts, _ = getopt.getopt(sys.argv[1:], "nh", ["names", "help"])
43
except getopt.GetoptError, e:
44
    print str(e)
45
    usage()
46
    sys.exit()
47
opts = dict(opts)
48
if "-h" in opts or "--help" in opts:
49
    usage()
50
    sys.exit()
51
if "-n" in opts or "--names" in opts:
52
    justnames = True
428 by mattgiuca
common.db: Added get_users function to get all users.
53
54
if os.getuid() != 0:
451 by mattgiuca
Updated common/db.py and listusers to deal with the new "state" column in the
55
    print "Must run listusers.py as root."
428 by mattgiuca
common.db: Added get_users function to get all users.
56
    sys.exit()
57
58
try:
59
    db = common.db.DB()
60
    list = db.get_users()
61
except Exception, message:
62
    print "Error: " + str(message)
63
    sys.exit(1)
64
500 by mattgiuca
db: get_user and get_users now return User objects instead of dictionaries.
65
list.sort(key=lambda user: user.login)
428 by mattgiuca
common.db: Added get_users function to get all users.
66
for user in list:
754 by mattgiuca
listusers.py: Added options processing (now accepts -h and -n).
67
    if justnames:
68
        print user.login
69
    else:
70
        print repr(user)