From 81ae44235c05db7cbde6cae1492446a24ca10f8b Mon Sep 17 00:00:00 2001
From: Friedel Wolff <friedel@translate.org.za>
Date: Fri, 17 Feb 2017 12:39:29 +0200
Subject: [PATCH] Check for PO header in input

---
 translate/tools/posegment.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/translate/tools/posegment.py b/translate/tools/posegment.py
index 190626a653..f3520bc680 100755
--- a/translate/tools/posegment.py
+++ b/translate/tools/posegment.py
@@ -25,7 +25,7 @@
 """
 
 from translate.lang import factory as lang_factory
-from translate.storage import factory
+from translate.storage import factory, poheader
 
 
 class segment:
@@ -62,9 +62,11 @@ def segmentunit(self, unit):
 
     def convertstore(self, fromstore):
         tostore = type(fromstore)()
-        # We don't want the default header in the case of PO, but rather the
-        # one from `fromstore`.
-        tostore.units = []
+        if isinstance(fromstore, poheader.poheader):
+            # We don't want the default header in the case of PO, but rather the
+            # one from `fromstore`.
+            if fromstore.header() is not None:
+                tostore.units = []
         for unit in fromstore.units:
             newunits = self.segmentunit(unit)
             if newunits: