#-*- coding:utf-8 -*-
import os
from ..base import *

baidu_download_mp3 = True

@register([u'百度汉语', u'Baidu-Hanyu'])
class Baidu_Chinese(WebService):

    def __init__(self):
        super(Baidu_Chinese, self).__init__()

    def _get_content(self):
        url = u"http://dict.baidu.com/s?wd={}#basicmean".format(self.quote_word)
        html = self.get_response(url, timeout=10)
        soup = parse_html(html)
        result = {
            'pinyin': '',
            'basicmean': '',
            'detailmean': '',
            'fanyi': '',
            'audio_url': '',
        }

        #拼音
        element = soup.find('div', id='pinyin')
        if element:
            tag = element.find_all('b')
            if tag:
                result['pinyin'] = u' '.join(x.get_text() for x in tag)
            if tag:
                tag = element.find('a')
                if tag:
                    result['audio_url'] = tag.get('url')

        #基本释义
        element = soup.find('div', id='basicmean-wrapper')
        if element:
            tag = element.find_all('div', {'class': 'tab-content'})
            if tag:
                result['basicmean'] = u''.join(str(x).decode('utf-8') for x in tag)

        #详细释义
        element = soup.find('div', id='detailmean-wrapper')
        if element:
            tag = element.find_all('div', {'class': 'tab-content'})
            if tag:
                result['detailmean'] = u''.join(str(x).decode('utf-8') for x in tag)

        #英文翻译
        element = soup.find('div', id='fanyi-wrapper')
        if element:
            tag = element.find_all('dt')
            if tag:
                result['fanyi'] = u'<br>'.join(x.get_text().strip() for x in tag)

        return self.cache_this(result)

    def _get_field(self, key, default=u''):
        return self.cache_result(key) if self.cached(key) else self._get_content().get(key, default)

    @with_styles(need_wrap_css=True, cssfile='_baidu.css')
    def _css(self, val):
        return val

    @export([u'拼音', u'Phoneticize'])
    def fld_pinyin(self):
        return self._get_field('pinyin')

    @export('PRON')
    def fld_pron(self):
        audio_url = self._get_field('audio_url')
        if baidu_download_mp3 and audio_url:
            filename = u'_baidu_chinese_{}_.mp3'.format(self.word)
            filename = get_hex_name(self.unique.lower(), filename, 'mp3')
            try:
                if os.path.exists(filename) or self.net_download(
                    filename,
                    audio_url,
                    require=dict(mime='audio/mp3', size=512)
                ):
                    return self.get_anki_label(filename, 'audio')
            except:
                pass
        return ''

    @export([u'基本释义', u'Basic Definitions'])
    def fld_basic(self):
        val = self._get_field('basicmean')
        if val is None or val == '':
            return ''
        return self._css(val)

    @export([u'详细释义', u'Detail Definitions'])
    def fld_detail(self):
        val = self._get_field('detailmean')
        if val is None or val == '':
            return ''
        return self._css(val)

    @export([u'英文翻译', u'Translation[En]'])
    def fld_fanyi(self):
        return self._get_field('fanyi')