OLD | NEW |
1 # This file is part of the Adblock Plus web scripts, | 1 # This file is part of the Adblock Plus web scripts, |
2 # Copyright (C) 2006-present eyeo GmbH | 2 # Copyright (C) 2006-present eyeo GmbH |
3 # | 3 # |
4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
7 # | 7 # |
8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
(...skipping 10 matching lines...) Expand all Loading... |
21 import getpass | 21 import getpass |
22 import os | 22 import os |
23 | 23 |
24 from cms.translations.xtm.xtm_api import ( | 24 from cms.translations.xtm.xtm_api import ( |
25 XTMCloudException, get_token, XTMCloudAPI, | 25 XTMCloudException, get_token, XTMCloudAPI, |
26 ) | 26 ) |
27 import cms.translations.xtm.constants as const | 27 import cms.translations.xtm.constants as const |
28 from cms.translations.xtm.projects_handler import ( | 28 from cms.translations.xtm.projects_handler import ( |
29 create_project, upload_files, download_files, | 29 create_project, upload_files, download_files, |
30 ) | 30 ) |
31 from cms.translations.xtm.utils import input_fn, read_token | 31 from cms.translations.xtm.utils import input_fn, read_token, get_api_url |
32 from cms.sources import create_source | 32 from cms.sources import create_source |
33 | 33 |
34 | 34 |
35 def handle_projects(args): | 35 def handle_projects(args): |
36 try: | 36 try: |
37 api = XTMCloudAPI(read_token()) | 37 token = read_token() |
| 38 with create_source(args.source_dir, cached=True) as fs: |
| 39 api = XTMCloudAPI(token, get_api_url(args, fs)) |
| 40 args.projects_func(args, api, fs) |
38 except Exception as err: | 41 except Exception as err: |
39 sys.exit(err) | 42 sys.exit(err) |
40 with create_source(args.source_dir, cached=True) as fs: | |
41 args.projects_func(args, api, fs) | |
42 | 43 |
43 | 44 |
44 def generate_token(args): | 45 def generate_token(args): |
45 """Generate an API token from username and password.""" | 46 """Generate an API token from username and password.""" |
46 username = input_fn('Username: ') | 47 username = input_fn('Username: ') |
47 user_id = input_fn('User ID: ') | 48 user_id = input_fn('User ID: ') |
48 password = getpass.getpass(prompt='Pasword: ') | 49 password = getpass.getpass(prompt='Password: ') |
49 | 50 |
50 logging.info(const.InfoMessages.GENERATING_TOKEN.format(username, user_id)) | 51 logging.info(const.InfoMessages.GENERATING_TOKEN.format(username, user_id)) |
51 try: | 52 try: |
52 token = get_token(username, password, int(user_id)) | 53 token = get_token(username, password, int(user_id), |
| 54 get_api_url(args, None)) |
53 logging.info(const.InfoMessages.TOKEN_GENERATED.format(token)) | 55 logging.info(const.InfoMessages.TOKEN_GENERATED.format(token)) |
54 | 56 |
55 cmd = const.Token.SAVE_COMMAND.format(const.Token.ENV_VAR, token) | 57 cmd = const.Token.SAVE_COMMAND.format(const.Token.ENV_VAR, token) |
56 sys.stdout.write(const.InfoMessages.TOKEN_SAVE_TO_ENV_VAR.format(cmd)) | 58 sys.stdout.write(const.InfoMessages.TOKEN_SAVE_TO_ENV_VAR.format(cmd)) |
57 except XTMCloudException as err: | 59 except XTMCloudException as err: |
58 sys.exit(err) | 60 sys.exit(err) |
59 except Exception as err: | 61 except Exception as err: |
60 sys.exit(err) | 62 sys.exit(err) |
61 | 63 |
62 | 64 |
63 def parse_args(): | 65 def parse_args(): |
64 parser = argparse.ArgumentParser() | 66 parser = argparse.ArgumentParser() |
65 subparsers = parser.add_subparsers() | 67 subparsers = parser.add_subparsers() |
66 | 68 |
67 # Universal arguments | 69 # Universal arguments |
68 parser.add_argument('-v', '--verbose', action='store_true', | 70 parser.add_argument('-v', '--verbose', action='store_true', |
69 help=const.ArgumentsHelp.VERBOSE) | 71 help=const.ArgumentsHelp.VERBOSE) |
| 72 parser.add_argument('--api-url', help=const.ArgumentsHelp.API_URL) |
70 | 73 |
71 # Subparser for generating token | 74 # Subparser for generating token |
72 token_parser = subparsers.add_parser('login', | 75 token_parser = subparsers.add_parser('login', |
73 help=const.ArgumentsHelp.LOGIN) | 76 help=const.ArgumentsHelp.LOGIN) |
74 token_parser.set_defaults(func=generate_token) | 77 token_parser.set_defaults(func=generate_token) |
75 | 78 |
76 # Subparser for project creation. | 79 # Subparser for project creation. |
77 project_create_parser = subparsers.add_parser( | 80 project_create_parser = subparsers.add_parser( |
78 'create', | 81 'create', |
79 help=const.ArgumentsHelp.ProjectCreate.MAIN, | 82 help=const.ArgumentsHelp.ProjectCreate.MAIN, |
(...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
150 ) | 153 ) |
151 | 154 |
152 return parser.parse_args() | 155 return parser.parse_args() |
153 | 156 |
154 | 157 |
155 def main(): | 158 def main(): |
156 """Run XTM integration script.""" | 159 """Run XTM integration script.""" |
157 args = parse_args() | 160 args = parse_args() |
158 | 161 |
159 if args.verbose: | 162 if args.verbose: |
160 logging.basicConfig(level=logging.INFO) | 163 logging.basicConfig(level=logging.INFO, stream=sys.stderr) |
161 | 164 |
162 args.func(args) | 165 args.func(args) |
163 | 166 |
164 | 167 |
165 if __name__ == '__main__': | 168 if __name__ == '__main__': |
166 main() | 169 main() |
OLD | NEW |