http://www.mediawiki.org/wiki/Special:Code/pywikipedia/11576 Revision: 11576 Author: valhallasw Date: 2013-05-24 15:10:37 +0000 (Fri, 24 May 2013) Log Message: ----------- + allow multiple users to be created in the user-config Modified Paths: -------------- branches/rewrite/generate_user_files.py Modified: branches/rewrite/generate_user_files.py =================================================================== --- branches/rewrite/generate_user_files.py 2013-05-24 14:52:50 UTC (rev 11575) +++ branches/rewrite/generate_user_files.py 2013-05-24 15:10:37 UTC (rev 11576) @@ -133,22 +133,29 @@ return True return False -def create_user_config(): - _fnc = os.path.join(base_dir, "user-config.py") - if not file_exists(_fnc): +def get_site_and_lang(): known_families = re.findall(r'(.+)_family.py\b', '\n'.join(os.listdir( os.path.join(pywikibot_dir, "pywikibot", "families")))) + known_families = sorted(known_families) fam = listchoice(known_families, "Select family of sites we are working on", default='wikipedia') mylang = raw_input( "The language code of the site we're working on (default: 'en'): ") or 'en' username = raw_input("Username (%s %s): " - % (mylang, fam)) or 'UnnamedBot' + % (mylang, fam)) username = unicode(username, console_encoding) + return fam, mylang, username + +def create_user_config(): + _fnc = os.path.join(base_dir, "user-config.py") + if not file_exists(_fnc): + fam, mylang, mainusername = get_site_and_lang() + mainusername = mainusername or "UnnamedBot" + while True: choice = raw_input( "Which variant of user_config.py:\n[S]mall or [E]xtended (with further information)? " @@ -200,13 +207,18 @@ usernames['%s']['%s'] = u'%s' -%s""" % (fam, mylang, fam, mylang, username, config_text)) +%s""" % (fam, mylang, fam, mylang, mainusername, config_text)) else: f.write("""# -*- coding: utf-8 -*- family = '%s' mylang = '%s' usernames['%s']['%s'] = u'%s' -""" % (fam, mylang, fam, mylang, username)) +""" % (fam, mylang, fam, mylang, mainusername)) + while(raw_input("Do you want to add any other projects? (y/N)").upper() == "Y"): + fam, mylang, username = get_site_and_lang() + username = username or mainusername + f.write("usernames['%(fam)s']['%(mylang)s'] = u'%(username)s'\n") + f.close() print("'%s' written." % _fnc)