1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#!/usr/bin/env python
"""
Utilities to manage a hash db of users for apache
"""
import anydbm
import htpasswd
__all__ = ["add_user", "delete_user", "update_user",
"get_user", "list_users", "raw_add"]
def get_user(user, db):
""" Get user info from the DB """
db_dict = anydbm.open(db, 'r')
ret_user = "%s:%s" %(user, db_dict[user])
db_dict.close()
return ret_user
def list_users(db):
""" List all the users in the DB """
db_dict = anydbm.open(db, 'r')
ret_str = '\n'.join(["%s:%s" %(k, v) for k, v in db_dict.iteritems()])
db_dict.close()
return ret_str
def add_user(user, passwd, db):
""" Add user to the DB, creating if need be """
db_dict = anydbm.open(db, 'c')
db_dict[user] = htpasswd.hash_password(passwd)
db_dict.close()
return True
def raw_add(line, db):
""" Add raw line to the DB """
db_dict = anydbm.open(db, 'w')
user, passwd = line.split(':')
db_dict[user] = passwd
db_dict.close()
return True
def delete_user(user, db):
""" Remove user from the DB """
db_dict = anydbm.open(db, 'w')
del db_dict[user]
db_dict.close()
return True
def update_user(user, passwd, db):
""" Change the users pass in the DB """
db_dict = anydbm.open(db, 'w')
if user in db_dict:
db_dict[user] = htpasswd.hash_password(passwd)
db_dict.close()
return True
else:
return False
|