class AWCore:
class Error(Exception):
def __init__(self, raison):
self.reason = raison
def __str__(self):
return self.reason
def __init__(self):
self.textEdit = TETextEdit(
language_name=TLPreferences['DFT_WRITING_LANGUAGE'])
self.metadata = DPMetaData()
self.filepath = None
self.lastFiles = CLLastFiles()
self.config_spelling = CLSpelling()
self.config_autocor = CLAutoCorrection()
def CMD_FileSave(self, filepath=None):
if filepath == None:
if self.filepath == None:
raise self.Error('Please specify the filepath')
filepath = self.filepath
else:
self.filepath = filepath
# filepath += 'ZIP'
tmpfile = False
if os.path.exists(filepath):
# For safety, we temporary copy the exixting file
tmpfile = os.path.join(
os.path.split(filepath)[0], '~' + os.path.split(filepath)[1])
shutil.copyfile(filepath, tmpfile)
# we save the main file, the 'w' option is to erase existing files
res = FMZipFileManagement.save(str(self.textEdit.toXml()),
zipfilepath=filepath,
filename='main.txt',
modezip='w')
if res:
# we will save the file .athw_meta as well
cur = self.textEdit.textCursor()
self.metadata.__setitem__('lastpos',
int(cur.position()),
protected=False)
self.metadata['language'] = self.textEdit.language.name
self.metadata.__setitem__('profile',
self.textEdit.language.profile,
protected=False)
self.metadata.__setitem__('athw_version',
__version__,
protected=False)
to_save = self.metadata.toxml()
res = FMZipFileManagement.save(to_save, filepath, 'meta.xml')
self.config_spelling.local_file = filepath
self.config_autocor.local_file = filepath
self.config_spelling.saveConfig('file')
self.config_autocor.saveConfig('file')
def updateLocal(config):
_, filepath_config, exists = config.get_file_list(
where='local')[0]
if exists:
config_tmp = config.__class__(local_file=filepath)
if config.dictConfig['local'] != None:
config_tmp.update(config.dictConfig['local'], 'local')
config = config_tmp
config.saveConfig('local')
return config
self.config_spelling = updateLocal(self.config_spelling)
self.config_autocor = updateLocal(self.config_autocor)
if tmpfile: # We remove the temporary file
os.remove(tmpfile)
if os.path.exists(filepath + '_meta'):
# if there was a metadata file from previous version
os.remove(filepath + '_meta')
def CMD_FileSaveCopy(self, filepath):
old_filepath = self.filepath
self.CMD_FileSave(filepath)
self.filepath = old_filepath
def CMD_FileOpen(self, filepath):
if filepath == None:
if self.filepath == None:
raise self.Error('Please specify the filepath')
filepath = self.filepath
else:
self.filepath = filepath
try:
text = FMZipFileManagement.open(self.filepath, 'main.txt')
if 'meta.xml' in FMZipFileManagement.listFiles(self.filepath):
metadata = DPMetaData.init_from_xml_string(
FMZipFileManagement.open(self.filepath, 'meta.xml'))
else:
metadata = DPMetaData()
self.config_spelling.changeFile(filepath)
self.config_autocor.changeFile(filepath)
except zipfile.BadZipFile:
text, metadata = self.CMD_FileOpen_version_1(filepath)
except KeyError as e:
raise IOError('Not a correct ATHW file' + str(e))
self.metadata = metadata
language = self.metadata['language']
lastpos = self.metadata['lastpos']
profile = self.metadata['profile']
if profile == None:
profile = TLPreferences['DFT_TYPO_PROFILE']
self.textEdit.setText(text,
type='xml',
new_language=language,
profile=profile)
if lastpos != None:
if lastpos == -1:
lastpos = self.textEdit.document().characterCount()
cur = self.textEdit.textCursor()
cur.setPosition(lastpos)
self.textEdit.setTextCursor(cur)
return True
def CMD_FileOpen_version_1(self, filepath):
"""
In order to open old Athw files that were not a zip file.
"""
# Get the text:
local_dir, tmp = os.path.split(self.filepath)
text = FMTextFileManagement.open(filepath)
meta_filepath, tmp = os.path.splitext(self.filepath)
meta_filepath += '.athw_meta'
if os.path.exists(meta_filepath):
metadata = DPMetaData.init_from_xml_string(
FMTextFileManagement.open(meta_filepath))
else:
metadata = DPMetaData()
return text, metadata
def CMD_FileImport(self, filepath, format_name, to_skip=None):
"""
- filepath : the path to the file to import
- format_name : the name of the format
- to_skip : the list of the lines to skip
"""
if to_skip == None: to_skip = []
# We check whenever we can import this type of files
list_extentions = list(DIDict.keys())
if format_name not in list_extentions:
raise self.Error(
'Unknown format <%s> to import: choose in '%format_name+\
str(list_extentions))
fiimport_instance = DIDict[format_name](file=filepath)
fiimport_instance.import2xml()
for i in to_skip:
fiimport_instance.skipLine(i)
text = fiimport_instance.text
self.textEdit.setText(text, type='xml')
self.textEdit.SLOT_actionRecheckTypography()
self.filepath = None
def CMD_FileExport(self,format_name,filepath=None,check_typo=False,\
**kargs):
if check_typo:
self.textEdit.SLOT_actionRecheckTypography()
list_extentions = [format.name for format in DEList]
if format_name not in list_extentions:
raise self.Error('Unknown format to import: choose in '+\
str(list_extentions))
index = list_extentions.index(format_name)
format = DEList[index](TSManager)
format.export(self.textEdit, **kargs)
res = format.export_file(
# default_saving_site=self.get_default_opening_saving_site(),
parent=self,
filepath=filepath,
)
return res
class AWWriterText(QtWidgets.QMainWindow, AWCore):
def __init__(self, *args, **kargs):
QtWidgets.QMainWindow.__init__(self, *args, **kargs)
AWCore.__init__(self)
# dict_autocorrection = FMAutoCorrectionFile.open()
self.textEdit = TETextEdit(
parent=self, language_name=TLPreferences['DFT_WRITING_LANGUAGE'])
self.setCentralWidget(self.textEdit)
self.fullScreened = False
self.setup_actions()
self.setup_connections()
self.setup_menus()
self.setWindowTitle("AthenaWriter : NewFile")
self.setGeometry(100, 100, AWPreferences["INIT_SIZE_X"],
AWPreferences["INIT_SIZE_Y"])
self.changeMessageStatusBar("Welcome in AthenaWriter :)")
if AWPreferences['AUTOSAVE']:
self.timer(start=True)
# self.setGeometry(100,100,500,1000)
def setup_actions(self):
self.actionFileSave = QtWidgets.QAction("Save", self)
self.actionFileSaveAs = QtWidgets.QAction("Save as", self)
self.actionFileNew = QtWidgets.QAction("New", self)
self.actionFileOpen = QtWidgets.QAction("Open", self)
self.actionFileImport = QtWidgets.QAction("Import", self)
self.actionFileExport = QtWidgets.QAction("Export", self)
self.actionFileStats = QtWidgets.QAction("Statistics", self)
self.actionFileMetaData = QtWidgets.QAction("Meta-data", self)
self.actionFileQuit = QtWidgets.QAction("Quit", self)
# self.actionFileExport.setEnabled(False) ######## TODO to debug
self.actionRecentFilesList = []
# dico=self.language.shortcuts_insert
for path in self.lastFiles.getValues():
#create the actions to open the last files
act = QtWidgets.QAction(path, self)
self.actionRecentFilesList.append(act)
self.actionSendToExternalSoftware = QtWidgets.QAction(
"Send to external software", self)
self.actionEditPreferences = QtWidgets.QAction("Preferences", self)
# self.actionViewZoomIn = QtWidgets.QAction("Zoom in" ,self)
# self.actionViewZoomOut = QtWidgets.QAction("Zoom out" ,self)
# self.actionViewZoomDft = QtWidgets.QAction("Zoom default" ,self)
self.actionViewFullScreen = QtWidgets.QAction("FullScreen", self)
self.actionAboutHelp = QtWidgets.QAction("Help", self)
self.actionAboutAbout = QtWidgets.QAction("About Athena Writer", self)
self.actionFileSave.setShortcuts(QtGui.QKeySequence.Save)
self.actionFileSaveAs.setShortcuts(
QtGui.QKeySequence("Ctrl+Shift+S")) #TODO for Windows
self.actionFileNew.setShortcuts(QtGui.QKeySequence.New)
self.actionFileOpen.setShortcuts(QtGui.QKeySequence.Open)
# self.actionFileStats.setShortcuts(QtGui.QKeySequence.Stats)
self.actionFileQuit.setShortcuts(
QtGui.QKeySequence("Ctrl+Shift+Q")) #TODO for Windows
self.actionFileSave.setEnabled(False)
self.actionViewFullScreen.setShortcuts(QtGui.QKeySequence("F11"))
# TODO: fix Import
self.actionFileImport.setEnabled(True)
def setup_connections(self):
self.actionFileSave.triggered.connect(self.SLOT_actionFileSave)
self.actionFileSaveAs.triggered.connect(self.SLOT_actionFileSaveAs)
self.actionFileNew.triggered.connect(self.SLOT_actionFileNew)
self.actionFileOpen.triggered.connect(self.SLOT_actionFileOpen)
self.actionFileImport.triggered.connect(self.SLOT_actionFileImport)
self.actionFileExport.triggered.connect(self.SLOT_actionFileExport)
self.actionFileStats.triggered.connect(self.SLOT_actionFileStats)
self.actionFileMetaData.triggered.connect(self.SLOT_actionFileMetaData)
self.actionFileQuit.triggered.connect(self.close)
self.actionSendToExternalSoftware.triggered.connect(
self.SLOT_actionSendToExternalSoftware)
self.actionEditPreferences.triggered.connect(
self.SLOT_actionEditPreferences)
# Add last file list
mapper = QtCore.QSignalMapper(self)
for path, act in zip(self.lastFiles.getValues(),
self.actionRecentFilesList):
act.triggered.connect(mapper.map)
# short.setContext(QtCore.Qt.WidgetShortcut)
mapper.setMapping(act, path)
# mapper.mapped.connect( self.SLOT_actionFileOpen )# TR:todelete
mapper.mapped[str].connect(self.SLOT_actionFileOpen)
self.actionViewFullScreen.triggered.connect(
self.SLOT_actionViewFullScreen)
self.actionAboutHelp.triggered.connect(self.SLOT_actionAboutHelp)
self.actionAboutAbout.triggered.connect(self.SLOT_actionAboutAbout)
self.textEdit.textChanged.connect(self.SLOT_somethingChanged)
self.textEdit.somethingChanged.connect(self.SLOT_somethingChanged)
self.textEdit.typographyModification.connect(
self.SLOT_typographyModification)
self.textEdit.protectedStyleModification.connect(
self.SLOT_protectedStyleModification)
def setup_menus(self):
menuFile = self.menuBar().addMenu("File")
menuFile.addAction(self.actionFileNew)
menuFile.addAction(self.actionFileOpen)
menuRecentFiles = menuFile.addMenu("RecentFiles")
for act in self.actionRecentFilesList:
menuRecentFiles.addAction(act)
menuFile.addSeparator()
menuFile.addAction(self.actionFileSave)
menuFile.addAction(self.actionFileSaveAs)
menuFile.addSeparator()
menuFile.addAction(self.actionFileImport)
menuFile.addAction(self.actionFileExport)
menuFile.addSeparator()
menuFile.addAction(self.actionFileQuit)
menuEdit = self.menuBar().addMenu("Edit")
menuEdit.addAction(self.textEdit.actionUndo)
menuEdit.addAction(self.textEdit.actionRedo)
menuEdit.addSeparator()
menuEdit.addAction(self.textEdit.actionCopy)
menuEdit.addAction(self.textEdit.actionCut)
menuEdit.addAction(self.textEdit.actionPaste)
menuEdit.addSeparator()
menuEdit.addAction(self.textEdit.actionLaunchFindDialog)
menuEdit.addAction(self.textEdit.actionFindNext)
menuEdit.addAction(self.textEdit.actionFindPrevious)
menuEdit.addAction(self.textEdit.actionLaunchCharWidgetTable)
menuEdit.addSeparator()
# menuEdit.addAction(self.textEdit.actionChangeLanguage)
menuEdit.addAction(self.actionEditPreferences)
menuDocument = self.menuBar().addMenu("Document")
menuDocument.addAction(self.textEdit.actionRecheckTypography)
menuProfiles = menuDocument.addMenu("Profiles")
for k, v in list(self.textEdit.actionProfileDict.items()):
menuProfiles.addAction(v)
menuDocument.addSeparator()
menuDocument.addAction(self.actionFileStats)
menuDocument.addAction(self.actionFileMetaData)
menuDocument.addSeparator()
menuDocument.addAction(self.actionSendToExternalSoftware)
menuDocument.addAction(self.textEdit.actionGuessLanguage)
menuStyle = self.menuBar().addMenu("Styles")
for id in list(TSManager.dictCharStyle.keys()):
if id in list(self.textEdit.actionStylesDict.keys()):
menuStyle.addAction(self.textEdit.actionStylesDict[id])
menuStyle.addSeparator()
for id in list(TSManager.dictBlockStyle.keys()):
if id in list(self.textEdit.actionStylesDict.keys()):
menuStyle.addAction(self.textEdit.actionStylesDict[id])
menuStyle.addSeparator()
menuStyle.addAction(self.textEdit.actionInsertImage)
menuStyle.addSeparator()
menuStyle.addAction(self.textEdit.actionResetStyle)
menuView = self.menuBar().addMenu("View")
menuView.addAction(self.textEdit.actionZoomIn)
menuView.addAction(self.textEdit.actionZoomOut)
menuView.addAction(self.textEdit.actionZoomNormal)
menuView.addAction(self.actionViewFullScreen)
menuAbout = self.menuBar().addMenu("About")
menuAbout.addAction(self.actionAboutHelp)
menuAbout.addAction(self.actionAboutAbout)
list_actions = [ self.actionFileNew,self.actionFileOpen,
self.actionFileImport,self.actionFileSave,self.actionFileSaveAs,
self.actionFileExport,self.actionFileStats,self.actionFileMetaData,
self.actionFileQuit,self.actionViewFullScreen,
self.textEdit.actionUndo,self.textEdit.actionRedo,
self.textEdit.actionCopy,self.textEdit.actionCut,
self.textEdit.actionPaste, self.textEdit.actionLaunchFindDialog,
self.textEdit.actionFindNext,self.textEdit.actionFindPrevious,
self.textEdit.actionLaunchCharWidgetTable,
# self.textEdit.actionChangeLanguage,
self.textEdit.actionRecheckTypography,
self.textEdit.actionEnableTypo,self.actionSendToExternalSoftware,
self.actionAboutHelp,self.actionAboutAbout] + \
list(self.textEdit.actionStylesDict.values())
for act in list_actions:
self.addAction(act)
################################## SLOTS ##################################
@QtCore.pyqtSlot()
def SLOT_actionFileSave(self, force_ask=False):
"""
Slot used when saving the current file.
force_ask : if True, will force asking for the new filepath
"""
if self.filepath == None or force_ask:
filepath = FMTextFileManagement.save_gui_filepath(
self.get_default_opening_saving_site(),
self,
filter="AthW files (*.athw);; All files (*.*)",
ext=".athw")
else:
filepath = self.filepath
if filepath:
filepath = str(filepath)
### Changed : ProgressBar is instable !
## progressBar = QtWidgets.QProgressBar(parent=self)
## progressBar.setMaximum(0)
## progressBar.setValue(0)
## # center the widget
## point = self.geometry().center()
## point1 = progressBar.geometry().center()
## progressBar.move(point-point1)
## # set the window to the correct flag
## progressBar.setWindowFlags(QtCore.Qt.SplashScreen)
## progressBar.setWindowModality(QtCore.Qt.WindowModal)
##
## progressBar.show()
##
## # this class is a very classic thread just to save the file while
## # displaying the progressBar:
## class TaskThread(QtCore.QThread):
## taskFinished = QtCore.pyqtSignal()
## def run(self1):
## self.CMD_FileSave(filepath=filepath)
## self1.taskFinished.emit()
##
## # What we have to do at the end of the saving
## def finishing():
## progressBar.close()
## self.actionFileSave.setEnabled(False)
## self.setWindowTitle("AthenaWriter : "+self.filepath)
## tmp,filename = os.path.split(self.filepath)
## self.changeMessageStatusBar("Has saved "+filename)
## self.lastFiles.addFile(self.filepath)
##
## self.myLongTask = TaskThread()
## self.myLongTask.taskFinished.connect(
## finishing)
##
## # self.myLongTask.taskFinished.connect(finishing)
## self.myLongTask.start()
self.textEdit.setReadOnly(True)
self.CMD_FileSave(filepath=filepath)
self.actionFileSave.setEnabled(False)
self.setWindowTitle("AthenaWriter : " + self.filepath)
tmp, filename = os.path.split(self.filepath)
self.changeMessageStatusBar("Has saved " + filename)
self.lastFiles.update([self.filepath])
self.textEdit.setReadOnly(False)
@QtCore.pyqtSlot()
def SLOT_actionFileSaveAs(self):
"""
Slot used when saving as the current file.
"""
self.SLOT_actionFileSave(force_ask=True)
@QtCore.pyqtSlot()
def SLOT_actionFileNew(self):
"""
Slot used when creating a new file.
"""
res = self.doSaveDialog()
if (res != QtWidgets.QMessageBox.Yes) and (res !=
QtWidgets.QMessageBox.No):
return False
self.clean_tmp_files() # we remove the previous tmp files
self.textEdit.setText("")
self.filepath = None
self.metadata = DPMetaData()
self.setWindowTitle("AthenaWriter : NewFile")
self.changeMessageStatusBar("Created a new file")
return True
@QtCore.pyqtSlot(str)
@QtCore.pyqtSlot()
def SLOT_actionFileOpen(self, filepath=None):
"""
If filepath==None, it will display the window to search the file
If filepath==None, it will display the window to search the file
"""
res = self.doSaveDialog()
if (res != QtWidgets.QMessageBox.Yes) and (res !=
QtWidgets.QMessageBox.No):
return False
if filepath == None:
filepath = FMTextFileManagement.open_gui_filepath(
self.get_default_opening_saving_site(),
self,
filter="AthW files (*.athw);; All files (*.*)")
else:
filepath = str(filepath)
if filepath:
self.clean_tmp_files() # we remove the previous tmp files
try:
self.CMD_FileOpen(filepath)
except IOError:
# if the file do not exist, propose to crreate it
r = QtWidgets.QMessageBox.question(self,"Non existing file",
"The file "+filepath+" do not exist. Do you want to "+\
"create it ?",
QtWidgets.QMessageBox.Yes|QtWidgets.QMessageBox.No)
if r != QtWidgets.QMessageBox.Yes:
return False
self.filepath = filepath
self.actionFileSave.setEnabled(False)
self.setWindowTitle("AthenaWriter : " + self.filepath)
tmp, filename = os.path.split(filepath)
self.changeMessageStatusBar("Has opened " + filename)
self.lastFiles.update([self.filepath])
return True
else:
return False
@QtCore.pyqtSlot()
def SLOT_actionFileImport(self, checked=False, filepath=None):
"""
If filepath==None, it will display the window to search the file
"""
res = self.doSaveDialog()
if (res != QtWidgets.QMessageBox.Yes) and (res !=
QtWidgets.QMessageBox.No):
return False
if filepath == None:
extensions = list(DIDict.keys())
filepath = FMTextFileManagement.open_gui_filepath(
parent=self,
dft_opening_saving_site=self.get_default_opening_saving_site(),
filter='*.' + ' *.'.join(extensions))
# filepath = FMTextFileManagement.open_gui_filepath(
# self.get_default_opening_saving_site(),
# self)
else:
filepath = str(filepath)
if filepath:
self.clean_tmp_files() # we remove the previous tmp files
path, e = os.path.splitext(filepath)
e = e[1:] #the [1:] is to skip the dot in the extension
self.CMD_FileImport(filepath=filepath, format_name=e)
self.actionFileSave.setEnabled(True)
self.setWindowTitle("AthenaWriter : NewFile")
tmp, filename = os.path.split(filepath)
self.changeMessageStatusBar("Has imported " + filename)
return True
else:
return False
@QtCore.pyqtSlot()
def SLOT_actionFileExport(self):
dft_opening_saving_site = self.get_default_opening_saving_site()
if self.filepath != None:
res = DEDialog.getFields(
dft_opening_saving_site=dft_opening_saving_site,
default_path=self.filepath,
source_d=self.metadata,
)
else:
res = DEDialog.getFields(
dft_opening_saving_site=dft_opening_saving_site,
source_d=self.metadata,
)
if res:
try:
res1 = self.CMD_FileExport(**res)
except IOError as e:
print(e)
res1 = False
if res1:
msg = "Successfull exportation"
self.changeMessageStatusBar("Has exported " + res['filepath'])
dia = QtWidgets.QMessageBox.information(
self, "Exportation", msg)
else:
msg = "Unsuccessfull exportation :\n"
msg += str(e)
dia = QtWidgets.QMessageBox.critical(self, "Exportation", msg)
# import TextFormat.TextFormat
# list_extension = [format.extension for format in FEList]
# res = QtWidgets.QInputDialog. getItem (self,'Export',
# 'Format of the exportation:', list_extension)
# if res[1]:
# res1 = self.CMD_FileExport(format_name = str(res[0]))
# if res1:
# self.changeMessageStatusBar("Has exported "+res1)
# return True
# return False
@QtCore.pyqtSlot()
def SLOT_actionFileStats(self):
dialog = DPStatisticsDialog(textedit=self.textEdit, parent=self)
dialog.show()
@QtCore.pyqtSlot()
def SLOT_actionFileMetaData(self, d=None):
if d == None:
# constraints_dict = {'language': [TLDico.keys()] }
d = CMDialog.getValueDict(parent=self,
constants_manager=self.metadata,
skip_same_as_init=True)
if d and len(d) > 0:
self.metadata.update(d)
if 'language' in list(d.keys()):
self.textEdit.changeLanguage(language_name=str(d['language']),
gui=True)
self.SLOT_somethingChanged()
# dialog = MDMetaDataDialog(metadata=self.metadata,parent=self)
# dialog.show()
@QtCore.pyqtSlot()
def SLOT_actionViewFullScreen(self):
if not self.fullScreened:
self.showFullScreen()
# self.textEdit.setVerticalScrollBarPolicy(
# QtCore.Qt.ScrollBarAlwaysOff)
self.menuBar().setHidden(True)
if AWPreferences['FULLSCREEN_CENTRAL_MAX_SIZE'] > 0:
rec = QtWidgets.QApplication.desktop().screenGeometry()
marg = rec.width()
marg -= AWPreferences['FULLSCREEN_CENTRAL_MAX_SIZE']
marg *= .5
if marg > 0:
self.setContentsMargins(marg, 0, marg, 0)
else:
print("CAUTION : FULLSCREEN_CENTRAL_MAX_SIZE too high")
self.fullScreened = True
else:
self.showNormal()
self.setContentsMargins(0, 0, 0, 0)
self.textEdit.setVerticalScrollBarPolicy(
QtCore.Qt.ScrollBarAsNeeded)
self.menuBar().setHidden(False)
self.fullScreened = False
@QtCore.pyqtSlot()
def SLOT_actionAboutHelp(self):
pass
@QtCore.pyqtSlot()
def SLOT_actionAboutAbout(self):
text = AWAbout.split("\n\n")
text = [' '.join(a.split('\n')) for a in text]
text = '\n\n'.join(text)
text = text.strip()
QtWidgets.QMessageBox.about(self, "About Athena Writer", text)
def SLOT_somethingChanged(self):
self.actionFileSave.setEnabled(True)
if self.filepath == None:
self.setWindowTitle("AthenaWriter : *NewFile")
else:
self.setWindowTitle("AthenaWriter : *" + self.filepath)
def SLOT_typographyModification(self, modification):
self.changeMessageStatusBar(modification[0].title)
def SLOT_protectedStyleModification(self, mess):
self.changeMessageStatusBar(mess)
def SLOT_autosave(self):
if self.filepath != None:
direct, file = os.path.split(self.filepath)
tmp_filepath = os.path.join(direct,
AWPreferences['TMP_FILE_MARK'] + file)
res = FMTextFileManagement.save(str(self.textEdit.toXml()),
tmp_filepath)
@QtCore.pyqtSlot()
def SLOT_actionSendToExternalSoftware(self):
# Send the scene to another software (mine is called Antidote and
# accept only a certain type of encoding).
path = AWPreferences['EXTERNAL_SOFT_PATH']
if path == "":
QtWidgets.QMessageBox.information(self,
"External Software Sender",
"Sorry, there is no path to the external software in the "+\
"configuration file.")
return False
text = self.textEdit.toXml()
i = 0
while AWPreferences['TMP_FILE_MARK']+"tmp"+str(i).zfill(3)+'.txt' in \
os.listdir('.'):
i += 1
name = AWPreferences['TMP_FILE_MARK'] + "tmp" + str(i).zfill(
3) + '.txt'
FMTextFileManagement.save(text, name, encoding='utf-8-sig', mode='w')
try:
s = subprocess.Popen(path + ' ' + os.path.abspath(name))
except OSError as e:
QtWidgets.QMessageBox.critical(
self, "Error with external software",
"Error when sending the file to the external software\n\n" +
str(e))
else:
res = QtWidgets.QMessageBox.question(
self, "External Software Sender",
"Have you finished to correct the file?",
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.Cancel)
if (res == QtWidgets.QMessageBox.Yes):
text = FMTextFileManagement.open(name,
encoding='utf-8-sig',
mode='rb')
self.textEdit.setText(text, type='xml')
@QtCore.pyqtSlot()
def SLOT_actionEditPreferences(self):
d = CLPreferencesDialog.getValueDict(
dict_preferences=AWDictPreferences, parent=self)
if d:
AWOverwritePreferences(d)
pref_dict, descr_dict = AWPreferencesToDict(skip_same_as_dft=True)
d = CLPreferencesFiles()
d.update(other_dict=pref_dict, where='user')
d.saveConfig(descriptions=descr_dict, where='user')
def closeEvent(self, event):
"""Check if we have changed something without saving"""
res = self.doSaveDialog()
if (res == QtWidgets.QMessageBox.Yes) or (res
== QtWidgets.QMessageBox.No):
self.lastFiles.saveConfig('user')
self.clean_tmp_files()
if AWPreferences['AUTOSAVE']: #We stop the thread of the autosave
self.threading_autosave.cancel()
event.accept()
else:
event.ignore()
###########################################################################
def get_default_opening_saving_site(self):
"""When open a file dialog, in which directory the dialog window
should begin :
"""
if self.filepath == None:
return str(AWPreferences['DLT_OPEN_SAVE_SITE'].expanduser())
else:
res, tmp = os.path.split(self.filepath)
return res
def doSaveDialog(self):
"""
Check if there is any modifications and ask if it has to save the file
if it not the case.
"""
if self.actionFileSave.isEnabled():
res=QtWidgets.QMessageBox.question(
self,
"Modification", "Do you want to save the modifications",
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No | \
QtWidgets.QMessageBox.Cancel
)
if (res == QtWidgets.QMessageBox.Yes):
self.SLOT_actionFileSave()
return res
return QtWidgets.QMessageBox.Yes
def changeMessageStatusBar(self, message=None):
"""This function is called to display a message in the status bar"""
if message == None: message = ""
self.statusBar().showMessage(message,
AWPreferences['TIME_STATUS_MESSAGE'])
def timer(self, start=False):
self.threading_autosave = threading.Timer(
AWPreferences['AUTOSAVE_TEMPO'], self.timer)
self.threading_autosave.start()
if not start:
self.changeMessageStatusBar('Autosave')
self.SLOT_autosave()
def clean_tmp_files(self):
if self.filepath != None:
direct, file = os.path.split(self.filepath)
tmp_filepath = os.path.join(direct,
AWPreferences['TMP_FILE_MARK'] + file)
try:
os.remove(tmp_filepath)
except:
pass
class AWWriterText(QtGui.QMainWindow,AWCore):
def __init__(self,*args,**kargs):
QtGui.QMainWindow.__init__(self,*args,**kargs)
AWCore.__init__(self)
# dict_autocorrection = FMAutoCorrectionFile.open()
self.textEdit = TETextEdit(
language_name=TLPreferences['DFT_WRITING_LANGUAGE'])
self.setCentralWidget (self.textEdit)
self.fullScreened=False
self.setup_actions()
self.setup_connections()
self.setup_menus()
self.setWindowTitle("AthenaWriter : NewFile")
self.setGeometry(100,100,AWPreferences["INIT_SIZE_X"],
AWPreferences["INIT_SIZE_Y"])
self.changeMessageStatusBar("Welcome in AthenaWriter :)")
if AWPreferences['AUTOSAVE']:
self.timer(start=True)
# self.setGeometry(100,100,500,1000)
def setup_actions(self):
self.actionFileSave = QtGui.QAction("Save" ,self)
self.actionFileSaveAs = QtGui.QAction("Save as" ,self)
self.actionFileNew = QtGui.QAction("New" ,self)
self.actionFileOpen = QtGui.QAction("Open" ,self)
self.actionFileImport = QtGui.QAction("Import" ,self)
self.actionFileExport = QtGui.QAction("Export" ,self)
self.actionFileStats = QtGui.QAction("Statistics" ,self)
self.actionFileMetaData = QtGui.QAction("Meta-data" ,self)
self.actionFileQuit = QtGui.QAction("Quit" ,self)
self.actionRecentFilesList = []
# dico=self.language.shortcuts_insert
for path in self.lastFiles.list_files:
#create the actions to open the last files
act = QtGui.QAction(path,self)
self.actionRecentFilesList.append(act)
self.actionSendToExternalSoftware = QtGui.QAction(
"Send to external software" ,self)
self.actionEditPreferences = QtGui.QAction("Preferences" ,self)
# self.actionViewZoomIn = QtGui.QAction("Zoom in" ,self)
# self.actionViewZoomOut = QtGui.QAction("Zoom out" ,self)
# self.actionViewZoomDft = QtGui.QAction("Zoom default" ,self)
self.actionViewFullScreen = QtGui.QAction("FullScreen" ,self)
self.actionAboutHelp = QtGui.QAction("Help" ,self)
self.actionAboutAbout = QtGui.QAction("About WolfText",self)
self.actionFileSave.setShortcuts(QtGui.QKeySequence.Save)
self.actionFileSaveAs.setShortcuts(QtGui.QKeySequence("Ctrl+Shift+S")) #TODO for Windows
self.actionFileNew.setShortcuts(QtGui.QKeySequence.New)
self.actionFileOpen.setShortcuts(QtGui.QKeySequence.Open)
# self.actionFileStats.setShortcuts(QtGui.QKeySequence.Stats)
self.actionFileQuit.setShortcuts(QtGui.QKeySequence("Ctrl+Shift+Q")) #TODO for Windows
self.actionFileSave.setEnabled(False)
self.actionViewFullScreen.setShortcuts(QtGui.QKeySequence("F11"))
# TODO: fix Import
self.actionFileImport.setEnabled(False)
def setup_connections(self):
trig = QtCore.SIGNAL("triggered()")
self.connect(self.actionFileSave ,trig,self.SLOT_actionFileSave )
self.connect(self.actionFileSaveAs ,trig,self.SLOT_actionFileSaveAs)
self.connect(self.actionFileNew ,trig,self.SLOT_actionFileNew )
self.connect(self.actionFileOpen ,trig,self.SLOT_actionFileOpen )
self.connect(self.actionFileImport ,trig,self.SLOT_actionFileImport)
self.connect(self.actionFileExport ,trig,self.SLOT_actionFileExport)
self.connect(self.actionFileStats ,trig,self.SLOT_actionFileStats )
self.connect(self.actionFileMetaData,trig,self.SLOT_actionFileMetaData)
self.connect(self.actionFileQuit ,trig,self.close )
self.connect(self.actionSendToExternalSoftware, trig,
self.SLOT_actionSendToExternalSoftware)
self.connect(self.actionEditPreferences, trig,
self.SLOT_actionEditPreferences)
# Add last file list
mapper = QtCore.QSignalMapper(self)
for path,act in zip(self.lastFiles.list_files,self.actionRecentFilesList):
QtCore.QObject.connect(act,QtCore.SIGNAL("triggered ()"), mapper, QtCore.SLOT("map()"))
# short.setContext(QtCore.Qt.WidgetShortcut)
mapper.setMapping(act, path)
self.connect(mapper, QtCore.SIGNAL("mapped(const QString &)"), self.SLOT_actionFileOpen )
self.connect(self.actionViewFullScreen ,QtCore.SIGNAL("triggered()"),
self.SLOT_actionViewFullScreen)
self.connect(self.actionAboutHelp,QtCore.SIGNAL("triggered()"),
self.SLOT_actionAboutHelp)
self.connect(self.actionAboutAbout,QtCore.SIGNAL("triggered()"),
self.SLOT_actionAboutAbout)
self.connect(
self.textEdit,
QtCore.SIGNAL('textChanged ()'),
self.SLOT_somethingChanged
)
self.connect(
self.textEdit,
QtCore.SIGNAL('somethingChanged ()'),
self.SLOT_somethingChanged
)
self.connect(
self.textEdit,
QtCore.SIGNAL('typographyModification (PyQt_PyObject)'),
self.SLOT_typographyModification
)
self.connect(
self.textEdit,
QtCore.SIGNAL('protectedStyleModification (PyQt_PyObject)'),
self.SLOT_protectedStyleModification
)
def setup_menus(self):
menuFile = self.menuBar().addMenu ( "File" )
menuFile.addAction(self.actionFileNew)
menuFile.addAction(self.actionFileOpen)
menuRecentFiles = menuFile.addMenu ("RecentFiles")
for act in self.actionRecentFilesList:
menuRecentFiles.addAction(act)
menuFile.addSeparator ()
menuFile.addAction(self.actionFileSave)
menuFile.addAction(self.actionFileSaveAs)
menuFile.addSeparator ()
menuFile.addAction(self.actionFileImport)
menuFile.addAction(self.actionFileExport)
menuFile.addSeparator ()
menuFile.addAction(self.actionFileQuit)
menuEdit = self.menuBar().addMenu ( "Edit" )
menuEdit.addAction(self.textEdit.actionUndo)
menuEdit.addAction(self.textEdit.actionRedo)
menuEdit.addSeparator ()
menuEdit.addAction(self.textEdit.actionCopy)
menuEdit.addAction(self.textEdit.actionCut)
menuEdit.addAction(self.textEdit.actionPaste)
menuEdit.addSeparator ()
menuEdit.addAction(self.textEdit.actionLaunchFindDialog)
menuEdit.addAction(self.textEdit.actionFindNext)
menuEdit.addAction(self.textEdit.actionFindPrevious)
menuEdit.addAction(self.textEdit.actionLaunchCharWidgetTable)
menuEdit.addSeparator ()
# menuEdit.addAction(self.textEdit.actionChangeLanguage)
menuEdit.addAction(self.actionEditPreferences)
menuDocument = self.menuBar().addMenu ( "Document" )
menuDocument.addAction(self.textEdit.actionRecheckTypography)
menuDocument.addAction(self.textEdit.actionEnableTypo)
menuDocument.addSeparator ()
menuDocument.addAction(self.actionFileStats)
menuDocument.addAction(self.actionFileMetaData)
menuDocument.addSeparator ()
menuDocument.addAction(self.actionSendToExternalSoftware)
menuStyle = self.menuBar().addMenu ( "Styles" )
for id in TSManager.dictCharStyle.keys():
if id in self.textEdit.actionStylesDict.keys():
menuStyle.addAction(self.textEdit.actionStylesDict[id])
menuStyle.addSeparator ()
for id in TSManager.dictBlockStyle.keys():
if id in self.textEdit.actionStylesDict.keys():
menuStyle.addAction(self.textEdit.actionStylesDict[id])
menuStyle.addSeparator ()
menuStyle.addAction(self.textEdit.actionInsertImage)
menuStyle.addSeparator ()
menuStyle.addAction (self.textEdit.actionResetStyle)
menuView = self.menuBar().addMenu ( "View" )
menuView.addAction(self.actionViewFullScreen)
menuAbout = self.menuBar().addMenu ( "About" )
menuAbout.addAction(self.actionAboutHelp)
menuAbout.addAction(self.actionAboutAbout)
list_actions = [ self.actionFileNew,self.actionFileOpen,
self.actionFileImport,self.actionFileSave,self.actionFileSaveAs,
self.actionFileExport,self.actionFileStats,self.actionFileMetaData,
self.actionFileQuit,self.actionViewFullScreen,
self.textEdit.actionUndo,self.textEdit.actionRedo,
self.textEdit.actionCopy,self.textEdit.actionCut,
self.textEdit.actionPaste, self.textEdit.actionLaunchFindDialog,
self.textEdit.actionFindNext,self.textEdit.actionFindPrevious,
self.textEdit.actionLaunchCharWidgetTable,
# self.textEdit.actionChangeLanguage,
self.textEdit.actionRecheckTypography,
self.textEdit.actionEnableTypo,self.actionSendToExternalSoftware,
self.actionAboutHelp,self.actionAboutAbout] + \
self.textEdit.actionStylesDict.values()
for act in list_actions:
self.addAction(act)
################################## SLOTS ##################################
def SLOT_actionFileSave(self):
"""
Slot used when saving the current file.
"""
if self.filepath==None:
filepath = FMFileManagement.save_gui_filepath(
self.get_default_opening_saving_site(),
self,filter="AthW files (*.athw);; All files (*.*)")
else:
filepath = self.filepath
if filepath :
filepath=unicode(filepath)
progressBar = QtGui.QProgressBar(parent=self)
progressBar.setMaximum(0)
progressBar.setValue(0)
# center the widget
point = self.geometry().center()
point1 = progressBar.geometry().center()
progressBar.move(point-point1)
# set the window to the correct flag
progressBar.setWindowFlags(QtCore.Qt.SplashScreen)
progressBar.setWindowModality(QtCore.Qt.WindowModal)
progressBar.show()
# this class is a very classic thread just to save the file while
# displaying the progressBar:
class TaskThread(QtCore.QThread):
taskFinished = QtCore.pyqtSignal()
def run(self1):
self.CMD_FileSave(filepath=filepath)
self1.taskFinished.emit()
# What we have to do at the end of the saving
def finishing():
progressBar.close()
self.actionFileSave.setEnabled(False)
self.setWindowTitle("AthenaWriter : "+self.filepath)
tmp,filename = os.path.split(self.filepath)
self.changeMessageStatusBar("Has saved "+filename)
self.lastFiles.addFile(self.filepath)
self.myLongTask = TaskThread()
self.connect(self.myLongTask,QtCore.SIGNAL("taskFinished()"),
finishing)
# self.myLongTask.taskFinished.connect(finishing)
self.myLongTask.start()
def SLOT_actionFileSaveAs(self):
"""
Slot used when saving as the current file.
"""
filepath = FMFileManagement.save_gui_filepath(
self.get_default_opening_saving_site(),
self,filter="AthW files (*.athw);; All files (*.*)")
if filepath :
self.clean_tmp_files() # we remove the previous tmp files
self.CMD_FileSave(filepath=unicode(filepath))
self.actionFileSave.setEnabled(False)
self.setWindowTitle("AthenaWriter : "+self.filepath)
tmp,filename = os.path.split(self.filepath)
self.changeMessageStatusBar("Has saved "+filename)
self.lastFiles.addFile(self.filepath)
def SLOT_actionFileNew(self):
"""
Slot used when creating a new file.
"""
res=self.doSaveDialog()
if (res != QtGui.QMessageBox.Yes) and (res != QtGui.QMessageBox.No):
return False
self.clean_tmp_files() # we remove the previous tmp files
self.textEdit.setText("")
self.filepath=None
self.metadata=DPMetaData()
self.setWindowTitle("AthenaWriter : NewFile")
self.changeMessageStatusBar("Created a new file")
return True
def SLOT_actionFileOpen(self,filepath=None):
"""
If filepath==None, it will display the window to search the file
If filepath==None, it will display the window to search the file
"""
res=self.doSaveDialog()
if (res != QtGui.QMessageBox.Yes) and (res != QtGui.QMessageBox.No):
return False
if filepath==None:
filepath = FMFileManagement.open_gui_filepath(
self.get_default_opening_saving_site(),
self,filter="AthW files (*.athw);; All files (*.*)")
else :
filepath=str(filepath)
if filepath:
self.clean_tmp_files() # we remove the previous tmp files
try:
self.CMD_FileOpen(filepath)
except IOError:
# if the file do not exist, propose to crreate it
r = QtGui.QMessageBox.question(self,"Non existing file",
"The file "+filepath+" do not exist. Do you want to "+\
"create it ?",
QtGui.QMessageBox.Yes|QtGui.QMessageBox.No)
if r!= QtGui.QMessageBox.Yes:
return False
self.filepath = filepath
self.actionFileSave.setEnabled(False)
self.setWindowTitle("AthenaWriter : "+self.filepath)
tmp,filename = os.path.split(filepath)
self.changeMessageStatusBar("Has opened "+filename)
self.lastFiles.addFile(self.filepath)
return True
else:
return False
def SLOT_actionFileImport(self,filepath=None):
"""
If filepath==None, it will display the window to search the file
"""
res=self.doSaveDialog()
if (res != QtGui.QMessageBox.Yes) and (res != QtGui.QMessageBox.No):
return False
if filepath==None:
filepath = FMFileManagement.open_gui_filepath(
self.get_default_opening_saving_site(),
self)
else :
filepath=str(filepath)
if filepath:
self.clean_tmp_files() # we remove the previous tmp files
path,e = os.path.splitext(filepath)
e = e[1:]#the [1:] is to skip the dot in the extension
self.CMD_FileImport(filepath=filepath,format_name=e)
self.actionFileSave.setEnabled(True)
self.setWindowTitle("AthenaWriter : NewFile")
tmp,filename = os.path.split(filepath)
self.changeMessageStatusBar("Has imported "+filename)
return True
else:
return False
def SLOT_actionFileExport(self):
dft_opening_saving_site = self.get_default_opening_saving_site()
if self.filepath!=None:
res = DEDialog.getFields(
dft_opening_saving_site=dft_opening_saving_site,
default_path = self.filepath,
)
else:
res = DEDialog.getFields(
dft_opening_saving_site=dft_opening_saving_site,
)
if res :
try :
res1 = self.CMD_FileExport(**res)
except IOError,e:
print e
res1 = False
if res1:
msg = "Successfull exportation"
self.changeMessageStatusBar("Has exported "+res['filepath'])
dia = QtGui.QMessageBox.information (
self,
"Exportation" ,
msg)
else:
msg = "Unsuccessfull exportation :\n"
msg += str(e)
dia = QtGui.QMessageBox.critical (
self,
"Exportation" ,
msg)
class AWCore:
class Error (Exception):
def __init__(self,raison):
self.raison = raison
def __str__(self):
return self.raison.encode('ascii','replace')
def __init__(self):
self.textEdit = TETextEdit(language_name=
TLPreferences['DFT_WRITING_LANGUAGE'])
self.metadata = DPMetaData()
self.filepath=None
self.lastFiles=CLLastFiles(FMLastFilesFile.open())
def CMD_FileSave(self,filepath=None):
if filepath==None:
if self.filepath==None:
raise self.Error('Please specify the filepath')
filepath = self.filepath
else:
self.filepath = filepath
res = FMFileManagement.save(unicode(self.textEdit.toXml()),filepath)
if res :
if not self.metadata.isEmpty():
# we will save the file .athw_meta as well
cur = self.textEdit.textCursor()
self.metadata.__setitem__('lastpos', int(cur.position()),
protected=False)
self.metadata['language'] = self.textEdit.language.name
to_save = self.metadata.toxml()
meta_filepath,tmp = os.path.splitext(self.filepath)
meta_filepath += '.athw_meta'
res = FMFileManagement.save(to_save,meta_filepath)
def CMD_FileSaveCopy(self,filepath):
old_filepath = self.filepath
self.CMD_FileSave(filepath)
self.filepath = old_filepath
def CMD_FileOpen(self,filepath):
if filepath==None:
if self.filepath==None:
raise self.Error('Please specify the filepath')
filepath = self.filepath
else:
self.filepath=filepath
# Get the text:
local_dir,tmp = os.path.split(self.filepath)
text = FMFileManagement.open(filepath)
meta_filepath,tmp = os.path.splitext(self.filepath)
meta_filepath += '.athw_meta'
if os.path.exists(meta_filepath):
self.metadata=DPMetaData.init_from_xml_string(
FMFileManagement.open(meta_filepath))
else:
self.metadata=DPMetaData()
language = self.metadata['language']
lastpos = self.metadata['lastpos']
self.textEdit.setText(text,type='xml',new_language=language)
if lastpos != None:
cur = self.textEdit.textCursor()
cur.setPosition(lastpos)
self.textEdit.setTextCursor(cur)
return True
def CMD_FileImport(self,filepath,format_name,to_skip=None):
"""
- filepath : the path to the file to import
- format_name : the name of the format
- to_skip : the list of the lines to skip
"""
if to_skip==None : to_skip=[]
# We check whenever we can import this type of files
list_extentions = [format.extension for format in DIList]
if format_name not in list_extentions:
raise self.Error('Unknown format to import: choose in '+\
str(list_extentions))
fiimport_instance = DIList[list_extentions.index(format_name)](
file=filepath)
fiimport_instance.import2xml()
for i in to_skip:
fiimport_instance.skipLine(i)
text = fiimport_instance.text
self.textEdit.setText(text,type='xml')
self.filepath = None
def CMD_FileExport(self,format_name,filepath=None,check_typo=False,\
**kargs):
if check_typo:
self.textEdit.SLOT_actionRecheckTypography()
list_extentions = [format.name for format in DEList]
if format_name not in list_extentions:
raise self.Error('Unknown format to import: choose in '+\
str(list_extentions))
index = list_extentions.index(format_name)
format = DEList[index](TSManager)
format.export(self.textEdit,**kargs)
res = format.export_file(
# default_saving_site=self.get_default_opening_saving_site(),
parent=self,
filepath=filepath,
)
return res