jvdZdZddlmZmZddlmZddlmZm Z m Z ddl m Z ddl mZmZGdd eZd S) zbase translator classz"Copyright (C) 2020 Nidhal Baccouri)ABCabstractmethod)Path)ListOptionalUnion)GOOGLE_LANGUAGES_TO_CODES)InvalidSourceOrTargetLanguageLanguageNotSupportedExceptionceZdZdZdedddddfdedededed eed eed eeffd Ze d Z e j dZ e dZ e j dZ dZ dZdefdZ d#dedeeeffdZdedefdZededefdZdefdZdefdZdedefd Zd!eedeefd"ZxZS)$BaseTranslatorzX Abstract class that serve as a base translator for other different translators Nautoenbase_url languagessourcetarget payload_key element_tag element_queryc |||_||_t|j|_|st ||st ||||\|_|_||_ ||_ ||_ ||_ tdS)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__s EC:\Projects\Historical\venv\Lib\site-packages\deep_translator/base.pyr$zBaseTranslator.__init__s"#$()=)=)?)?$@$@! 8/77 7 8/77 7%)%?%?%O%O" dl%'+& c|jSNrr%s r(rzBaseTranslator.source3 |r)c||_dSr+r,r%langs r(rzBaseTranslator.source7  r)c|jSr+rr-s r(rzBaseTranslator.target;r.r)c||_dSr+r4r0s r(rzBaseTranslator.target?r2r)c|jjSr+)r'__name__r-s r(_typezBaseTranslator._typeCs ~&&r)c'K|D]k}||jvs|dkr|V(||jvr|j|VSt|d|jdS)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 rzSNo support for the provided language. Please select on of the supported languages: )messageN)rvaluesrr )r%rlanguages r(rz$BaseTranslator._map_language_to_codeFs"  H4?113333x67I7IT_113333oh/////3)))  r)returnc"|j|jkSr+)rrr-s r(_same_source_targetz"BaseTranslator._same_source_target[s|t|++r)Fas_dictc "|s|jn|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%r@kwargss r(get_supported_languagesz&BaseTranslator.get_supported_languages^s18Lt((T_Lr)r<c |dks6||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 rTF)rrr;)r%r<rBs r(is_language_supportedz$BaseTranslator.is_language_supportedisK   4?//11114?11333345r)textc tdS)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%rFrBs r( translatezBaseTranslator.translatexsKLLLr)fc4ddl}||SNr)docx2txtprocess)r%rJrMs r( _read_docxzBaseTranslator._read_docxs"""r)cvddl}||}|jd}|SrL)pypdf PdfReaderpages extract_text)r%rJrQreaderpages r( _read_pdfzBaseTranslator._read_pdfs9 ##|A  """r)pathc .t|tst|}|stdt d|j}|dkr$|t|}nz|dkr$|t|}nPt|dd5}| }d d d n #1swxYwY| |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)rJz.pdfrzutf-8)encodingN) isinstancerexistsprintexitsuffixrOstrrWopenreadstriprI)r%rXrBextrFrJs r(_translate_filezBaseTranslator._translate_files,$%% ::D{{}}  . / / / GGGk '>>??SYY?//DD F]]>>CII>..DDdC'222 (avvxx~~'' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (~~d###s'C77C;>C;batchc |stdg}t|D](\}}|j|fi|}||)|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 enumeraterIappend)r%rhrBarrirF translateds r(_translate_batchzBaseTranslator._translate_batchso  OMNN N '' # #GAt'7777J JJz " " " " r))F)r7 __module__ __qualname____doc__r rbdictrr$propertyrsetterrr8rboolr?rrrCrErrIrOrWrgrrp __classcell__)r's@r(r r sr 3%)%)(,    c] c] ~<X ]]X ]]'''*,T,,,,$ M M M tTz  M M M M c     McMMMM^M#C#### #3####$C$c$$$$8 d3i d3i        r)r N)rs __copyright__abcrrpathlibrtypingrrrdeep_translator.constantsr deep_translator.exceptionsr r r r)r(rs4 ########((((((((((?????? gggggSgggggr)