diff --git a/findStray.py b/findStray.py index 6e0f035..54c2667 100755 --- a/findStray.py +++ b/findStray.py @@ -3,7 +3,7 @@ # @Author: KevinMidboe # @Date: 2017-03-04 16:50:09 # @Last Modified by: KevinMidboe -# @Last Modified time: 2017-03-05 00:49:03 +# @Last Modified time: 2017-03-05 13:09:10 import os, sqlite3, re from fuzzywuzzy import process @@ -103,12 +103,38 @@ def getEpisodeInfo(folderItem): 'episode': episode, 'media_items': media_items, 'subtitles': subtitles, - 'trash': trash} + 'trash': trash, + 'tweet_id': 'NULL', + 'verified': '0'} + addToDB(episodeInfo) return episodeInfo +def addToDB(episodeInfo): + conn = sqlite3.connect(dbPath) + c = conn.cursor() + + original = '"' + episodeInfo['original'] + '",' + full_path = '"' + episodeInfo['full_path'] + '",' + name = '"' + episodeInfo['name'] + '",' + season = '"' + episodeInfo['season'] + '",' + episode = '"' + episodeInfo['episode'] + '",' + media_items = '"' + str(episodeInfo['media_items']) + '",' + subtitles = '"' + str(episodeInfo['subtitles']) + '",' + trash = '"' + str(episodeInfo['trash']) + '",' + tweet_id = episodeInfo['tweet_id'] + ',' + verified = '"' + episodeInfo['verified'] + '"' + + try: + c.execute('INSERT INTO stray_episodes VALUES ('+ original + full_path + name + season\ + + episode + media_items + subtitles + trash + 'NULL,' + verified + ')') + except sqlite3.IntegrityError: + print('Episode already registered') + + conn.commit() + conn.close() def main(): for item in getNewFolderContents():