From c63096b3485b883f18c4678cf6c078236bc6c1f5 Mon Sep 17 00:00:00 2001
From: James McCreight <jamesmcc@ucar.edu>
Date: Wed, 2 Feb 2022 15:09:15 -0700
Subject: [PATCH 1/2] get attribute methods extension

---
 modflowapi/modflowapi.py | 61 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/modflowapi/modflowapi.py b/modflowapi/modflowapi.py
index 882fa0a..197cf6e 100644
--- a/modflowapi/modflowapi.py
+++ b/modflowapi/modflowapi.py
@@ -1,3 +1,15 @@
+from ctypes import (
+    CDLL,
+    byref,
+    POINTER,
+    c_char_p,
+    c_double,
+    c_int,
+    c_void_p,
+    cdll,
+    create_string_buffer,
+)
+from typing import Tuple
 from xmipy import XmiWrapper
 
 
@@ -25,3 +37,52 @@ def __init__(
             working_directory=working_directory,
             timing=timing,
         )
+
+    def get_attrs_item_count(self) -> int:
+        count = c_int(0)
+        self.execute_function(self.lib.get_attrs_item_count, byref(count))
+        return count.value
+
+    def get_attrs_keys(self) -> Tuple[str]:
+        len_attr_name = self.get_constant_int("BMI_LENATTRNAME")
+        nr_output_vars = self.get_attrs_item_count()
+        len_names = nr_output_vars * len_attr_name
+        names = create_string_buffer(len_names)
+
+        # get a (1-dim) char array (char*) containing the output variable
+        # names as \x00 terminated sub-strings
+        self.execute_function(self.lib.get_attrs_keys, byref(names))
+
+        # decode
+        output_vars = [
+            names[i * len_attr_name : (i + 1) * len_attr_name]
+            .split(b"\0", 1)[0]
+            .decode("ascii")
+            for i in range(nr_output_vars)
+        ]
+        return tuple(output_vars)
+
+    def get_var_attrs(self, name: str) -> Tuple[str]:
+        len_attr_name = self.get_constant_int("BMI_LENATTRNAME")
+        nr_output_vars = self.get_attrs_item_count()
+        len_attrs = nr_output_vars * len_attr_name
+        attrs = create_string_buffer(len_attrs)
+
+        # get a (1-dim) char array (char*) containing the output variable
+        # names as \x00 terminated sub-strings
+        self.execute_function(
+            self.lib.get_var_attrs,
+            c_char_p(name.encode()),
+            byref(attrs),
+            detail="for variable " + name
+        )
+
+        # decode
+        output_vars = [
+            attrs[i * len_attr_name : (i + 1) * len_attr_name]
+            .split(b"\0", 1)[0]
+            .decode("ascii")
+            for i in range(nr_output_vars)
+        ]
+
+        return tuple(output_vars)

From 9ad75919eb80e32df43aade30e86a5f5cdbe85b6 Mon Sep 17 00:00:00 2001
From: James McCreight <jamesmcc@ucar.edu>
Date: Wed, 2 Feb 2022 15:26:46 -0700
Subject: [PATCH 2/2] tidy up

---
 modflowapi/modflowapi.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/modflowapi/modflowapi.py b/modflowapi/modflowapi.py
index 197cf6e..a24a3f6 100644
--- a/modflowapi/modflowapi.py
+++ b/modflowapi/modflowapi.py
@@ -1,12 +1,7 @@
 from ctypes import (
-    CDLL,
     byref,
-    POINTER,
     c_char_p,
-    c_double,
     c_int,
-    c_void_p,
-    cdll,
     create_string_buffer,
 )
 from typing import Tuple