Core sampler: add ability to accept either a project code or an SDM name

• Core sampler now accepts either a project code or an SDM name

apps/cli/utilities/core_sampler/core_sampler/core_sampler.py

@@ -28,9 +28,6 @@ import argparse
 
 import psycopg2 as pg
 from psycopg2 import extras
-
-# pylint: disable=C0103, E0402, R0201
-
 from pycapo import CapoConfig
 
 from .database import PGTable
@@ -41,6 +38,8 @@ from .row_writer import (
     UniquifyingRowWriter,
 )
 
+# pylint: disable=C0103, E0402, R0201, R0903
+
 # stolen shamelessly from aat_wrest
 class MDDBConnector:
     """Use this connection to interrogate this science product locator"""
@@ -105,16 +104,18 @@ class CoreSampler:
         self.save(requested)
         self.writer.close()
 
-    def sample_eb(self, exec_block_id: int):
+    def sample_eb(self, sdm_name: str):
         """
         Pull execution block metadata from the archive database.
 
-        :param exec_block_id: execution_block_id of interest
+        :param sdm_name: SDM of interest
         :return:
         """
         ebs = self.table("execution_blocks")
 
-        requested = ebs.fetch({"execution_block_id": exec_block_id})
+        finder = ExecBlockFinder(self.connection)
+        eb_id = finder.find_eb_id(sdm_name)
+        requested = ebs.fetch({"execution_block_id": eb_id})
         self.save(requested)
         self.writer.close()
 
@@ -155,6 +156,23 @@ class CoreSampler:
         rows.write_to(self.writer)
 
 
+class ExecBlockFinder:
+    """Looks up execution block ID for an SDM"""
+
+    def __init__(self, connection: MDDBConnector):
+        self.connection = connection
+
+    def find_eb_id(self, sdm_name: str) -> int:
+        cursor = self.connection.cursor()
+        sql = """
+SELECT execution_block_id FROM execution_blocks
+WHERE ngas_fileset_id=%(sdm_name)s
+        """
+        cursor.execute(sql, {"sdm_name": sdm_name})
+        data = cursor.fetchone()
+        return data["execution_block_id"]
+
+
 def main():
     parser = argparse.ArgumentParser()
     group = parser.add_mutually_exclusive_group(required=True)
@@ -163,18 +181,18 @@ def main():
     )
     group.add_argument(
         "-e",
-        "--exec_block_id",
-        type=int,
+        "--sdm_name",
+        type=str,
         nargs=1,
-        help="execution_block_id of SDM for which to pull a core sample",
+        help="name of SDM for which to pull a core sample, e.g., 21A-409.sb39530397.eb39561636.59309.07888592593",
         action="store",
     )
     ns = parser.parse_args()
 
     if ns.project_code:
         CoreSampler(MDDBConnector()).sample_project(ns.project_code[0])
-    elif ns.exec_block_id:
-        CoreSampler(MDDBConnector()).sample_eb(int(ns.exec_block_id[0]))
+    elif ns.sdm_name:
+        CoreSampler(MDDBConnector()).sample_eb(ns.sdm_name[0])
 
     return 0