o NK&h@shdZdZddlmZmZddlmZddlmZm Z m Z ddl m Z ddl mZmZGdd d eZd S) zbase translator classz"Copyright (C) 2020 Nidhal Baccouri)ABCabstractmethod)Path)ListOptionalUnion)GOOGLE_LANGUAGES_TO_CODES)InvalidSourceOrTargetLanguageLanguageNotSupportedExceptioncs:eZdZdZdedddddfdedededed eed eed eeffd d Ze ddZ e j ddZ e ddZ e j ddZ ddZ ddZdefddZ d0dedeeeffddZdedefd d!Zed"edefd#d$Zd%efd&d'Zd%efd(d)Zd*edefd+d,Zd-eedeefd.d/ZZS)1BaseTranslatorzX Abstract class that serve as a base translator for other different translators Nautoenbase_url languagessourcetarget payload_key element_tag element_queryc  sn||_||_t|j|_|st||st||||\|_|_||_ ||_ ||_ ||_ t dS)zy @param source: source language to translate from @param target: target language to translate to N) _base_url _languageslistkeys_supported_languagesr _map_language_to_code_source_target _url_params _element_tag_element_queryrsuper__init__) selfrrrrrrr url_params __class__G/var/www/html/venv/lib/python3.10/site-packages/deep_translator/base.pyr!szBaseTranslator.__init__cC|jSNrr"r&r&r'r3zBaseTranslator.sourcecC ||_dSr)r*r"langr&r&r'r7 cCr(r)rr+r&r&r'r;r,zBaseTranslator.targetcCr-r)r1r.r&r&r'r?r0cCs|jjSr))r%__name__r+r&r&r'_typeCszBaseTranslator._typecgs\|D](}||jvs|dkr|Vq||jvr"|j|Vqt|d|jddS)a  map language to its corresponding code (abbreviation) if the language was passed by its full name by the user @param languages: list of languages @return: mapped value of the language or raise an exception if the language is not supported r zSNo support for the provided language. Please select on of the supported languages: )messageN)rvaluesrr )r"rlanguager&r&r'rFsz$BaseTranslator._map_language_to_codereturncCs |j|jkSr))rrr+r&r&r'_same_source_target[s z"BaseTranslator._same_source_targetFas_dictcKs|s|jS|jS)z return the supported languages by the Google translator @param as_dict: if True, the languages will be returned as a dictionary mapping languages to their abbreviations @return: list or dict )rr)r"r9kwargsr&r&r'get_supported_languages^s z&BaseTranslator.get_supported_languagesr6cKs,|dks||jvs||jvrdSdS)z check if the language is supported by the translator @param language: a string for 1 language @return: bool or raise an Exception r TF)rrr5)r"r6r:r&r&r'is_language_supportedis z$BaseTranslator.is_language_supportedtextcKstdS)z translate a text using a translator under the hood and return the translated text @param text: text to translate @param kwargs: additional arguments @return: str z+You need to implement the translate method!)NotImplemented)r"r=r:r&r&r' translatexs zBaseTranslator.translatefcCsddl}||SNr)docx2txtprocess)r"r@rBr&r&r' _read_docxs zBaseTranslator._read_docxcCs$ddl}||}|jd}|SrA)pypdf PdfReaderpages extract_text)r"r@rEreaderpager&r&r' _read_pdfs  zBaseTranslator._read_pdfpathcKst|ts t|}|stdtd|j}|dkr%|jt|d}n*|dkr2|jt|d}nt |ddd}| }Wd n1sJwY| |S) z translate directly from file @param path: path to the target file @type path: str @param kwargs: additional args @return: str zPath to the file is wrong!z.docx)r@z.pdfrzutf-8)encodingN) isinstancerexistsprintexitsuffixrDstrrKopenreadstripr?)r"rLr:extr=r@r&r&r'_translate_files  zBaseTranslator._translate_filebatchcKsB|stdg}t|D]\}}|j|fi|}||q |S)z translate a list of texts @param batch: list of texts you want to translate @return: list of translations z/Enter your text list that you want to translate) Exception enumerater?append)r"r[r:arrir= translatedr&r&r'_translate_batchs zBaseTranslator._translate_batch)F)r2 __module__ __qualname____doc__rrUdictrr!propertyrsetterrr3rboolr8rrr;r<rr?rDrKrZrrb __classcell__r&r&r$r'r s`        "r N)re __copyright__abcrrpathlibrtypingrrrdeep_translator.constantsrdeep_translator.exceptionsr r r r&r&r&r's