Limnoria/scripts/supybot-adduser

135 lines
5.1 KiB
Plaintext
Raw Normal View History

#!/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.
###
2003-10-02 07:01:44 +02:00
import supybot
2003-10-05 15:02:56 +02:00
import fix
from questions import *
import os
2003-10-02 07:01:44 +02:00
import sys
import optparse
def main():
2004-02-07 13:25:56 +01:00
import log
import conf
conf.supybot.log.stdout.setValue(False)
2004-02-07 23:23:12 +01:00
#conf.supybot.log.level.set('DEBUG')
2004-02-07 13:25:56 +01:00
parser = optparse.OptionParser(usage='Usage: %prog [options] <configFile>',
2004-02-07 13:25:56 +01:00
version='supybot %s' % conf.version)
2003-10-02 07:01:44 +02:00
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.')
2003-10-02 07:01:44 +02:00
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 configuration file you\'d like to use.')
2004-02-07 23:23:12 +01:00
filename = os.path.abspath(args[0])
conf.supybot.directories.conf.setValue('/')
conf.supybot.databases.users.filename.setValue(filename)
import ircdb
2003-10-02 07:01:44 +02:00
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
2003-10-02 07:01:44 +02:00
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
2003-10-02 07:01:44 +02:00
if not options.password:
2003-11-17 05:13:06 +01:00
password = getpass('What is %s\'s password? ' % name)
2003-10-02 07:01:44 +02:00
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
2003-10-02 07:01:44 +02:00
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
2003-10-02 07:01:44 +02:00
else:
capabilities = options.capabilities
(id, user) = ircdb.users.newUser()
user.name = name
user.setPassword(password, hashed=hashed)
2003-10-02 07:01:44 +02:00
for capability in capabilities:
user.addCapability(capability)
ircdb.users.setUser(id, user)
2004-02-07 23:23:12 +01:00
ircdb.users.flush()
os.system('cat %s' % filename)
ircdb.users.close()
2003-10-02 07:01:44 +02:00
print 'User %s added.' % name
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
pass