feat: Adding filter_cdx and warc_by_cdx commands (2)

[malteos]

The PR adds two new commands to the CLI:

• filter_cdx: Filter CDX files based on a given URL or SURT white list.
• warc_by_cdx: Fetch WARC files like warc but instead of filtering based on a single URL this commands filters based on a given CDX file.

Both commands support reading and writing to local or remote paths (like S3) and gzip compression. To make the code more readable, parts of cli.py were moved to utils.py.

Example usage

Filter CDX files from S3 based on URL or SURT whitelist (input and output paths can be local or remote, e.g., S3):

# - one CDX file takes ~ 200 seconds with a whitelist of 1M SURTs)
# - for all 300 cdx files from CC-MAIN-2024-30 take  ~ 1.4 hrs with the EOTA2024 list and an c5n.xlarge instance
cdxt -v filter_cdx \
    s3://commoncrawl/cc-index/collections \
    s3://commoncrawl-dev/eot-archive/malte-test/ccf-gov-federal-web-graph-2024-jun-jul-aug.txt \
    s3://commoncrawl-dev/eot-archive/malte-test/filtered-cdxs --filter-type url \
    --input-glob "/CC-MAIN-2024-30/indexes/*.gz" --overwrite

Fetch WARC records based on filtered CDX (output is written to a local dir or S3)

# Input and output paths can be local or remote, e.g., S3.
cdxt -v --cc  warc_by_cdx \
    s3://commoncrawl-dev/eot-archive/malte-test/filtered-cdxs --index-glob "*.gz" \
    --prefix ./output/filtered-warcs/ \
    --warc-download-prefix=s3://commoncrawl \
   --creator foo --operator bob

For better throughput for S3 read/write, we have also a specific implementation using aioboto3 that you can enable with the --implementation=aioboto3 argument:

#  fetching takes ~ 10 hours for 3M records (filter from above) using an c5n.xlarge instance
cdxt -v --cc  warc_by_cdx \
    s3://commoncrawl-dev/eot-archive/malte-test/filtered-cdxs --index-glob "*.gz" \
    --prefix s3://commoncrawl-dev/eot-archive/malte-test/filtered-warcs/ \
    --warc-download-prefix=s3://commoncrawl \
   --creator foo --operator bob --implementation=aioboto3

To maintain the filter inputs, e.g., the whitelist list, we can add one or multiple files with metadata as resource records to the extracted WARC. To do this, you need to provide the corresponding file paths as arguments --write-paths-as-resource-records=s3://commoncrawl-dev/eot-archive/malte-test/ccf-gov-federal-web-graph-2024-jun-jul-aug.txt and --write-paths-as-resource-records-metadata=path/to/metadata.json. The metadata file is optional and can have the following optional fields:

- warc_content_type: str
- uri: str
- http_headers: dict
- warc_headers_dict: str

This in one example for a metadata JSON file:

{
    "uri": "filter_cdx.cdx.gz",
    "warc_content_type": "application/cdx",
}

The full WARC extraction command would look like this:

cdxt -v --cc  warc_by_cdx \
    s3://commoncrawl-dev/eot-archive/malte-test/filtered-cdxs --index-glob "*.gz" \
    --prefix s3://commoncrawl-dev/eot-archive/malte-test/filtered-warcs/ \
    --warc-download-prefix=s3://commoncrawl \
   --creator foo --operator bob \
   --write-paths-as-resource-records=s3://commoncrawl-dev/eot-archive/malte-test/ccf-gov-federal-web-graph-2024-jun-jul-aug.txt \
   --write-paths-as-resource-records-metadata=path/to/metadata.json

TODOs

• filter CDX: glob via fspec to support remote input paths, like S3.
• warc by CDX: support multiple input CDX files
• unit tests
• explicit parallel does not work
• formatting + linting (wait until feat: Adding linting with flake8 #49 is merged)

[codecov]

Codecov Report

❌ Patch coverage is 96.94750% with 25 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.35%. Comparing base (b6f5f02) to head (e1cd976).

Files with missing lines	Patch %	Lines
cdx_toolkit/filter_warc/warc_filter.py	92.51%	17 Missing ⚠️
cdx_toolkit/filter_warc/command.py	92.10%	3 Missing ⚠️
cdx_toolkit/warc.py	93.10%	2 Missing ⚠️
cdx_toolkit/cli.py	92.85%	1 Missing ⚠️
cdx_toolkit/commoncrawl.py	80.00%	1 Missing ⚠️
cdx_toolkit/utils.py	97.22%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #54      +/-   ##
==========================================
+ Coverage   95.78%   96.35%   +0.57%     
==========================================
  Files           7       22      +15     
  Lines         877     1645     +768     
==========================================
+ Hits          840     1585     +745     
- Misses         37       60      +23     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.