aboutsummaryrefslogtreecommitdiffhomepage
path: root/custom_libs/subliminal_patch/extensions.py
blob: 800a1d547cb3bcec14c5144852e0036af0e41744 (plain)
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
# coding=utf-8
from __future__ import absolute_import
from collections import OrderedDict

import subliminal
import babelfish


class ProviderRegistry(object):
    providers = None

    def __init__(self):
        self.providers = OrderedDict()

    def __cmp__(self, d):
        return cmp(self.providers, d)

    def __contains__(self, item):
        return item in self.providers

    def __setitem__(self, key, item):
        self.providers[key] = item

    def __iter__(self):
        return iter(self.providers)

    def __getitem__(self, key):
        if key in self.providers:
            return self.providers[key]

    def __repr__(self):
        return repr(self.providers)

    def __str__(self):
        return str(self.providers)

    def __len__(self):
        return len(self.providers)

    def __delitem__(self, key):
        del self.providers[key]

    def register(self, name, cls):
        self.providers[name] = cls

    def names(self):
        return list(self.providers.keys())


provider_registry = ProviderRegistry()

# add language converters
try:
    babelfish.language_converters.unregister('addic7ed = subliminal.converters.addic7ed:Addic7edConverter')
except ValueError:
    pass

babelfish.language_converters.register('addic7ed = subliminal_patch.language:PatchedAddic7edConverter')
babelfish.language_converters.register('szopensubtitles = subliminal_patch.language:PatchedOpenSubtitlesConverter')
subliminal.refiner_manager.register('sz_metadata = subliminal_patch.refiners.metadata:refine')
subliminal.refiner_manager.register('sz_omdb = subliminal_patch.refiners.omdb:refine')
subliminal.refiner_manager.register('sz_tvdb = subliminal_patch.refiners.tvdb:refine')
subliminal.refiner_manager.register('drone = subliminal_patch.refiners.drone:refine')
subliminal.refiner_manager.register('filebot = subliminal_patch.refiners.filebot:refine')
subliminal.refiner_manager.register('file_info_file = subliminal_patch.refiners.file_info_file:refine')
subliminal.refiner_manager.register('symlinks = subliminal_patch.refiners.symlinks:refine')