Support plotname, if given, as name of indicator in csv output

shecenon · May 3, 2018 · b239e7e · b239e7e
1 parent 9bfaaff
commit b239e7e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/backtrader/strategy.py b/backtrader/strategy.py
@@ -405,7 +405,8 @@ def getwriterheaders(self):
 
         # prepare the indicators/observers data headers
         for iocsv in self.indobscsv:
-            headers.append(iocsv.__class__.__name__)
+            name = iocsv.plotinfo.plotname or iocsv.__class__.__name__
+            headers.append(name)
             headers.append('len')
             headers.extend(iocsv.getlinealiases())
 
@@ -415,7 +416,8 @@ def getwritervalues(self):
         values = list()
 
         for iocsv in self.indobscsv:
-            values.append(iocsv.__class__.__name__)
+            name = iocsv.plotinfo.plotname or iocsv.__class__.__name__
+            values.append(name)
             lio = len(iocsv)
             values.append(lio)
             if lio: