Veronica K. B. Olsen
109 lines
2.6 KiB
Python
109 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""novelWriter Test Tools
|
|
"""
|
|
|
|
import os
|
|
import shutil
|
|
|
|
from itertools import chain
|
|
|
|
from PyQt5.QtWidgets import qApp
|
|
|
|
def cmpFiles(fileOne, fileTwo, ignoreLines=None):
|
|
"""Compare two files, but optionally ignore lines given by a list.
|
|
"""
|
|
if ignoreLines is None:
|
|
ignoreLines = []
|
|
|
|
try:
|
|
foOne = open(fileOne, mode="r", encoding="utf8")
|
|
except Exception as e:
|
|
print(str(e))
|
|
return False
|
|
|
|
try:
|
|
foTwo = open(fileTwo, mode="r", encoding="utf8")
|
|
except Exception as e:
|
|
print(str(e))
|
|
return False
|
|
|
|
txtOne = foOne.readlines()
|
|
txtTwo = foTwo.readlines()
|
|
|
|
if len(txtOne) != len(txtTwo):
|
|
print("Files are not the same length")
|
|
return False
|
|
|
|
diffFound = False
|
|
for n in range(len(txtOne)):
|
|
lnOne = txtOne[n].strip()
|
|
lnTwo = txtTwo[n].strip()
|
|
|
|
if n+1 in ignoreLines:
|
|
print("Ignoring line %d" % (n+1))
|
|
continue
|
|
|
|
if lnOne != lnTwo:
|
|
print("Diff on line %d:" % (n+1))
|
|
print(" << '%s'" % lnOne)
|
|
print(" >> '%s'" % lnTwo)
|
|
diffFound = True
|
|
|
|
foOne.close()
|
|
foTwo.close()
|
|
|
|
return not diffFound
|
|
|
|
def cmpList(listOne, listTwo):
|
|
"""Compare two iterable objects.
|
|
"""
|
|
flatOne = list(chain.from_iterable([listOne]))
|
|
flatTwo = list(chain.from_iterable([listTwo]))
|
|
if len(flatOne) != len(flatTwo):
|
|
return False
|
|
for i in range(len(flatOne)):
|
|
if flatOne[i] != flatTwo[i]:
|
|
return False
|
|
return True
|
|
|
|
def getGuiItem(theName):
|
|
"""Returns a QtWidget based on its objectName.
|
|
"""
|
|
for qWidget in qApp.topLevelWidgets():
|
|
if qWidget.objectName() == theName:
|
|
return qWidget
|
|
return None
|
|
|
|
def readFile(fileName):
|
|
"""Returns the content of a file as a string.
|
|
"""
|
|
with open(fileName, mode="r", encoding="utf8") as inFile:
|
|
return inFile.read()
|
|
|
|
def writeFile(fileName, fileData):
|
|
"""Write the contents of a string to a file.
|
|
"""
|
|
with open(fileName, mode="w", encoding="utf8") as outFile:
|
|
outFile.write(fileData)
|
|
|
|
def cleanProject(projPath):
|
|
"""Delete all generated files in a project.
|
|
"""
|
|
cacheDir = os.path.join(projPath, "cache")
|
|
if os.path.isdir(cacheDir):
|
|
shutil.rmtree(cacheDir)
|
|
|
|
metaDir = os.path.join(projPath, "meta")
|
|
if os.path.isdir(metaDir):
|
|
shutil.rmtree(metaDir)
|
|
|
|
bakFile = os.path.join(projPath, "nwProject.bak")
|
|
if os.path.isfile(bakFile):
|
|
os.unlink(bakFile)
|
|
|
|
tocFile = os.path.join(projPath, "ToC.txt")
|
|
if os.path.isfile(tocFile):
|
|
os.unlink(tocFile)
|
|
|
|
return
|