mirror of
https://gitdl.cn/https://github.com/chakralinux/core.git
synced 2025-02-05 15:37:14 +08:00
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
|
# userinfo.py
|
||
|
#
|
||
|
# (c) Copyright 2009-2010 Michael Towers (larch42 at googlemail dot com)
|
||
|
#
|
||
|
# This file is part of the larch project.
|
||
|
#
|
||
|
# larch is free software; you can redistribute it and/or modify
|
||
|
# it under the terms of the GNU General Public License as published by
|
||
|
# the Free Software Foundation; either version 2 of the License, or
|
||
|
# (at your option) any later version.
|
||
|
#
|
||
|
# larch is distributed in the hope that it will be useful,
|
||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
# GNU General Public License for more details.
|
||
|
#
|
||
|
# You should have received a copy of the GNU General Public License
|
||
|
# along with larch; if not, write to the Free Software Foundation, Inc.,
|
||
|
# 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
||
|
#
|
||
|
#----------------------------------------------------------------------------
|
||
|
# 2010.07.11
|
||
|
|
||
|
import os
|
||
|
from ConfigParser import SafeConfigParser
|
||
|
|
||
|
# Default list of 'additional' groups for a new user
|
||
|
BASEGROUPS = 'video,audio,optical,storage,scanner,power,camera'
|
||
|
|
||
|
class Userinfo:
|
||
|
def __init__(self, profile):
|
||
|
self.profile_dir = profile
|
||
|
|
||
|
def getusers(self):
|
||
|
"""Read user information by means of a SafeConfigParser instance.
|
||
|
This is then available as self.userconf.
|
||
|
"""
|
||
|
self.userconf = SafeConfigParser({'pw':'', 'maingroup':'', 'uid':'',
|
||
|
'skel':'', 'xgroups':BASEGROUPS, 'expert':''})
|
||
|
users = self.profile_dir + '/users'
|
||
|
if os.path.isfile(users):
|
||
|
try:
|
||
|
self.userconf.read(users)
|
||
|
except:
|
||
|
error0(_("Invalid 'users' file"))
|
||
|
|
||
|
def allusers(self):
|
||
|
self.getusers()
|
||
|
return self.userconf.sections()
|
||
|
|
||
|
def get(self, user, field):
|
||
|
return self.userconf.get(user, field)
|
||
|
|
||
|
def userinfo(self, user, fields):
|
||
|
"""Get an ordered list of the given field data for the given user.
|
||
|
"""
|
||
|
return [self.userconf.get(user, f) for f in fields]
|
||
|
|
||
|
def userset(self, uname, field, text):
|
||
|
self.userconf.set(uname, field, text)
|
||
|
|
||
|
def newuser(self, user):
|
||
|
try:
|
||
|
self.userconf.add_section(user)
|
||
|
return self.saveusers()
|
||
|
except:
|
||
|
error0(_("Couldn't add user '%s'") % user)
|
||
|
return False
|
||
|
|
||
|
def deluser(self, user):
|
||
|
try:
|
||
|
self.userconf.remove_section(user)
|
||
|
return self.saveusers()
|
||
|
except:
|
||
|
error0(_("Couldn't remove user '%s'") % user)
|
||
|
return False
|
||
|
|
||
|
def saveusers(self):
|
||
|
"""Save the user configuration data (in 'INI' format)
|
||
|
"""
|
||
|
try:
|
||
|
fh = None
|
||
|
fh = open(self.profile_dir + '/users', 'w')
|
||
|
self.userconf.write(fh)
|
||
|
fh.close()
|
||
|
return True
|
||
|
except:
|
||
|
if fh:
|
||
|
fh.close()
|
||
|
error0(_("Couldn't save 'users' file"))
|
||
|
self.getusers()
|
||
|
return False
|
||
|
|