mirror of
				https://github.com/Mikaela/Limnoria.git
				synced 2025-11-04 01:27:21 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			111 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			111 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python
 | 
						|
 | 
						|
###
 | 
						|
# Copyright (c) 2002, Jeremiah Fincher
 | 
						|
# All rights reserved.
 | 
						|
#
 | 
						|
# Redistribution and use in source and binary forms, with or without
 | 
						|
# modification, are permitted provided that the following conditions are met:
 | 
						|
#
 | 
						|
#   * Redistributions of source code must retain the above copyright notice,
 | 
						|
#     this list of conditions, and the following disclaimer.
 | 
						|
#   * Redistributions in binary form must reproduce the above copyright notice,
 | 
						|
#     this list of conditions, and the following disclaimer in the
 | 
						|
#     documentation and/or other materials provided with the distribution.
 | 
						|
#   * Neither the name of the author of this software nor the name of
 | 
						|
#     contributors to this software may be used to endorse or promote products
 | 
						|
#     derived from this software without specific prior written consent.
 | 
						|
#
 | 
						|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 | 
						|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | 
						|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 | 
						|
# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 | 
						|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 | 
						|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 | 
						|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 | 
						|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 | 
						|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 | 
						|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 | 
						|
# POSSIBILITY OF SUCH DAMAGE.
 | 
						|
###
 | 
						|
 | 
						|
import supybot
 | 
						|
 | 
						|
import fix
 | 
						|
from questions import *
 | 
						|
 | 
						|
import os
 | 
						|
import sys
 | 
						|
import optparse
 | 
						|
 | 
						|
import conf
 | 
						|
import debug
 | 
						|
 | 
						|
debug.minimumPriority = 'high'
 | 
						|
 | 
						|
def main():
 | 
						|
    parser = optparse.OptionParser(usage='Usage: %prog [options]',
 | 
						|
                                   version='supybot %s' % conf.version)
 | 
						|
    parser.add_option('-u', '--username', action='store', default='',
 | 
						|
                      dest='name',
 | 
						|
                      help='username for the user.')
 | 
						|
    parser.add_option('-p', '--password', action='store', default='',
 | 
						|
                      dest='password',
 | 
						|
                      help='password for the user.')
 | 
						|
    parser.add_option('-c', '--capability', action='append',
 | 
						|
                      dest='capabilities', metavar='CAPABILITY',
 | 
						|
                      help='capability the user should have; '
 | 
						|
                           'this option may be given multiple times.')
 | 
						|
    filename = os.path.join(conf.confDir, conf.userfile)
 | 
						|
    parser.add_option('-f', '--filename', action='store', default=filename,
 | 
						|
                      dest='filename',
 | 
						|
                      help='filename of your users.conf; '
 | 
						|
                           'defaults to %s' % filename)
 | 
						|
 | 
						|
    (options, args) = parser.parse_args()
 | 
						|
    if not options.name:
 | 
						|
        name = something('What is the user\'s name?')
 | 
						|
    else:
 | 
						|
        name = options.name
 | 
						|
 | 
						|
    if not options.password:
 | 
						|
        password = getpass('What is %s\'s password? ' % name)
 | 
						|
    else:
 | 
						|
        password = options.password
 | 
						|
 | 
						|
    if not options.capabilities:
 | 
						|
        capabilities = []
 | 
						|
        prompt = 'Would you like to give %s a capability?' % name
 | 
						|
        while yn(prompt) == 'y':
 | 
						|
            capabilities.append(anything('What capability?'))
 | 
						|
            prompt = 'Would you like to give %s another capability?' % name
 | 
						|
    else:
 | 
						|
        capabilities = options.capabilities
 | 
						|
 | 
						|
    conf.confDir = os.path.dirname(options.filename)
 | 
						|
    conf.userfile = os.path.basename(options.filename)
 | 
						|
    import ircdb
 | 
						|
 | 
						|
    try:
 | 
						|
        # First, let's check to see if the user is already in the database.
 | 
						|
        _ = ircdb.users.getUser(name)
 | 
						|
        # Uh oh.  That user already exists; otherwise we'd have KeyError'ed.
 | 
						|
        sys.stderr.write('That user already exists.  Try another name.\n')
 | 
						|
        sys.exit(-1)
 | 
						|
    except KeyError:
 | 
						|
        # Good.  No such user exists.  We'll pass.
 | 
						|
        pass
 | 
						|
    (id, user) = ircdb.users.newUser()
 | 
						|
    user.name = name
 | 
						|
    user.setPassword(password)
 | 
						|
    for capability in capabilities:
 | 
						|
        user.addCapability(capability)
 | 
						|
    ircdb.users.setUser(id, user)
 | 
						|
    print 'User %s added.' % name
 | 
						|
 | 
						|
if __name__ == '__main__':
 | 
						|
    try:
 | 
						|
        main()
 | 
						|
    except KeyboardInterrupt:
 | 
						|
        pass
 |