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']
