100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright © 2007-2008 Red Hat, Inc. All rights reserved.
|
|
#
|
|
# This copyrighted material is made available to anyone wishing to use, modify,
|
|
# copy, or redistribute it subject to the terms and conditions of the GNU
|
|
# General Public License v.2. This program is distributed in the hope that it
|
|
# will be useful, but WITHOUT ANY WARRANTY expressed or implied, including the
|
|
# implied warranties 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 this program;
|
|
# if not, write to the Free Software Foundation, Inc., 51 Franklin Street,
|
|
# Fifth Floor, Boston, MA 02110-1301, USA. Any Red Hat trademarks that are
|
|
# incorporated in the source code or documentation are not subject to the GNU
|
|
# General Public License and may only be used or replicated with the express
|
|
# permission of Red Hat, Inc.
|
|
#
|
|
# Red Hat Author(s): Mike McGrath <mmcgrath@redhat.com>
|
|
#
|
|
|
|
|
|
from fedora.tg.client import BaseClient, AuthError, ServerError
|
|
|
|
import sys
|
|
import os
|
|
from shutil import move
|
|
import logging
|
|
|
|
FAS_URL = 'http://localhost:8080/fas/json/'
|
|
|
|
|
|
class MakeShellAccounts(BaseClient):
|
|
def group_list(self, search='*'):
|
|
params = {'search' : search}
|
|
data = self.send_request('group_list', auth=False, input=params)
|
|
return data
|
|
|
|
|
|
|
|
def groups_text(self, groups=None, people=None):
|
|
i = 0
|
|
file = open('/tmp/group.txt', 'w')
|
|
if not groups:
|
|
groups = self.group_list()
|
|
if not people:
|
|
people = self.people_list()
|
|
|
|
''' First create all of our users/groups combo '''
|
|
usernames = {}
|
|
for person in people:
|
|
uid = person['id']
|
|
username = person['username']
|
|
usernames['%s' % uid] = username
|
|
file.write("=%i %s:x:%i:\n" % (uid, username, uid))
|
|
file.write( "0%i %s:x:%i:\n" % (i, username, uid))
|
|
file.write( ".%s %s:x:%i:\n" % (username, username, uid))
|
|
i = i + 1
|
|
|
|
for group in groups['groups']:
|
|
gid = group['id']
|
|
name = group['name']
|
|
# print groups['memberships'][m]
|
|
# print groups['memberships'][1228]
|
|
memberships = ''
|
|
try:
|
|
''' Shoot me now I know this isn't right '''
|
|
members = []
|
|
for member in groups['memberships'][u'%s' % gid]:
|
|
members.append(usernames['%s' % member['person_id']])
|
|
memberships = ','.join(members)
|
|
except KeyError:
|
|
''' No users exist in the group '''
|
|
pass
|
|
file.write( "=%i %s:x:%i:%s\n" % (gid, name, gid, memberships))
|
|
file.write("0%i %s:x:%i:%s\n" % (i, name, gid, memberships))
|
|
file.write(".%s %s:x:%i:%s\n" % (name, name, gid, memberships))
|
|
i = i + 1
|
|
|
|
file.close()
|
|
|
|
|
|
def people_list(self, search='*'):
|
|
params = {'search' : search}
|
|
data = self.send_request('people_list', auth=False, input=params)
|
|
return data['people']
|
|
|
|
def make_group_db(self):
|
|
self.groups_text()
|
|
os.system('makedb -o /tmp/group.db /tmp/group.txt')
|
|
|
|
def install_group_db(self):
|
|
move('/tmp/group.db', '/var/db/group.db')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
fas = MakeShellAccounts(FAS_URL, None, None, None)
|
|
fas.make_group_db()
|
|
fas.install_group_db()
|
|
|