|
| 1 | +#!/usr/bin/python3 |
| 2 | + |
| 3 | +# iterate over monthly crawls and store |
| 4 | +# the joined data as a partition of the result table |
| 5 | + |
| 6 | +import logging |
| 7 | +import re |
| 8 | +import sys |
| 9 | + |
| 10 | +from pyathena import connect |
| 11 | + |
| 12 | +logging.basicConfig(level='INFO', |
| 13 | + format='%(asctime)s %(levelname)s %(name)s: %(message)s') |
| 14 | + |
| 15 | +join_template = """ |
| 16 | +CREATE TABLE {db}._tmp_overlap |
| 17 | +WITH (external_location = '{s3_location}/crawl={crawl}/', |
| 18 | + partitioned_by = ARRAY['subset'], |
| 19 | + format = 'PARQUET', |
| 20 | + parquet_compression = 'GZIP') |
| 21 | +AS SELECT |
| 22 | + {tid}.id AS id, |
| 23 | + {tid}.title AS title, |
| 24 | + {tid}.link AS link, |
| 25 | + {tid}.language AS language, |
| 26 | + {tid}.url_surtkey AS url_surtkey_prefix, |
| 27 | + cc.url_surtkey AS url_surtkey, |
| 28 | + cc.url_host_tld AS url_host_tld, |
| 29 | + cc.url_host_registered_domain AS url_host_registered_domain, |
| 30 | + cc.url_host_name AS url_host_name, |
| 31 | + cc.url AS url, |
| 32 | + cc.fetch_status AS fetch_status, |
| 33 | + cc.fetch_time AS fetch_time, |
| 34 | + cc.warc_filename AS warc_filename, |
| 35 | + cc.warc_record_offset AS warc_record_offset, |
| 36 | + cc.warc_record_length AS warc_record_length, |
| 37 | + cc.fetch_redirect AS fetch_redirect, |
| 38 | + cc.content_mime_detected AS content_mime_detected, |
| 39 | + cc.content_languages AS content_languages, |
| 40 | + cc.subset AS subset |
| 41 | +FROM ccindex.ccindex AS cc |
| 42 | + RIGHT OUTER JOIN {db}.{seed_table} AS {tid} |
| 43 | + ON cc.url_host_registered_domain = {tid}.url_host_registered_domain |
| 44 | + AND strpos(cc.url_surtkey, {tid}.url_surtkey) = 1 |
| 45 | +WHERE cc.crawl = '{crawl}' |
| 46 | +""" |
| 47 | + |
| 48 | +drop_tmp_table = 'DROP TABLE `{db}._tmp_overlap`;' |
| 49 | + |
| 50 | +# list of crawls |
| 51 | +# Note: in order to get a list of released crawls: |
| 52 | +# - query Athena |
| 53 | +# SHOW PARTITIONS ccindex |
| 54 | +# - see |
| 55 | +# https://commoncrawl.s3.amazonaws.com/crawl-data/index.html |
| 56 | +crawls = [ |
| 57 | + 'CC-MAIN-2013-20', 'CC-MAIN-2013-48', |
| 58 | + # |
| 59 | + 'CC-MAIN-2014-10', 'CC-MAIN-2014-15', 'CC-MAIN-2014-23', |
| 60 | + 'CC-MAIN-2014-35', 'CC-MAIN-2014-41', 'CC-MAIN-2014-42', |
| 61 | + 'CC-MAIN-2014-49', 'CC-MAIN-2014-52', |
| 62 | + # |
| 63 | + 'CC-MAIN-2015-06', 'CC-MAIN-2015-11', 'CC-MAIN-2015-14', |
| 64 | + 'CC-MAIN-2015-18', 'CC-MAIN-2015-22', 'CC-MAIN-2015-27', |
| 65 | + 'CC-MAIN-2015-32', 'CC-MAIN-2015-35', 'CC-MAIN-2015-40', |
| 66 | + 'CC-MAIN-2015-48', |
| 67 | + # |
| 68 | + 'CC-MAIN-2016-07', 'CC-MAIN-2016-18', 'CC-MAIN-2016-22', |
| 69 | + 'CC-MAIN-2016-26', 'CC-MAIN-2016-30', 'CC-MAIN-2016-36', |
| 70 | + 'CC-MAIN-2016-40', 'CC-MAIN-2016-44', 'CC-MAIN-2016-50', |
| 71 | + # |
| 72 | + 'CC-MAIN-2017-04', 'CC-MAIN-2017-09', 'CC-MAIN-2017-13', |
| 73 | + 'CC-MAIN-2017-17', 'CC-MAIN-2017-22', 'CC-MAIN-2017-26', |
| 74 | + 'CC-MAIN-2017-30', 'CC-MAIN-2017-34', 'CC-MAIN-2017-39', |
| 75 | + 'CC-MAIN-2017-43', 'CC-MAIN-2017-47', 'CC-MAIN-2017-51', |
| 76 | + # |
| 77 | + 'CC-MAIN-2018-05', 'CC-MAIN-2018-09', 'CC-MAIN-2018-13', |
| 78 | + 'CC-MAIN-2018-17', 'CC-MAIN-2018-22', 'CC-MAIN-2018-26', |
| 79 | + 'CC-MAIN-2018-30', 'CC-MAIN-2018-34', 'CC-MAIN-2018-39', |
| 80 | + 'CC-MAIN-2018-43', 'CC-MAIN-2018-47', 'CC-MAIN-2018-51', |
| 81 | + # |
| 82 | + 'CC-MAIN-2019-04', 'CC-MAIN-2019-09', 'CC-MAIN-2019-13', |
| 83 | + 'CC-MAIN-2019-18', 'CC-MAIN-2019-22', 'CC-MAIN-2019-26', |
| 84 | + 'CC-MAIN-2019-30', 'CC-MAIN-2019-35', 'CC-MAIN-2019-39', |
| 85 | + 'CC-MAIN-2019-43', 'CC-MAIN-2019-47', 'CC-MAIN-2019-51', |
| 86 | + # |
| 87 | + 'CC-MAIN-2020-05', 'CC-MAIN-2020-10', 'CC-MAIN-2020-16', |
| 88 | + 'CC-MAIN-2020-24', 'CC-MAIN-2020-29', 'CC-MAIN-2020-34', |
| 89 | + 'CC-MAIN-2020-40', 'CC-MAIN-2020-45', 'CC-MAIN-2020-50', |
| 90 | + # |
| 91 | + 'CC-MAIN-2021-04', 'CC-MAIN-2021-10', 'CC-MAIN-2021-17', |
| 92 | + 'CC-MAIN-2021-21', 'CC-MAIN-2021-25', 'CC-MAIN-2021-31', |
| 93 | + 'CC-MAIN-2021-39', 'CC-MAIN-2021-43', 'CC-MAIN-2021-49', |
| 94 | + # |
| 95 | +] |
| 96 | + |
| 97 | + |
| 98 | +s3_location = sys.argv[1] |
| 99 | +s3_location = s3_location.rstrip('/') # no trailing slash! |
| 100 | + |
| 101 | +seed_table = sys.argv[2] |
| 102 | + |
| 103 | +crawl_selector = re.compile(sys.argv[3], re.IGNORECASE) |
| 104 | + |
| 105 | + |
| 106 | +crawls = filter(lambda c: crawl_selector.match(c), crawls) |
| 107 | + |
| 108 | + |
| 109 | +cursor = connect(s3_staging_dir="{}/staging".format(s3_location), |
| 110 | + region_name="us-east-1").cursor() |
| 111 | + |
| 112 | +for crawl in crawls: |
| 113 | + query = join_template.format(crawl=crawl, |
| 114 | + s3_location="{}/cc".format(s3_location), |
| 115 | + db='bigscience', |
| 116 | + seed_table=seed_table, |
| 117 | + tid='bs') |
| 118 | + logging.info("Athena query: %s", query) |
| 119 | + |
| 120 | + cursor.execute(query) |
| 121 | + logging.info("Athena query ID %s: %s", cursor.query_id, cursor.result_set.state) |
| 122 | + logging.info(" data_scanned_in_bytes: %d", cursor.result_set.data_scanned_in_bytes) |
| 123 | + logging.info(" total_execution_time_in_millis: %d", cursor.result_set.total_execution_time_in_millis) |
| 124 | + |
| 125 | + cursor.execute(drop_tmp_table.format(db='bigscience')) |
| 126 | + logging.info("Drop temporary table: %s", cursor.result_set.state) |
| 127 | + |
| 128 | + |
0 commit comments