images

[iramuteq] / parse_factiva_xml.py

diff --git a/parse_factiva_xml.py b/parse_factiva_xml.py
index 6dd2d56..2be8bed 100644 (file)
--- a/parse_factiva_xml.py
+++ b/parse_factiva_xml.py
@@ -12,7 +12,9 @@ import codecs
 import sys
 import re
 import wx
-
+from parse_factiva_mail import ParseFactivaMail
+from parse_factiva_txt import ParseFactivaPaste
+from import_txm import TXM2IRA
 
 def ParseDocument(filename) :
     print filename
@@ -80,17 +82,20 @@ def getcorpus_from_xml(xmldir, corpus_out):
     return 'ok'
 
 class PrefImport(wx.Dialog):
-    def __init__(self, parent, size=wx.DefaultSize, pos=wx.DefaultPosition, style=wx.DEFAULT_DIALOG_STYLE):
+    def __init__(self, parent, size=wx.DefaultSize, pos=wx.DefaultPosition, style=wx.DEFAULT_DIALOG_STYLE, methode = 'mail'):
         pre = wx.PreDialog()
         pre.SetExtraStyle(wx.DIALOG_EX_CONTEXTHELP)
         pre.Create(parent, -1, '', pos, size, style)
         self.PostCreate(pre)
-
+        if methode in ['xml', 'txm'] :
+            txt = _(u'Select a directory of xml files').decode('utf8')
+        else :
+            txt = _(u'Select a directory of txt files').decode('utf8')
         self.parent = parent
-        self.txt1 = wx.StaticText(self, -1, u"Répertoire des fichiers xml")
+        self.txt1 = wx.StaticText(self, -1, txt.encode('utf8'))
         self.dbb = filebrowse.DirBrowseButton(self, -1, size=(450, -1), changeCallback = self.fbbCallback)
         self.dbb.SetLabel("")
-        self.txt2 = wx.StaticText(self, -1, u"Fichier en sortie")
+        self.txt2 = wx.StaticText(self, -1, _(u'Output file').decode('utf8'))
         self.fbb = filebrowse.FileBrowseButton(self, -1, size=(450, -1), fileMode = 2)
         self.fbb.SetLabel("")
 
@@ -154,18 +159,25 @@ class PrefImport(wx.Dialog):
 
 
 class ImportFactiva():
-    def __init__(self,parent):
-       self.dial =  PrefImport(parent)
+    def __init__(self, parent, methode):
+       self.dial =  PrefImport(parent, methode=methode)
        self.dial.CenterOnParent()
        val = self.dial.ShowModal()
        if val == wx.ID_OK :
            xmldir = self.dial.dbb.GetValue()
            corp_out = self.dial.fbb.GetValue()
-           res = getcorpus_from_xml(xmldir, corp_out)
-           if res != 'ok' :
+           if methode == 'xml' :
+                res = getcorpus_from_xml(xmldir, corp_out)
+           elif methode == 'mail' :
+               res = ParseFactivaMail(xmldir, corp_out, 'utf8', parent.syscoding)
+           elif methode == 'txt' :
+               res = ParseFactivaPaste(xmldir, corp_out, 'utf8', parent.syscoding)
+           elif methode == 'txm' :
+               res = TXM2IRA(xmldir, corp_out, 'utf8', parent.syscoding)
+           if res == 'nofile' :
                 dlg = wx.MessageDialog(parent, u"Pas de fichier \'.xml\' dans %s" % xmldir, 'ATTENTION', wx.OK | wx.NO_DEFAULT | wx.ICON_WARNING)
                 dlg.CenterOnParent()
                 dlg.ShowModal()
-           else :
-                parent.filename = corp_out
-                parent.OpenText()
+           #else :
+           #     parent.filename = corp_out
+           #     parent.OpenText()