2018-09-17 00:27:00 +00:00
|
|
|
#!/usr/bin/env python
|
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
"""
|
|
|
|
|
JSON Utils
|
|
|
|
|
"""
|
|
|
|
|
import json
|
|
|
|
|
|
2020-05-20 15:29:39 +00:00
|
|
|
from six import text_type
|
2018-09-17 00:27:00 +00:00
|
|
|
from rebulk.match import Match
|
|
|
|
|
|
|
|
|
|
class GuessitEncoder(json.JSONEncoder):
|
|
|
|
|
"""
|
|
|
|
|
JSON Encoder for guessit response
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
def default(self, o): # pylint:disable=method-hidden
|
|
|
|
|
if isinstance(o, Match):
|
2020-05-20 15:29:39 +00:00
|
|
|
return o.advanced
|
|
|
|
|
if hasattr(o, 'name'): # Babelfish languages/countries long name
|
|
|
|
|
return text_type(o.name)
|
|
|
|
|
# pragma: no cover
|
|
|
|
|
return text_type(o)
|
