mirror of https://github.com/morpheus65535/bazarr
73 lines
2.9 KiB
Python
73 lines
2.9 KiB
Python
# coding=utf-8
|
|
|
|
from flask_restx import Resource, Namespace, reqparse
|
|
from operator import itemgetter
|
|
|
|
from app.database import TableHistory, TableHistoryMovie, TableSettingsLanguages, database, select
|
|
from languages.get_languages import alpha2_from_alpha3, language_from_alpha2, alpha3_from_alpha2
|
|
|
|
from ..utils import authenticate, False_Keys
|
|
|
|
api_ns_system_languages = Namespace('System Languages', description='Get languages list')
|
|
|
|
|
|
@api_ns_system_languages.route('system/languages')
|
|
class Languages(Resource):
|
|
get_request_parser = reqparse.RequestParser()
|
|
get_request_parser.add_argument('history', type=str, required=False, help='Language name for history stats')
|
|
|
|
@authenticate
|
|
@api_ns_system_languages.doc(parser=get_request_parser)
|
|
@api_ns_system_languages.response(200, 'Success')
|
|
@api_ns_system_languages.response(401, 'Not Authenticated')
|
|
def get(self):
|
|
"""List languages for history filter or for language filter menu"""
|
|
args = self.get_request_parser.parse_args()
|
|
history = args.get('history')
|
|
if history and history not in False_Keys:
|
|
languages = database.execute(
|
|
select(TableHistory.language)
|
|
.where(TableHistory.language.is_not(None)))\
|
|
.all()
|
|
languages += database.execute(
|
|
select(TableHistoryMovie.language)
|
|
.where(TableHistoryMovie.language.is_not(None)))\
|
|
.all()
|
|
languages_list = [lang.language.split(':')[0] for lang in languages]
|
|
languages_dicts = []
|
|
for language in languages_list:
|
|
code2 = None
|
|
if len(language) == 2:
|
|
code2 = language
|
|
elif len(language) == 3:
|
|
code2 = alpha2_from_alpha3(language)
|
|
else:
|
|
continue
|
|
|
|
if not any(x['code2'] == code2 for x in languages_dicts):
|
|
try:
|
|
languages_dicts.append({
|
|
'code2': code2,
|
|
'code3': alpha3_from_alpha2(code2),
|
|
'name': language_from_alpha2(code2),
|
|
# Compatibility: Use false temporarily
|
|
'enabled': False
|
|
})
|
|
except Exception:
|
|
continue
|
|
else:
|
|
languages_dicts = [{
|
|
'name': x.name,
|
|
'code2': x.code2,
|
|
'code3': x.code3,
|
|
'enabled': x.enabled == 1
|
|
} for x in database.execute(
|
|
select(TableSettingsLanguages.name,
|
|
TableSettingsLanguages.code2,
|
|
TableSettingsLanguages.code3,
|
|
TableSettingsLanguages.enabled)
|
|
.order_by(TableSettingsLanguages.name))
|
|
.all()]
|
|
|
|
return sorted(languages_dicts, key=itemgetter('name'))
|