Added old project, half finished and in many parts. More great improvements coming soon

findStray.py

@@ -0,0 +1,115 @@
+#!/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()