520 write an archiver for doe low income energy affordability data lead

src/pudl_archiver/archivers/classes.py

@@ -223,6 +223,7 @@ async def get_hyperlinks(
         url: str,
         filter_pattern: typing.Pattern | None = None,
         verify: bool = True,
+        headers: dict | None = None,
     ) -> list[str]:
         """Return all hyperlinks from a specific web page.
 
@@ -235,12 +236,18 @@ async def get_hyperlinks(
             url: URL of web page.
             filter_pattern: If present, only return links that contain pattern.
             verify: Verify ssl certificate (EPACEMS https source has bad certificate).
+            headers: Additional headers to send in the GET request.
         """
         # Parse web page to get all hyperlinks
         parser = _HyperlinkExtractor()
 
         response = await retry_async(
-            self.session.get, args=[url], kwargs={"ssl": verify}
+            self.session.get,
+            args=[url],
+            kwargs={
+                "ssl": verify,
+                **({"headers": headers} if headers is not None else {}),
+            },
         )
         text = await retry_async(response.text)
         parser.feed(text)
@@ -253,8 +260,8 @@ async def get_hyperlinks(
         # Warn if no links are found
         if not hyperlinks:
             self.logger.warning(
-                f"The archiver couldn't find any hyperlinks that match {filter_pattern}."
-                f"Make sure your filter_pattern is correct or if the structure of the {url} page changed."
+                f"The archiver couldn't find any hyperlinks{('that match: ' + filter_pattern.pattern) if filter_pattern else ''}."
+                f"Make sure your filter_pattern is correct, check if the structure of the {url} page changed, or if you are missing HTTP headers."
             )
 
         return hyperlinks

src/pudl_archiver/archivers/doelead.py

@@ -0,0 +1,138 @@
+"""Download DOE LEAD data.
+
+Each partition includes:
+- Data Dictionary
+- Census Tracts List
+- Cities List
+- Counties List
+- States List
+- Tribal Areas List
+- Cities Census Track Overlaps
+- Tribal Areas Tract Overlaps
+- One .zip file per state, each of which includes:
+  - AMI Census Tracts
+  - SMI Census Tracts
+  - LLSI Census Tracts
+  - FPL Census Tracts
+  - LLSI Counties
+  - SMI Counties
+  - FPL Counties
+  - AMI Counties
+"""
+
+import re
+
+from pudl_archiver.archivers.classes import (
+    AbstractDatasetArchiver,
+    ArchiveAwaitable,
+    ResourceInfo,
+)
+from pudl_archiver.frictionless import ZipLayout
+
+# This site is no longer online as of 01/28/2025.
+# TOOL_URL = "https://www.energy.gov/scep/low-income-energy-affordability-data-lead-tool"
+
+YEARS_DOIS = {
+    2022: "https://doi.org/10.25984/2504170",
+    2018: "https://doi.org/10.25984/1784729",
+}
+
+# verified working 2025-01-22 via
+# $ wget "https://www.energy.gov/scep/low-income-energy-affordability-data-lead-tool" -O foo.html -U "Mozilla/5.0 Catalyst/2025 Cooperative/2025"
+HEADERS = {"User-Agent": "Mozilla/5.0 Catalyst/2025 Cooperative/2025"}
+
+
+class DoeLeadArchiver(AbstractDatasetArchiver):
+    """DOE LEAD archiver."""
+
+    name = "doelead"
+
+    async def get_resources(self) -> ArchiveAwaitable:
+        """Download DOE LEAD resources.
+
+        The DOE LEAD Tool is down as of 01/28/2025. It didn't provide direct access
+        to the raw data, but instead linked to the current raw data release hosted on
+        OEDI. It did not provide links to past data releases. So, we hard-code the
+        DOIs for all known releases and archive those. Based on the removal of the main
+        page, it's safe to assume this won't be updated any time soon. If it is, we'll
+        need to manually update the DOIs.
+        """
+        # e.g.: https://data.openei.org/files/6219/DC-2022-LEAD-data.zip
+        #       https://data.openei.org/files/6219/Data%20Dictionary%202022.xlsx
+        #       https://data.openei.org/files/6219/LEAD%20Tool%20States%20List%202022.xlsx
+        data_link_pattern = re.compile(r"([^/]+(\d{4})(?:-LEAD-data.zip|.xlsx))")
+        """Regex for matching the data files in a release on the OEDI page. Captures the year, and supports both .zip and .xlsx file names."""
+
+        for year, doi in YEARS_DOIS.items():
+            self.logger.info(f"Processing DOE LEAD raw data release for {year}: {doi}")
+            filenames_links = {}
+            for data_link in await self.get_hyperlinks(doi, data_link_pattern):
+                matches = data_link_pattern.search(data_link)
+                if not matches:
+                    continue
+                link_year = int(matches.group(2))
+                if link_year != year:
+                    raise AssertionError(
+                        f"We expect all files at {doi} to be for {year}, but we found: {link_year} from {data_link}"
+                    )
+                filenames_links[matches.group(1)] = data_link
+            if filenames_links:
+                self.logger.info(f"Downloading: {year}, {len(filenames_links)} items")
+                yield self.get_year_resource(filenames_links, year)
+
+        # Download LEAD methodology PDF and other metadata separately
+        metadata_links = {
+            "lead-methodology-122024.pdf": "https://www.energy.gov/sites/default/files/2024-12/lead-methodology_122024.pdf",
+            "lead-tool-factsheet-072624.pdf": "https://www.energy.gov/sites/default/files/2024-07/lead-tool-factsheet_072624.pdf",
+        }
+        for filename, link in metadata_links.items():
+            yield self.get_metadata_resource(filename=filename, link=link)
+
+    async def get_year_resource(self, links: dict[str, str], year: int) -> ResourceInfo:
+        """Download all available data for a year.
+
+        Resulting resource contains one zip file of CSVs per state/territory, plus a handful of .xlsx dictionary and geocoding files.
+
+        Args:
+            links: filename->URL mapping for files to download
+            year: the year we're downloading data for
+        """
+        host = "https://data.openei.org"
+        zip_path = self.download_directory / f"doelead-{year}.zip"
+        data_paths_in_archive = set()
+        for filename, link in sorted(links.items()):
+            self.logger.info(f"Downloading {link}")
+            download_path = self.download_directory / filename
+            await self.download_file(f"{host}{link}", download_path)
+            self.add_to_archive(
+                zip_path=zip_path,
+                filename=filename,
+                blob=download_path.open("rb"),
+            )
+            data_paths_in_archive.add(filename)
+            # Don't want to leave multiple giant files on disk, so delete
+            # immediately after they're safely stored in the ZIP
+            download_path.unlink()
+        return ResourceInfo(
+            local_path=zip_path,
+            partitions={"year": year},
+            layout=ZipLayout(file_paths=data_paths_in_archive),
+        )
+
+    async def get_metadata_resource(self, filename: str, link: str) -> ResourceInfo:
+        """Download metadata resource.
+
+        Resulting resource contains one PDF file with metadata about the LEAD dataset.
+
+        Args:
+            links: filename->URL mapping for files to download
+            year: the year we're downloading data for
+        """
+        self.logger.info(f"Downloading {link}")
+        download_path = self.download_directory / filename
+        await self.download_file(url=link, file_path=download_path, headers=HEADERS)
+
+        return ResourceInfo(
+            local_path=download_path,
+            partitions={},
+        )

src/pudl_archiver/package_data/zenodo_doi.yaml

@@ -7,6 +7,9 @@ censuspep:
 doeiraec:
   production_doi: 10.5281/zenodo.14757121
   sandbox_doi: 10.5072/zenodo.157934
+doelead:
+  production_doi: 10.5281/zenodo.14758684
+  # sandbox_doi: TODO once server 413 error resolves
 eia176:
   production_doi: 10.5281/zenodo.7682357
   sandbox_doi: 10.5072/zenodo.3158