Allow access to PDB procedures via strings

CHANGELOG.md

@@ -1,3 +1,9 @@
+1.1
+===
+
+* Allowed access to PDB procedures via strings as `pdb['some-procedure-name']`.
+* Python exceptions are now raised if attempting to access non-existent procedure names.
+
 1.0
 ===
 

pdb-wrapper/pypdb.py

@@ -29,11 +29,32 @@ def last_error(self):
   def __getattr__(self, name):
     proc_name = name.replace('_', '-')
 
+    if not self._procedure_exists(proc_name):
+      raise AttributeError(f'procedure "{proc_name}" does not exist')
+
+    return self._get_proc_by_name(proc_name)
+
+  def __getitem__(self, name):
+    proc_name = name.replace('_', '-')
+
+    if not self._procedure_exists(proc_name):
+      raise KeyError(f'procedure "{proc_name}" does not exist')
+
+    return self._get_proc_by_name(proc_name)
+
+  def __contains__(self, name):
+    return self._procedure_exists(name)
+
+  def _get_proc_by_name(self, proc_name):
     if proc_name not in self._proc_cache:
       self._proc_cache[proc_name] = PyPDBProcedure(self, proc_name)
 
     return self._proc_cache[proc_name]
 
+  @staticmethod
+  def _procedure_exists(proc_name):
+    return _gimp_pdb.procedure_exists(proc_name)
+
 
 class PyPDBProcedure: