def testAddFolder(self) : rename = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) assert rename.addFolder( self.folder1 ) == self.folder1 assert rename.addFolder( self.folder1 ) == None assert rename.addFolder( self.folder1 ) == self.folder1 assert rename.addFolder( Folder(os.path.join(self.Tools.rootDir, self.Tools.testDirs[1]), self.Tools.databaseDir) ) == None assert rename.addFolder( self.folder3 ) == self.folder3
def testRename(self): self.folder1.loadFiles() rename1 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename1.addFolder(self.folder1) rename1.addFolder(self.folder2) rename1.getMatchingShows() rename1.generatePreviews('ext3') assert os.listdir(self.folder1.path) == ['bb.s03e05.avi', 'blackbooks.s01e02.avi'] rename1.rename() assert os.listdir(self.folder1.path) == ["Black Books - S01E02 - Manny's First Day.avi", 'Black Books - S03E05 - The Travel Writer.avi']
def testRename(self): self.folder1.loadFiles() rename1 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename1.addFolder(self.folder1) rename1.addFolder(self.folder2) rename1.getMatchingShows() rename1.generatePreviews('ext3') assert os.listdir( self.folder1.path) == ['bb.s03e05.avi', 'blackbooks.s01e02.avi'] rename1.rename() assert os.listdir(self.folder1.path) == [ "Black Books - S01E02 - Manny's First Day.avi", 'Black Books - S03E05 - The Travel Writer.avi' ]
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 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' ]
def testGeneratePreviews(self): self.folder1.loadFiles() self.folder2.loadFiles() self.folder3.loadFiles() rename1 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename2 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename3 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename1.addFolder(self.folder1) rename1.addFolder(self.folder2) rename2.addFolder(self.folder1) rename2.addFolder(self.folder3) rename3.addFolder(self.folder3) rename1.getMatchingShows() rename2.getMatchingShows() rename3.getMatchingShows() assert [fo for fo in rename1.generatePreviews('ext3') ] == [[('bb.s03e05.avi', 'Black Books - S03E05 - The Travel Writer.avi'), ('blackbooks.s01e02.avi', "Black Books - S01E02 - Manny's First Day.avi")], [('csiS01E11.avi', 'C.S.I - S01E11 - I-15 Murders.avi'), ('CSI.2x12.avi', "C.S.I - S02E12 - You've Got Male.avi") ]] assert [ fo for fo in rename2.generatePreviews('ntfs', '%show[%season](%title)') ] == [[('bb.s03e05.avi', 'Black Books[03](The Travel Writer).avi'), ('blackbooks.s01e02.avi', 'Black Books[01](Mannys First Day).avi')], [('Spaced.S02E03.avi', 'Spaced[02](Mettle).avi'), ('Spaced.2x4.avi', 'Spaced[02](Help).avi')]] assert [ fo for fo in rename3.generatePreviews( 'ext3', '%show - %seasonx%episode - %title') ] == [[('Spaced.S02E03.avi', 'Spaced - 02x03 - Mettle.avi'), ('Spaced.2x4.avi', 'Spaced - 02x04 - Help.avi')]]
def testGetMatchingShows(self): self.folder1.loadFiles() self.folder2.loadFiles() self.folder3.loadFiles() rename1 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename2 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename3 = Rename(self.Tools.databaseDir, self.Tools.filetypesXML) rename1.addFolder(self.folder1) rename1.addFolder(self.folder2) rename2.addFolder(self.folder1) rename2.addFolder(self.folder3) rename3.addFolder(self.folder3) assert [ fn.CorrectShow.name for fo in rename1.getMatchingShows() for fn in fo.getMatchingShows() ] == ['Black Books', 'Black Books', 'C.S.I', 'C.S.I'] assert [ fn.CorrectShow.name for fo in rename2.getMatchingShows() for fn in fo.getMatchingShows() ] == ['Black Books', 'Black Books', 'Spaced', 'Spaced'] assert [ fn.CorrectShow.name for fo in rename3.getMatchingShows() for fn in fo.getMatchingShows() ] == ['Spaced', 'Spaced']
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']
def testGeneratePreviews(self): self.folder1.loadFiles() self.folder2.loadFiles() self.folder3.loadFiles() rename1 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename2 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename3 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename1.addFolder(self.folder1) rename1.addFolder(self.folder2) rename2.addFolder(self.folder1) rename2.addFolder(self.folder3) rename3.addFolder(self.folder3) rename1.getMatchingShows() rename2.getMatchingShows() rename3.getMatchingShows() assert [ fo for fo in rename1.generatePreviews('ext3') ] == [[('bb.s03e05.avi', 'Black Books - S03E05 - The Travel Writer.avi'), ('blackbooks.s01e02.avi', "Black Books - S01E02 - Manny's First Day.avi")], [('csiS01E11.avi', 'C.S.I - S01E11 - I-15 Murders.avi'), ('CSI.2x12.avi', "C.S.I - S02E12 - You've Got Male.avi")]] assert [ fo for fo in rename2.generatePreviews('ntfs', '%show[%season](%title)') ] == [[('bb.s03e05.avi', 'Black Books[03](The Travel Writer).avi'), ('blackbooks.s01e02.avi', 'Black Books[01](Mannys First Day).avi')], [('Spaced.S02E03.avi', 'Spaced[02](Mettle).avi'), ('Spaced.2x4.avi', 'Spaced[02](Help).avi')]] assert [ fo for fo in rename3.generatePreviews('ext3', '%show - %seasonx%episode - %title') ] == [[('Spaced.S02E03.avi', 'Spaced - 02x03 - Mettle.avi'), ('Spaced.2x4.avi', 'Spaced - 02x04 - Help.avi')]]
def testGetMatchingShows(self): self.folder1.loadFiles() self.folder2.loadFiles() self.folder3.loadFiles() rename1 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename2 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename3 = Rename( self.Tools.databaseDir, self.Tools.filetypesXML ) rename1.addFolder(self.folder1) rename1.addFolder(self.folder2) rename2.addFolder(self.folder1) rename2.addFolder(self.folder3) rename3.addFolder(self.folder3) assert [ fn.CorrectShow.name for fo in rename1.getMatchingShows() for fn in fo.getMatchingShows()] == ['Black Books', 'Black Books', 'C.S.I', 'C.S.I'] assert [ fn.CorrectShow.name for fo in rename2.getMatchingShows() for fn in fo.getMatchingShows()] == ['Black Books', 'Black Books', 'Spaced', 'Spaced'] assert [ fn.CorrectShow.name for fo in rename3.getMatchingShows() for fn in fo.getMatchingShows()] == ['Spaced', 'Spaced']