Add get_schema_id to Schema class

lsst · Aug 21, 2024 · 0ea4f4e · 0ea4f4e
1 parent fac0a32
commit 0ea4f4e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/python/lsst/alert/packet/bin/syncAllSchemasToRegistry.py b/python/lsst/alert/packet/bin/syncAllSchemasToRegistry.py
@@ -23,7 +23,6 @@
 
 import argparse
 import json
-import re
 import fastavro
 import requests
 

diff --git a/python/lsst/alert/packet/schema.py b/python/lsst/alert/packet/schema.py
@@ -26,11 +26,13 @@
 
 import io
 import tempfile
+import re
+import fastavro
 from importlib import resources
 from pathlib import PurePath
+
 from lsst.resources import ResourcePath
 
-import fastavro
 
 __all__ = ["get_schema_root", "get_latest_schema_version", "get_schema_path",
            "Schema", "get_path_to_latest_schema", "get_schema_root_uri",
@@ -428,3 +430,12 @@ def from_file(cls, filename=None):
                                      if schema['name'] == root_name)
 
         return cls(schema_definition)
+
+    def get_schema_id(self):
+        """Retrieve the schema id used in the schema registry.
+        """
+        numbers = re.findall(r'\d+', self.definition['name'])
+        numbers[1] = str(numbers[1]).zfill(2)
+        schema_id = int(''.join(numbers))
+
+        return schema_id
diff --git a/python/lsst/alert/packet/schemaRegistry.py b/python/lsst/alert/packet/schemaRegistry.py
@@ -22,10 +22,8 @@
 """Provide a lookup table for alert schemas.
 """
 
-import json
 import os
 import re
-import zlib
 
 __all__ = ["SchemaRegistry"]