Fixed some issues with the Py2/3 changes to the populate_yang_model_dirs management command.

 - Legacy-Id: 16520

Author: levkowetz

ietf/utils/management/commands/populate_yang_model_dirs.py

@@ -9,7 +9,7 @@
 import time
 
 from pathlib2 import Path
-from io import StringIO
+from six import StringIO
 from textwrap import dedent
 from xym import xym
 
@@ -83,15 +83,18 @@ def extract_from(file, dir, strict=True):
                 model_list = [ n.replace('"','') for n in model_list ]
             except Exception as e:
                 self.stderr.write("** Error when extracting from %s: %s" % (file, str(e)))
-            sys.stdout = saved_stdout
-            sys.stderr = saved_stderr
+            finally:
+                sys.stdout = saved_stdout
+                sys.stderr = saved_stderr
             #
             if verbosity > 1:
                 outmsg = xymout.getvalue()
-                self.stdout.write(outmsg)
+                if outmsg.strip():
+                    self.stdout.write(outmsg)
             if verbosity>2:            
                 errmsg = xymerr.getvalue()
-                self.stderr.write(errmsg)
+                if errmsg.strip():
+                    self.stderr.write(errmsg)
             return model_list
 
         # Extract from new RFCs