| #!/usr/bin/python |
| # Copyright (C) 2012 Google Inc. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """ |
| Configures git to authenticate to *.googlesource.com. |
| |
| Adds a cookie to ~/.git-credential-cache/cookies and sets |
| Git variable http.cookiefile to this path. |
| """ |
| |
| from __future__ import print_function |
| |
| import os |
| from subprocess import Popen, PIPE, check_call |
| import sys |
| |
| if sys.version_info[0] > 2: |
| # Python 3 imports |
| import http.cookiejar as cookielib |
| else: |
| # Python 2 imports |
| import cookielib |
| |
| try: |
| # We want to use raw_input in Python 2. |
| input = raw_input |
| except NameError: |
| # raw_input does not exist in in Python 3 and has been |
| # renamed to input. |
| pass |
| |
| def write_cookie(token): |
| git_config = ['git', 'config', '--global', 'http.cookiefile'] |
| set_config = False |
| |
| fn = Popen(git_config, stdout=PIPE).communicate()[0].decode('utf-8') |
| if fn == '': |
| home = os.environ['HOME'] |
| fn = os.path.join(home, '.git-credential-cache', 'cookies') |
| set_config = True |
| elif fn[-1] == '\n': |
| fn = fn[:-1] |
| |
| dir = os.path.dirname(fn) |
| if os.path.exists(dir): |
| os.chmod(dir, 0o700) |
| else: |
| os.mkdir(dir, 0o700) |
| |
| cj = cookielib.MozillaCookieJar(fn) |
| if os.path.exists(cj.filename): |
| cj.load() |
| cj.set_cookie(cookielib.Cookie( |
| version = 0, |
| name = 'o', |
| value = token, |
| port = None, |
| port_specified = False, |
| domain = '.googlesource.com', |
| domain_specified = True, |
| domain_initial_dot = True, |
| path = '/', |
| path_specified = True, |
| secure = True, |
| expires = (1 << 31) - 1, |
| discard = False, |
| comment = None, |
| comment_url = None, |
| rest = None)) |
| cj.save() |
| os.chmod(cj.filename, 0o600) |
| if set_config: |
| check_call(git_config + [cj.filename]) |
| return cj.filename |
| |
| def main(): |
| print('Open https://www.googlesource.com/new-password ...') |
| t = input("Password (under 'Login for Git'): ") |
| p = write_cookie(t) |
| print("Saved to %s" % (p,)) |
| |
| if __name__ == '__main__': |
| main() |