mirror of https://github.com/Mikaela/Limnoria.git
143 lines
5.7 KiB
Plaintext
143 lines
5.7 KiB
Plaintext
![]() |
#!/usr/bin/env python
|
||
|
|
||
|
###
|
||
|
# Copyright (c) 2002-2004, 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 supybot.fix as fix
|
||
|
from supybot.questions import *
|
||
|
|
||
|
import os
|
||
|
import sys
|
||
|
import optparse
|
||
|
|
||
|
def main():
|
||
|
import supybot.log as log
|
||
|
import supybot.conf as conf
|
||
|
conf.supybot.log.stdout.setValue(False)
|
||
|
parser = optparse.OptionParser(usage='Usage: %prog [options] <users.conf>',
|
||
|
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('-x', '--hashed', action='store_const', const=1,
|
||
|
default=0, dest='hashed',
|
||
|
help='hash encrypt the password.')
|
||
|
parser.add_option('-n', '--plain', action='store_const', const=2,
|
||
|
default=0, dest='hashed',
|
||
|
help='store the password in plain text.')
|
||
|
parser.add_option('-c', '--capability', action='append',
|
||
|
dest='capabilities', metavar='CAPABILITY',
|
||
|
help='capability the user should have; '
|
||
|
'this option may be given multiple times.')
|
||
|
(options, args) = parser.parse_args()
|
||
|
if len(args) is not 1:
|
||
|
parser.error('Specify the users.conf file you\'d like to use. '
|
||
|
'Be sure *not* to specify your registry file, generated '
|
||
|
'by supybot-wizard. This is not the file you want. '
|
||
|
'Instead, take a look in your conf directory (usually '
|
||
|
'named "conf") and take a gander at the file '
|
||
|
'"users.conf". That\'s the one you want.')
|
||
|
|
||
|
filename = os.path.abspath(args[0])
|
||
|
conf.supybot.directories.log.setValue('/')
|
||
|
conf.supybot.directories.conf.setValue('/')
|
||
|
conf.supybot.directories.data.setValue('/')
|
||
|
conf.supybot.directories.plugins.setValue(['/'])
|
||
|
conf.supybot.databases.users.filename.setValue(filename)
|
||
|
import supybot.ircdb as ircdb
|
||
|
|
||
|
if not options.name:
|
||
|
name = ''
|
||
|
while not name:
|
||
|
name = something('What is the user\'s name?')
|
||
|
try:
|
||
|
# 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.
|
||
|
output('That user already exists. Try another name.')
|
||
|
name = ''
|
||
|
except KeyError:
|
||
|
# Good. No such user exists. We'll pass.
|
||
|
pass
|
||
|
else:
|
||
|
try:
|
||
|
# Same as above. We exit here instead.
|
||
|
_ = ircdb.users.getUser(options.name)
|
||
|
output('That user already exists. Try another name.')
|
||
|
sys.exit(-1)
|
||
|
except KeyError:
|
||
|
name = options.name
|
||
|
|
||
|
if not options.password:
|
||
|
password = getpass('What is %s\'s password? ' % name)
|
||
|
else:
|
||
|
password = options.password
|
||
|
|
||
|
if options.hashed is 0:
|
||
|
hashed = yn('Do you want the password to be hashed instead of '
|
||
|
'storing it as plain text?', default=False)
|
||
|
elif options.hashed is 1:
|
||
|
hashed = True
|
||
|
else:
|
||
|
hashed = False
|
||
|
|
||
|
if not options.capabilities:
|
||
|
capabilities = []
|
||
|
prompt = 'Would you like to give %s a capability?' % name
|
||
|
while yn(prompt):
|
||
|
capabilities.append(anything('What capability?'))
|
||
|
prompt = 'Would you like to give %s another capability?' % name
|
||
|
else:
|
||
|
capabilities = options.capabilities
|
||
|
|
||
|
user = ircdb.users.newUser()
|
||
|
user.name = name
|
||
|
user.setPassword(password, hashed=hashed)
|
||
|
for capability in capabilities:
|
||
|
user.addCapability(capability)
|
||
|
ircdb.users.setUser(user)
|
||
|
ircdb.users.flush()
|
||
|
#os.system('cat %s' % filename) # Was this here just for debugging?
|
||
|
ircdb.users.close()
|
||
|
print 'User %s added.' % name
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
try:
|
||
|
main()
|
||
|
except KeyboardInterrupt:
|
||
|
pass
|