Extra function for getting fault_ids

nshmdb/nshmdb.py

@@ -440,7 +440,21 @@ def get_fault_names(self) -> set[str]:
                 name
                 for (name,) in conn.execute("SELECT name FROM parent_fault").fetchall()
             }
+
+    def get_fault_ids(self) -> set[int]:
+        """Get the list of fault ids in the database.
 
+        Returns
+        -------
+        set[int]
+            The list of fault ids.
+        """
+        with self.connection() as conn:
+            return {
+                fault_id
+                for (fault_id,) in conn.execute("SELECT fault_id FROM fault").fetchall()
+            }
+
     def query(
         self,
         query_str: str,

nshmdb/scripts/nshm_db_generator.py

@@ -302,3 +302,6 @@ def main(
                 rupture_fault_join_df.to_sql(
                     "rupture_faults", conn, index=False, if_exists="append"
                 )
+
+if __name__ == "__main__":
+    app()