summaryrefslogtreecommitdiffhomepage
path: root/bazarr/languages/get_languages.py
blob: 881dd6ce063fbc43963960c8481bb601deed7ea5 (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
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
# coding=utf-8

import pycountry

from subzero.language import Language

from .custom_lang import CustomLanguage
from app.database import TableSettingsLanguages, database, insert, update, select


def load_language_in_db():
    # Get languages list in langs tuple
    langs = [{'code3': lang.alpha_3, 'code2': lang.alpha_2, 'name': lang.name, 'enabled': 0}
             for lang in pycountry.languages
             if hasattr(lang, 'alpha_2')]

    # Insert standard languages in database table
    database.execute(
        insert(TableSettingsLanguages)
        .values(langs)
        .on_conflict_do_nothing())

    # Update standard languages with code3b if available
    langs = [{'code3b': lang.bibliographic, 'code3': lang.alpha_3}
             for lang in pycountry.languages
             if hasattr(lang, 'alpha_2') and hasattr(lang, 'bibliographic')]

    # Update languages in database table
    database.execute(
        update(TableSettingsLanguages), langs)

    # Insert custom languages in database table
    CustomLanguage.register(TableSettingsLanguages)

    # Create languages dictionary for faster conversion than calling database
    create_languages_dict()


def create_languages_dict():
    global languages_dict
    # replace chinese by chinese simplified
    database.execute(
        update(TableSettingsLanguages)
        .values(name='Chinese Simplified')
        .where(TableSettingsLanguages.code3 == 'zho'))

    languages_dict = [{
        'code3': x.code3,
        'code2': x.code2,
        'name': x.name,
        'code3b': x.code3b,
    } for x in database.execute(
        select(TableSettingsLanguages.code3, TableSettingsLanguages.code2, TableSettingsLanguages.name,
               TableSettingsLanguages.code3b))
        .all()]

def audio_language_from_alpha2(lang):
    lang_map = {
        'Chinese': 'zh',
        'Portuguese (Brazil)': 'pb'
    }

    return language_from_alpha2(lang_map.get(lang, lang))

def language_from_alpha2(lang):
    return next((item['name'] for item in languages_dict if item['code2'] == lang[:2]), None)


def language_from_alpha3(lang):
    return next((item['name'] for item in languages_dict if lang[:3] in [item['code3'], item['code3b']]), None)


def alpha2_from_alpha3(lang):
    return next((item['code2'] for item in languages_dict if lang[:3] in [item['code3'], item['code3b']]), None)


def alpha2_from_language(lang):
    return next((item['code2'] for item in languages_dict if item['name'] == lang), None)


def alpha3_from_alpha2(lang):
    return next((item['code3'] for item in languages_dict if item['code2'] == lang[:2]), None)


def alpha3_from_language(lang):
    return next((item['code3'] for item in languages_dict if item['name'] == lang), None)


def get_language_set():
    languages = database.execute(
        select(TableSettingsLanguages.code3)
        .where(TableSettingsLanguages.enabled == 1))\
        .all()

    language_set = set()

    for lang in languages:
        custom = CustomLanguage.from_value(lang.code3, "alpha3")
        if custom is None:
            language_set.add(Language(lang.code3))
        else:
            language_set.add(custom.subzero_language())

    return language_set