115 lines
2.8 KiB
Python
Executable File
115 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# @Author: KevinMidboe
|
|
# @Date: 2017-03-04 16:50:09
|
|
# @Last Modified by: KevinMidboe
|
|
# @Last Modified time: 2017-03-04 22:15:21
|
|
|
|
import os, sqlite3, re
|
|
from fuzzywuzzy import process
|
|
from langdetect import detect
|
|
from pprint import pprint
|
|
|
|
showDir = '/Volumes/media/tv/'
|
|
dbPath = 'shows.db'
|
|
mediaExtensions = ['mkv', 'mp4', 'avi']
|
|
subExtensions = ['srt']
|
|
|
|
def XOR(list1, list2):
|
|
return set(list1) ^ set(list2)
|
|
|
|
def getFuzzyName(query):
|
|
return process.extractOne(query, getShowNames().keys())
|
|
|
|
def removeUploader(mediaItem):
|
|
match = re.search('-[a-zA-Z\[\]\-]*.[a-z]{3}', mediaItem)
|
|
if match:
|
|
uploader, ext = match.group(0).split('.')
|
|
return uploader
|
|
|
|
return subtitles
|
|
|
|
def getLanguage(path, subtitles):
|
|
f = open(path + subtitles, 'r', encoding= 'ISO-8859-15')
|
|
language = detect(f.read())
|
|
f.close()
|
|
|
|
return language
|
|
|
|
# Finds the correct show name
|
|
def getShowNames():
|
|
conn = sqlite3.connect(dbPath)
|
|
c = conn.cursor()
|
|
|
|
c.execute('SELECT show_names, date_added, date_modified FROM shows')
|
|
|
|
returnList = {}
|
|
for name, added, modified in c.fetchall():
|
|
returnList[name] = [added, modified]
|
|
|
|
conn.close()
|
|
return returnList
|
|
|
|
def getNewFolderContents():
|
|
showNames = getShowNames().keys()
|
|
folderContents = filter( lambda f: not f.startswith('.'), os.listdir(showDir))
|
|
|
|
return XOR(folderContents, showNames)
|
|
|
|
def checkForSingleEpisodes(folderItem):
|
|
showName, hit = getFuzzyName(folderItem)
|
|
episodeMatch = re.findall(re.sub(' ', '.', showName)+'\.S[0-9]{1,2}E[0-9]{1,2}\.', folderItem)
|
|
|
|
if episodeMatch:
|
|
return True
|
|
|
|
def getByIdentifier(folderItem, identifier):
|
|
itemMatch = re.findall(identifier + '[0-9]{1,2}', folderItem)
|
|
item = re.sub(identifier, '', itemMatch[0])
|
|
return item
|
|
|
|
def getItemChildren(folder):
|
|
children = os.listdir(showDir + folder)
|
|
|
|
media_items = []
|
|
subtitles = []
|
|
trash = []
|
|
for childItem in children:
|
|
if childItem[-3:] in mediaExtensions:
|
|
media_items.append([childItem, removeUploader(childItem)])
|
|
elif childItem[-3:] in subExtensions:
|
|
subtitles.append([childItem, removeUploader(childItem), getLanguage(showDir + folder + '/', childItem)])
|
|
else:
|
|
trash.append(childItem)
|
|
|
|
return media_items, subtitles, trash
|
|
|
|
|
|
def getEpisodeInfo(folderItem):
|
|
showName, hit = getFuzzyName(folderItem)
|
|
season = getByIdentifier(folderItem, 'S')
|
|
episode = getByIdentifier(folderItem, 'E')
|
|
media_items, subtitles, trash = getItemChildren(folderItem)
|
|
|
|
episodeInfo = []
|
|
episodeInfo = {'original': folderItem,
|
|
'full_path': showDir + folderItem,
|
|
'name': showName,
|
|
'season': season,
|
|
'episode': episode,
|
|
'media_items': media_items,
|
|
'subtitles': subtitles,
|
|
'trash': trash}
|
|
|
|
|
|
return episodeInfo
|
|
|
|
|
|
def main():
|
|
for item in getNewFolderContents():
|
|
if checkForSingleEpisodes(item):
|
|
pprint(getEpisodeInfo(item))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |