Sub-Zero.bundle/Contents/Code/__init__.py at 9cf1b759d7480c6930ef2a16896d5d1c800a8889 · pannal/Sub-Zero.bundle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# hdbits.org

import string, os, urllib, zipfile, re, copy
from babelfish import Language
from datetime import timedelta
import subliminal
import logger

OS_PLEX_USERAGENT = 'plexapp.com v9.0'

DEPENDENCY_MODULE_NAMES = ['subliminal', 'enzyme', 'guessit', 'requests']

SUPPORTED_PROVIDERS = ['opensubtitles', 'thesubdb', 'podnapisi', 'addic7ed', 'tvsubtitles']

ENABLED_PROVIDERS = {'opensubtitles' : Prefs['provider.opensubtitles'],
                     'thesubdb' : Prefs['provider.thesubdb'],
                     'podnapisi' : Prefs['provider.podnapisi'],
                     'addic7ed' : Prefs['provider.addic7ed'],
                     'tvsubtitles' : Prefs['provider.tvsubtitles']
                     }
# dict((key,value) for key, value in a.iteritems() if key == 1)
SUPPORTED_PROVIDER_SETTINGS = {'addic7ed': {
                                            'username': Prefs['provider.addic7ed.username'],
                                            'password': Prefs['provider.addic7ed.password']
                                            }
                               }

def Start():
    HTTP.CacheTime = 0
    HTTP.Headers['User-agent'] = OS_PLEX_USERAGENT
    Log.Debug("START CALLED")
    logger.registerLoggingHander(DEPENDENCY_MODULE_NAMES)

def ValidatePrefs():
    Log.Debug("Validate Prefs called.")
    return

# Prepare a list of languages we want subs for
def getLangList():
    langList = {Language.fromietf(Prefs["langPref1"])}
    if(Prefs["langPref2"] != "None"):
        langList.update({Language.fromietf(Prefs["langPref2"])})

    return langList

def getProviders():
    enabledProviders = []
    for supportedProvider in SUPPORTED_PROVIDERS:
        if Prefs["provider." + supportedProvider]:
            Log.Debug("Provider %s is enabled" % supportedProvider)
            enabledProviders.append(supportedProvider)
    return enabledProviders

def getProviderSettings():
    return SUPPORTED_PROVIDER_SETTINGS

class SubliminalSubtitlesAgentMovies(Agent.Movies):
    name = 'Subliminal Movie Subtitles'
    languages = [Locale.Language.English]
    primary_provider = False
    contributes_to = ['com.plexapp.agents.imdb']

    def search(self, results, media, lang):
        Log.Debug("MOVIE SEARCH CALLED")
        results.Append(MetadataSearchResult(id='null', score=100))


    def update(self, metadata, media, lang):
        videos = []
        Log.Debug("MOVIE UPDATE CALLED")
        for item in media.items:
            for part in item.parts:
                Log.Debug("Append: %s" % part.file)
                try:
                    scannedVideo = subliminal.video.scan_video(part.file, subtitles=True, embedded_subtitles=True)
                except ValueError:
                    Log.Warn("File could not be guessed by subliminal")
                    continue

                videos.append(scannedVideo)

        subtitles = subliminal.api.download_best_subtitles(videos, getLangList(), getProviders(), getProviderSettings())
        subliminal.api.save_subtitles(subtitles)

class SubliminalSubtitlesAgentTvShows(Agent.TV_Shows):

    name = 'Subliminal TV Subtitles'
    languages = [Locale.Language.English]
    primary_provider = False
    contributes_to = ['com.plexapp.agents.thetvdb']

    def search(self, results, media, lang):
        Log.Debug("TV SEARCH CALLED")
        results.Append(MetadataSearchResult(id='null', score=100))

    def update(self, metadata, media, lang):
        videos = []
        Log.Debug("TvUpdate. Lang %s" % lang)
        for season in media.seasons:
            for episode in media.seasons[season].episodes:
                for item in media.seasons[season].episodes[episode].items:
                    for part in item.parts:
                        Log.Debug("Append: %s" % part.file)
                        try:
                            scannedVideo = subliminal.video.scan_video(part.file, subtitles=True, embedded_subtitles=True)
                        except ValueError:
                            Log.Warn("File could not be guessed by subliminal")
                            continue

                        videos.append(scannedVideo)

        subtitles = subliminal.api.download_best_subtitles(videos, getLangList(), getProviders(), getProviderSettings())
        subliminal.api.save_subtitles(subtitles)