o qi @sdZddlZddlmZddlmZmZddlZddl m Z de defd d Z de d efd d Z dde de de dee fddZde fddZdS)z%Different file types read/write utilsN)Path)AnyOptional)URDUHACK_BASE_DIR file_namedatacCsBt|d}tj||tjdWddS1swYdS)z Save the python object in pickle format Args: file_name (str): file name data (Any): Any data type wb)protocolN)openpickledumpHIGHEST_PROTOCOL)rrfilerE/home/ubuntu/.local/lib/python3.10/site-packages/urduhack/utils/io.py pickle_dump s "rreturncCs8t|d }t|WdS1swYdS)zz Load the pickle file Args: file_name (str): file name Returns: Any: python object type rbN)r r load)rrrrr pickle_loads $rurl download_dir cache_dircCst|ts t|dt|tst|dt|ts$t|d|dur*t}t|jdddtjjj ||||dddS)aC Download anything from HTTP url Args: file_name (str): Save file as provided file name url (str): HTTP url download_dir (str): location to store file cache_dir (str): Main download dir Raises: TypeError: If any of the url, file_path and file_name are not str Type.  must be str type.NT)parentsexist_ok)fnameorigin cache_subdirrextract) isinstancestr TypeErrorrrmkdirtfkerasutilsget_file)rrrrrrrdownload_from_url&s   r)cCs(t|ts t|dt|dS)z Delete the local file Args: file_name (str): File to be deleted Raises: TypeError: If file_name is not str Type. FileNotFoundError: If file_name does not exist rN)r!r"r#runlink)rrrr remove_file@s r+)N)__doc__r pathlibrtypingrr tensorflowr%configrr"rrr)r+rrrrs