def testRemoveFolder(self):
rename = Rename(self.Tools.databaseDir, self.Tools.filetypesXML)
rename.addFolder(self.folder1)
rename.addFolder(self.folder1)
assert rename.removeFolder(self.folder1) == self.folder1
assert rename.removeFolder(self.folder1) == None
assert rename.removeFolder(
Folder(os.path.join(self.Tools.rootDir, self.Tools.testDirs[2]),
self.Tools.databaseDir)) == None
assert rename.removeFolder(
Folder(os.path.join(self.Tools.rootDir, self.Tools.testDirs[1]),
self.Tools.databaseDir)) == self.folder1
def setUp(self):
self.Tools = Tools()
self.Tools.createRootDir()
self.Tools.createTempFiles()
self.Tools.createDatabaseFiles()
self.Tools.createFilesystemXML()
self.database = Database(self.Tools.databaseDir)
self.database.loadDB()
self.folder1 = Folder(
os.path.join(self.Tools.rootDir, self.Tools.testDirs[0]),
self.Tools.databaseDir)
self.folder2 = Folder(
os.path.join(self.Tools.rootDir, self.Tools.testDirs[1]),
self.Tools.databaseDir)
self.folder3 = Folder(
os.path.join(self.Tools.rootDir, self.Tools.testDirs[2]),
self.Tools.databaseDir)
def testRenameFile(self):
self.folder1 = Folder(
os.path.join(self.Tools.rootDir, self.Tools.testDirs[0]),
self.Tools.databaseDir)
self.folder1.loadFiles()
self.filename1 = FileName('blackbooks.s01e02.avi', self.database)
self.filename1.getMatchingShows()
self.filename1.generatePreview(
self.Tools.filetypesXML, 'ext3',
'%show - %seasonx%episode - %title ( %arc - %airdate )')
assert os.listdir(
self.folder1.path) == ['bb.s03e05.avi', 'blackbooks.s01e02.avi']
self.filename1.renameFile(self.folder1.path)
assert os.listdir(self.folder1.path) == [
'bb.s03e05.avi',
"Black Books - 01x02 - Manny's First Day ( none - 6 October 2000 ).avi"
]
def testAddFoldersRecursively(self):
rename = Rename(self.Tools.databaseDir, self.Tools.filetypesXML)
rename.addFolder(self.folder1)
assert rename.addFoldersRecursively(Folder(
self.Tools.rootDir)) == Folder(self.Tools.rootDir)
assert [folder.path for folder in rename.folders] == [
'/tmp/veefire/Black Books', '/tmp/veefire', '/tmp/veefire/Spaced',
'/tmp/veefire/CSI', '/tmp/veefire/database'
]
rename.addFolder(self.folder1)
rename.addFolder(self.folder1)
rename.addFolder(Folder('/tmp/veefire/Spaced'))
rename.addFolder(Folder('/tmp/veefire/Spaced'))
rename.addFolder(Folder('/tmp/veefire/Spaced'))
assert rename.removeFolder(
Folder('/tmp/veefire')).path == '/tmp/veefire'
assert [folder.path for folder in rename.folders] == [
'/tmp/veefire/Black Books', '/tmp/veefire/Spaced',
'/tmp/veefire/CSI', '/tmp/veefire/database'
]
assert rename.removeFolder(
Folder('/tmp/veefire/Spaced')).path == '/tmp/veefire/Spaced'
assert rename.removeFolder(Folder('/tmp/veefire/Spaced')) == None
assert [folder.path for folder in rename.folders] == [
'/tmp/veefire/Black Books', '/tmp/veefire/CSI',
'/tmp/veefire/database'
]
