New _StepsView.stepstr property

This can be used to name output files depending on which timesteps
were requested when relevant.

Author: amorison

stagpy/rprof.py

@@ -1,6 +1,4 @@
 """Plot radial profiles."""
-import re
-
 import matplotlib.pyplot as plt
 
 from . import conf, misc
@@ -17,10 +15,7 @@ def plot_rprofs(rprofs, lovs):
             produced by :func:`stagpy.misc.list_of_vars`.
     """
     try:
-        stepstr = repr(rprofs.steps)
-        reg = re.compile(
-            r'^StagyyData\(.*\)\.(steps|snaps)\[(.*)\](?:.filter\(.*\))?$')
-        stepstr = '_'.join(reg.match(stepstr).groups())
+        stepstr = rprofs.steps.stepstr
     except AttributeError:
         stepstr = str(rprofs.step.istep)
     sdat = rprofs.step.sdat

stagpy/stagyydata.py

@@ -547,15 +547,25 @@ def rprofs_averaged(self):
             self._rprofs_averaged = _RprofsAveraged(self)
         return self._rprofs_averaged
 
-    def __repr__(self):
-        rep = repr(self._col)
+    @crop
+    def stepstr(self):
+        """String representation of the requested set of steps."""
         items = []
+        no_slice = True
         for item in self._items:
             if isinstance(item, slice):
                 items.append('{}:{}:{}'.format(*item.indices(len(self._col))))
+                no_slice = False
             else:
                 items.append(repr(item))
-        rep += '[{}]'.format(','.join(items))
+        item_str = ','.join(items)
+        if no_slice and len(items) == 1:
+            item_str += ','
+        colstr = repr(self._col).rsplit('.', maxsplit=1)[-1]
+        return f'{colstr}[{item_str}]'
+
+    def __repr__(self):
+        rep = f'{self._col.sdat!r}.{self.stepstr}'
         flts = []
         for flt in ('snap', 'rprofs', 'fields'):
             if self._flt[flt]: