86 lines
2.4 KiB
Python
Executable File
86 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# @Author: KevinMidboe
|
|
# @Date: 2017-07-11 19:16:23
|
|
# @Last Modified by: kevinmidboe
|
|
# @Last Modified time: 2017-08-05 16:47:03
|
|
|
|
import fire, re, os
|
|
from pprint import pprint
|
|
|
|
class seasonMover(object):
|
|
''' Moving multiple files to multiple folders with
|
|
identifer '''
|
|
workingDir = os.getcwd()
|
|
mediaExt = ['mkv', 'mp4']
|
|
subExt = ['srt']
|
|
|
|
def create(self, name, interval):
|
|
pass
|
|
|
|
def move(self, fileSyntax, folderName):
|
|
episodeRange = self.findInterval(fileSyntax)
|
|
episodeList = self.getFiles()
|
|
|
|
self.motherMover(fileSyntax, folderName, episodeRange)
|
|
|
|
|
|
def getFiles(self):
|
|
epDir = os.path.join(self.workingDir)
|
|
dirContent = os.listDir(epDir)
|
|
fileList = sorted(dirContent)
|
|
return fileList
|
|
|
|
def getEpisodeNumber(self):
|
|
m = re.search('[eE][0-9]{1,2}', self.parent)
|
|
if m:
|
|
return re.sub('[eE]', '', m.group(0))
|
|
|
|
def findInterval(self, item):
|
|
if (re.search(r'\((.*)\)', item) is None):
|
|
raise ValueError('Need to declare an identifier e.g. (1..3) in: \n\t' + item)
|
|
|
|
start = int(re.search('\((\d+)\.\.', item).group(1))
|
|
end = int(re.search('\.\.(\d+)\)', item).group(1))
|
|
|
|
return list(range(start, end+1))
|
|
|
|
def removeUploadSign(self, file):
|
|
match = re.search('-[a-zA-Z\[\]\-]*.[a-z]{3}', file)
|
|
if match:
|
|
uploader = match.group(0)[:-4]
|
|
return re.sub(uploader, '', file)
|
|
|
|
return file
|
|
|
|
def motherMover(self, fileSyntax, folderName, episodeRange):
|
|
# Call for sub of fileList
|
|
# TODO check if range is same as folderContent
|
|
# print(fileSyntax, folderName, episodeRange)
|
|
mediaFiles = []
|
|
subtitleFiles = []
|
|
seasonDirectory = sorted(os.listdir(os.path.join(self.workingDir)))
|
|
|
|
for file in seasonDirectory:
|
|
if file[-3:] in self.mediaExt:
|
|
mediaFiles.append(file)
|
|
elif file[-3:] in self.subExt:
|
|
subtitleFiles.append(file)
|
|
|
|
if (len(mediaFiles) is not len(episodeRange)):
|
|
raise ValueError('Range defined does not match directory file content')
|
|
|
|
for epMedia, epNum in zip(mediaFiles, episodeRange):
|
|
leadingZeroNumber = "%02d" % epNum
|
|
fileName = re.sub(r'\((.*)\)', leadingZeroNumber, fileSyntax)
|
|
oldPath = os.path.join(self.workingDir, epMedia)
|
|
newFolder = os.path.join(self.workingDir, folderName + leadingZeroNumber)
|
|
newPath = os.path.join(newFolder, self.removeUploadSign(fileName))
|
|
|
|
# os.makedirs(newFolder)
|
|
# os.rename(oldPath, newPath)
|
|
print(oldPath + ' --> ' + newPath)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
fire.Fire(seasonMover) |