| Index: remoting/tools/gettoken.py
|
| diff --git a/remoting/tools/gettoken.py b/remoting/tools/gettoken.py
|
| index a36408bd009b8ccc5cabf6a0fe2a18e024213769..4f867bfcc476ed73ca4e5e9e359d1af0bab57a46 100755
|
| --- a/remoting/tools/gettoken.py
|
| +++ b/remoting/tools/gettoken.py
|
| @@ -1,11 +1,12 @@
|
| #!/usr/bin/env python
|
| -#
|
| -# Copyright (c) 2010 The Chromium Authors. All rights reserved.
|
| +# Copyright (c) 2011 The Chromium Authors. All rights reserved.
|
| # Use of this source code is governed by a BSD-style license that can be
|
| # found in the LICENSE file.
|
| -#
|
| -# gettoken.py can be used to get auth token from Gaia. It asks username and
|
| -# password and then prints token on the screen.
|
| +
|
| +"""Get auth token from Gaia.
|
| +
|
| +It asks username and password and then prints token on the screen.
|
| +"""
|
|
|
| import getpass
|
| import os
|
| @@ -13,46 +14,53 @@ import urllib
|
|
|
| import gaia_auth
|
|
|
| -chromoting_auth_filepath = os.path.join(os.path.expanduser('~'),
|
| - '.chromotingAuthToken')
|
| -chromoting_dir_auth_filepath = os.path.join(os.path.expanduser('~'),
|
| - '.chromotingDirectoryAuthToken')
|
| -
|
| -print "Email:",
|
| -email = raw_input()
|
| -
|
| -passwd = getpass.getpass("Password: ")
|
| -
|
| -chromoting_authenticator = gaia_auth.GaiaAuthenticator('chromiumsync');
|
| -chromoting_auth_token = chromoting_authenticator.authenticate(email, passwd)
|
| -
|
| -chromoting_dir_authenticator = gaia_auth.GaiaAuthenticator('chromoting');
|
| -chromoting_dir_auth_token = \
|
| - chromoting_dir_authenticator.authenticate(email, passwd)
|
| -
|
| -# Set permission mask for created files.
|
| -os.umask(0066)
|
| -
|
| -chromoting_auth_file = open(chromoting_auth_filepath, 'w')
|
| -chromoting_auth_file.write(email)
|
| -chromoting_auth_file.write('\n')
|
| -chromoting_auth_file.write(chromoting_auth_token)
|
| -chromoting_auth_file.close()
|
| -
|
| -print
|
| -print 'Chromoting (sync) Auth Token:'
|
| -print
|
| -print chromoting_auth_token
|
| -print '...saved in', chromoting_auth_filepath
|
| -
|
| -chromoting_dir_auth_file = open(chromoting_dir_auth_filepath, 'w')
|
| -chromoting_dir_auth_file.write(email)
|
| -chromoting_dir_auth_file.write('\n')
|
| -chromoting_dir_auth_file.write(chromoting_dir_auth_token)
|
| -chromoting_dir_auth_file.close()
|
| -
|
| -print
|
| -print 'Chromoting Directory Auth Token:'
|
| -print
|
| -print chromoting_dir_auth_token
|
| -print '...saved in', chromoting_dir_auth_filepath
|
| +
|
| +def main():
|
| + basepath = os.path.expanduser('~')
|
| + chromoting_auth_filepath = os.path.join(basepath, '.chromotingAuthToken')
|
| + chromoting_dir_auth_filepath = os.path.join(
|
| + basepath, '.chromotingDirectoryAuthToken')
|
| +
|
| + print "Email:",
|
| + email = raw_input()
|
| +
|
| + passwd = getpass.getpass("Password: ")
|
| +
|
| + chromoting_authenticator = gaia_auth.GaiaAuthenticator('chromiumsync');
|
| + chromoting_auth_token = chromoting_authenticator.authenticate(email, passwd)
|
| +
|
| + chromoting_dir_authenticator = gaia_auth.GaiaAuthenticator('chromoting');
|
| + chromoting_dir_auth_token = chromoting_dir_authenticator.authenticate(
|
| + email, passwd)
|
| +
|
| + # Set permission mask for created files.
|
| + os.umask(0066)
|
| +
|
| + chromoting_auth_file = open(chromoting_auth_filepath, 'w')
|
| + chromoting_auth_file.write(email)
|
| + chromoting_auth_file.write('\n')
|
| + chromoting_auth_file.write(chromoting_auth_token)
|
| + chromoting_auth_file.close()
|
| +
|
| + print
|
| + print 'Chromoting (sync) Auth Token:'
|
| + print
|
| + print chromoting_auth_token
|
| + print '...saved in', chromoting_auth_filepath
|
| +
|
| + chromoting_dir_auth_file = open(chromoting_dir_auth_filepath, 'w')
|
| + chromoting_dir_auth_file.write(email)
|
| + chromoting_dir_auth_file.write('\n')
|
| + chromoting_dir_auth_file.write(chromoting_dir_auth_token)
|
| + chromoting_dir_auth_file.close()
|
| +
|
| + print
|
| + print 'Chromoting Directory Auth Token:'
|
| + print
|
| + print chromoting_dir_auth_token
|
| + print '...saved in', chromoting_dir_auth_filepath
|
| + return 0
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + sys.exit(main())
|
|
|