6 from autotest_lib
.client
.common_lib
import global_config
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]
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")
20 con
= MySQLdb
.connect(host
=host
, user
=user
,
21 passwd
=password
, db
=db_name
)
24 for username
in sys
.argv
[1:]:
28 WHERE login = %s""", username
)
32 print "User %s does not exist. Creating..." % username
34 INSERT INTO afe_users (login, access_level)
35 VALUES (%s, 100)""", username
)
38 print "Updating user %s..." % username
41 SET access_level = 100
42 WHERE login = %s""", username
)
43 if (cur
.rowcount
== 1):
46 print " %s is already a superuser!" % username