Browse Source

Scan files to library after rename

pull/62/head
Ruud 14 years ago
parent
commit
ac0be3da67
  1. 5
      couchpotato/core/plugins/renamer/main.py
  2. 26
      couchpotato/core/plugins/scanner/main.py

5
couchpotato/core/plugins/renamer/main.py

@ -280,10 +280,7 @@ class Renamer(Plugin):
log.info('Removing release %s' % release)
# Add this release to the library
if not group['destination_dir'] is destination:
fireEventAsync('scanner.to_library', folder = group['destination_dir'])
else:
log.error('Single destination folder not fully supported yet.')
fireEvent('scanner.files', folder = destination, files = rename_files)
# Search for trailers etc
fireEventAsync('renamer.after', group)

26
couchpotato/core/plugins/scanner/main.py

@ -72,10 +72,19 @@ class Scanner(Plugin):
addEvent('scanner.create_file_identifier', self.createStringIdentifier)
addEvent('scanner.scan', self.scan)
addEvent('scanner.to_library', self.scanToLibrary)
addEvent('scanner.files', self.scanToFilesLibrary)
addEvent('scanner.folder', self.scanToFolderLibrary)
addEvent('scanner.name_year', self.getReleaseNameYear)
def scanToLibrary(self, folder = None, files = []):
def scanToFilesLibrary(self, folder = None, files = None):
groups = self.scan(folder = folder, files = files)
for group in groups.itervalues():
if group['library']:
fireEvent('release.add', group = group)
def scanToFolderLibrary(self, folder = None):
if not os.path.isdir(folder):
return
@ -115,7 +124,7 @@ class Scanner(Plugin):
db.remove()
def scan(self, folder = None):
def scan(self, folder = None, files = []):
if not folder or not os.path.isdir(folder):
log.error('Folder doesn\'t exists: %s' % folder)
@ -124,10 +133,15 @@ class Scanner(Plugin):
# Get movie "master" files
movie_files = {}
leftovers = []
for root, dirs, files in os.walk(folder):
for filename in files:
file_path = os.path.join(root, filename)
# Scan all files of the folder if no files are set
if len(files) == 0:
files = []
for root, dirs, walk_files in os.walk(folder):
for filename in walk_files:
files.append(os.path.join(root, filename))
for file_path in files:
# Remove ignored files
if not self.keepFile(file_path):

Loading…
Cancel
Save