frontend/make_superuser.py

   1 #!/usr/bin/python
   2
   3 import common
   4 import MySQLdb
   5 import sys
   6 from autotest_lib.client.common_lib import global_config
   7
   8 if (len(sys.argv) < 2 or
   9     [arg for arg in sys.argv[1:] if arg.startswith('-')]):
  10     print "Usage: %s username [username ...]" %sys.argv[0]
  11     sys.exit(1)
  12
  13 config = global_config.global_config
  14 section = 'AUTOTEST_WEB'
  15 host = config.get_config_value(section, "host")
  16 db_name = config.get_config_value(section, "database")
  17 user = config.get_config_value(section, "user")
  18 password = config.get_config_value(section, "password")
  19
  20 con = MySQLdb.connect(host=host, user=user,
  21                       passwd=password, db=db_name)
  22 cur = con.cursor()
  23
  24 for username in sys.argv[1:]:
  25     cur.execute("""
  26         SELECT access_level
  27         FROM afe_users
  28         WHERE login = %s""", username)
  29     row = cur.fetchone()
  30
  31     if row is None:
  32         print "User %s does not exist. Creating..." % username
  33         cur.execute("""
  34             INSERT INTO afe_users (login, access_level)
  35             VALUES (%s, 100)""", username)
  36         print "    Done"
  37     else:
  38         print "Updating user %s..." % username
  39         cur.execute("""
  40             UPDATE afe_users
  41             SET access_level = 100
  42             WHERE login = %s""", username)
  43         if (cur.rowcount == 1):
  44             print "    Done"
  45         else:
  46             print "    %s is already a superuser!" % username
  47
  48 cur.close()
  49 con.commit()
  50 con.close()