def test_arguments(self): parser = argparse.ArgumentParser() antibody.add_argparse_options(parser) args = parser.parse_args([ '--cache-path', '/ab', '--git-checkout-path', '/bcd', '--sql-password-file', 'abd.txt', '--write-html', '--run-antibody', '--parse-git-rietveld', '--output-dir-path', '/efg', '--since', '2000', '--database', 'db_name', '--repo-list', 'db_name1', 'db_name2' ]) self.assertEqual(args.cache_path, '/ab') self.assertEqual(args.git_checkout_path, '/bcd') self.assertEqual(args.sql_password_file, 'abd.txt') self.assertTrue(args.write_html) self.assertTrue(args.run_antibody) self.assertTrue(args.parse_git_rietveld) self.assertEqual(args.output_dir_path, '/efg') self.assertEqual(args.since, '2000') self.assertEqual(args.database, 'db_name') self.assertEqual(args.repo_list, ['db_name1', 'db_name2']) args = parser.parse_args([ '-c', '/ab', '-g', '/bcd', '-p', 'abd.txt', '-w', '-a', '-r', '-d', '/efg', '-s', '2000', '-b', 'db_name', '-l', 'db_name1', 'db_name2' ]) self.assertEqual(args.cache_path, '/ab') self.assertEqual(args.git_checkout_path, '/bcd') self.assertEqual(args.sql_password_file, 'abd.txt') self.assertTrue(args.write_html) self.assertTrue(args.run_antibody) self.assertTrue(args.parse_git_rietveld) self.assertEqual(args.output_dir_path, '/efg') self.assertEqual(args.since, '2000') self.assertEqual(args.database, 'db_name') self.assertEqual(args.repo_list, ['db_name1', 'db_name2'])
def test_arguments(self): parser = argparse.ArgumentParser() antibody.add_argparse_options(parser) args = parser.parse_args(['--cache-path', '/ab', '--git-checkout-path', '/bcd', '--sql-password-file', 'abd.txt','--write-html', '--run-antibody', '--parse-git-rietveld', '--output-dir-path', '/efg', '--since', '2000', '--database', 'db_name', '--repo-list', 'db_name1', 'db_name2']) self.assertEqual(args.cache_path, '/ab') self.assertEqual(args.git_checkout_path, '/bcd') self.assertEqual(args.sql_password_file, 'abd.txt') self.assertTrue(args.write_html) self.assertTrue(args.run_antibody) self.assertTrue(args.parse_git_rietveld) self.assertEqual(args.output_dir_path, '/efg') self.assertEqual(args.since, '2000') self.assertEqual(args.database, 'db_name') self.assertEqual(args.repo_list, ['db_name1', 'db_name2']) args = parser.parse_args(['-c', '/ab', '-g', '/bcd', '-p', 'abd.txt', '-w', '-a', '-r', '-d', '/efg', '-s', '2000', '-b', 'db_name', '-l', 'db_name1', 'db_name2']) self.assertEqual(args.cache_path, '/ab') self.assertEqual(args.git_checkout_path, '/bcd') self.assertEqual(args.sql_password_file, 'abd.txt') self.assertTrue(args.write_html) self.assertTrue(args.run_antibody) self.assertTrue(args.parse_git_rietveld) self.assertEqual(args.output_dir_path, '/efg') self.assertEqual(args.since, '2000') self.assertEqual(args.database, 'db_name') self.assertEqual(args.repo_list, ['db_name1', 'db_name2'])
def main(argv): parser = argparse.ArgumentParser( prog="antibody", description=sys.modules['__main__'].__doc__) antibody.add_argparse_options(parser) infra_libs.logs.add_argparse_options(parser) args = parser.parse_args(argv) infra_libs.logs.process_argparse_options(args) if args.cache_path: requests_cache.install_cache(args.cache_path) else: requests_cache.install_cache(os.path.join(DATA_DIR, 'rietveld_cache')) # Do more processing here LOGGER.info('Antibody starting') with open(args.sql_password_file, 'r') as f: password = f.read().strip() connection, cc = csql.connect(password, args.database) checkout = args.git_checkout_path if args.parse_git_rietveld or args.run_antibody: antibody.setup_antibody_db(cc, os.path.join( DATA_DIR, 'ANTIBODY_DB_schema_setup.sql'), args.database) git_commit_parser.upload_to_sql(cc, checkout, args.since) git_commits_with_review_urls = git_commit_parser.get_urls_from_git_commit( cc) for review_url in git_commits_with_review_urls: # cannot get access into chromereview.googleplex.com if not any(host in review_url for host in ( 'chromereviews.googleplex', )): code_review_parse.add_code_review_data_to_db(review_url, cc, checkout) csql.commit(connection) if args.write_html or args.run_antibody: if not os.path.exists(args.output_dir_path): os.makedirs(args.output_dir_path) gitiles_prefix = antibody.get_gitiles_prefix(checkout) if not gitiles_prefix: gitiles_prefix = '' project_name = antibody.get_project_name(checkout).lower() antibody.generate_antibody_ui(cc, gitiles_prefix, project_name, args.since, args.output_dir_path, code_review_parse.get_tbr_no_lgtm(cc, 'author'), args.repo_list) csql.close(connection, cc)
def main(argv): parser = argparse.ArgumentParser( prog="antibody", description=sys.modules['__main__'].__doc__) antibody.add_argparse_options(parser) infra_libs.logs.add_argparse_options(parser) args = parser.parse_args(argv) infra_libs.logs.process_argparse_options(args) if args.cache_path: requests_cache.install_cache(args.cache_path) else: requests_cache.install_cache(os.path.join(DATA_DIR, 'rietveld_cache')) # Do more processing here LOGGER.info('Antibody starting') with open(args.sql_password_file, 'r') as f: password = f.read().strip() connection, cc = csql.connect(password, args.database) checkout = args.git_checkout_path if args.parse_git_rietveld or args.run_antibody: antibody.setup_antibody_db( cc, os.path.join(DATA_DIR, 'ANTIBODY_DB_schema_setup.sql'), args.database) git_commit_parser.upload_to_sql(cc, checkout, args.since) git_commits_with_review_urls = git_commit_parser.get_urls_from_git_commit( cc) for review_url in git_commits_with_review_urls: # cannot get access into chromereview.googleplex.com if not any(host in review_url for host in ('chromereviews.googleplex', )): code_review_parse.add_code_review_data_to_db( review_url, cc, checkout) csql.commit(connection) if args.write_html or args.run_antibody: if not os.path.exists(args.output_dir_path): os.makedirs(args.output_dir_path) gitiles_prefix = antibody.get_gitiles_prefix(checkout) if not gitiles_prefix: gitiles_prefix = '' project_name = antibody.get_project_name(checkout).lower() antibody.generate_antibody_ui( cc, gitiles_prefix, project_name, args.since, args.output_dir_path, code_review_parse.get_tbr_no_lgtm(cc, 'author'), args.repo_list) csql.close(connection, cc)