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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
# indirectly based on ctypes implementation: Victor Stinner, 2008-05-08
"""
This module provides access to the Unix password database.
It is available on all Unix versions.
Password database entries are reported as 7-tuples containing the following
items from the password database (see `<pwd.h>'), in order:
pw_name, pw_passwd, pw_uid, pw_gid, pw_gecos, pw_dir, pw_shell.
The uid and gid items are integers, all others are strings. An
exception is raised if the entry asked for cannot be found.
"""
from _pwdgrp_cffi import ffi, lib
import _structseq
import thread
_lock = thread.allocate_lock()
try: from __pypy__ import builtinify
except ImportError: builtinify = lambda f: f
class struct_passwd:
"""
pwd.struct_passwd: Results from getpw*() routines.
This object may be accessed either as a tuple of
(pw_name,pw_passwd,pw_uid,pw_gid,pw_gecos,pw_dir,pw_shell)
or via the object attributes as named in the above tuple.
"""
__metaclass__ = _structseq.structseqtype
name = "pwd.struct_passwd"
pw_name = _structseq.structseqfield(0)
pw_passwd = _structseq.structseqfield(1)
pw_uid = _structseq.structseqfield(2)
pw_gid = _structseq.structseqfield(3)
pw_gecos = _structseq.structseqfield(4)
pw_dir = _structseq.structseqfield(5)
pw_shell = _structseq.structseqfield(6)
def _mkpwent(pw):
return struct_passwd([
ffi.string(pw.pw_name),
ffi.string(pw.pw_passwd),
pw.pw_uid,
pw.pw_gid,
ffi.string(pw.pw_gecos),
ffi.string(pw.pw_dir),
ffi.string(pw.pw_shell)])
@builtinify
def getpwuid(uid):
"""
getpwuid(uid) -> (pw_name,pw_passwd,pw_uid,
pw_gid,pw_gecos,pw_dir,pw_shell)
Return the password database entry for the given numeric user ID.
See pwd.__doc__ for more on password database entries.
"""
with _lock:
pw = lib.getpwuid(uid)
if not pw:
raise KeyError("getpwuid(): uid not found: %s" % uid)
return _mkpwent(pw)
@builtinify
def getpwnam(name):
"""
getpwnam(name) -> (pw_name,pw_passwd,pw_uid,
pw_gid,pw_gecos,pw_dir,pw_shell)
Return the password database entry for the given user name.
See pwd.__doc__ for more on password database entries.
"""
if not isinstance(name, basestring):
raise TypeError("expected string")
name = str(name)
with _lock:
pw = lib.getpwnam(name)
if not pw:
raise KeyError("getpwname(): name not found: %s" % name)
return _mkpwent(pw)
@builtinify
def getpwall():
"""
getpwall() -> list_of_entries
Return a list of all available password database entries, in arbitrary order.
See pwd.__doc__ for more on password database entries.
"""
users = []
with _lock:
lib.setpwent()
while True:
pw = lib.getpwent()
if not pw:
break
users.append(_mkpwent(pw))
lib.endpwent()
return users
__all__ = ('struct_passwd', 'getpwuid', 'getpwnam', 'getpwall')
if __name__ == "__main__":
# Uncomment next line to test CPython implementation
# from pwd import getpwuid, getpwnam, getpwall
from os import getuid
uid = getuid()
pw = getpwuid(uid)
print("uid %s: %s" % (pw.pw_uid, pw))
name = pw.pw_name
print("name %r: %s" % (name, getpwnam(name)))
print("All:")
for pw in getpwall():
print(pw)
|
