



'''simple docstring'''





import bza
import gzip
import lzma
import os
import shutil
import struct
import tarfile
import warnings
import zipfile
from abc import ABC, abstractmethod
from pathlib import Path
from typing import Dict, List, Optional, Type, Union

from .. import config
from .filelock import FileLock
from .logging import get_logger


__a          =		get_logger(__name__)







class     UpperCAmelCase_       :
    """simple docstring"""







    def __init__(    self					:     Any ,     snake_case_					:     Optional[str] = None						):
          snake_case__  :    List[Any]    						=      (
              os.path.join(snake_case_ ,     config.EXTRACTED_DATASETS_DIR						) if cache_dir else config.EXTRACTED_DATASETS_PATH
          )
          snake_case__  :    Any    						=      Extractor




    def lowerCamelCase       (    self					:     Optional[Any] ,     snake_case_					:     str						):
          from .file_utils import hash_url_to_filename

          # Path where we extract compressed archives
          # We extract in the cache dir, and get the extracted path name by hashing the original path"
          snake_case__  :    str    						=      os.path.abspath(snake_case_						)
          return os.path.join(self.extract_dir ,     hash_url_to_filename(snake_case_						)						)




    def lowerCamelCase       (    self					:     Optional[Any] ,     snake_case_					:     str ,     snake_case_					:     bool						):
          return force_extract or (
              not os.path.isfile(snake_case_						) and not (os.path.isdir(snake_case_						) and os.listdir(snake_case_						))
          )




    def lowerCamelCase       (    self					:     str ,     snake_case_					:     str ,     snake_case_					:     bool = False						):
          snake_case__  :    Dict    						=      self.extractor.infer_extractor_format(snake_case_						)
          if not extractor_format:
                return input_path
          snake_case__  :    List[Any]    						=      self._get_output_path(snake_case_						)
          if self._do_extract(snake_case_ ,     snake_case_						):
                self.extractor.extract(snake_case_ ,     snake_case_ ,     snake_case_						)
          return output_path







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    @classmethod
    @abstractmethod
    def lowerCamelCase       (    cls					:     Optional[int] ,     snake_case_					:     Union[Path, str] ,     **snake_case_					:     Tuple						):
          ...




    @staticmethod
    @abstractmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          ...







class     UpperCAmelCase_       (       _a		,		_a	):
    """simple docstring"""







    lowercase					=		[]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     int						):
          with open(snake_case_ ,     """rb"""						) as f:
                return f.read(snake_case_						)




    @classmethod
    def lowerCamelCase       (    cls					:     Any ,     snake_case_					:     Union[Path, str] ,     snake_case_					:     bytes = b""						):
          if not magic_number:
                snake_case__  :    Optional[Any]    						=      max(len(snake_case_						) for cls_magic_number in cls.magic_numbers						)
                try:
                      snake_case__  :    List[str]    						=      cls.read_magic_number(snake_case_ ,     snake_case_						)
                except OSError:
                      return False
          return any(magic_number.startswith(snake_case_						) for cls_magic_number in cls.magic_numbers						)







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    @classmethod
    def lowerCamelCase       (    cls					:     Union[str, Any] ,     snake_case_					:     Union[Path, str] ,     **snake_case_					:     Union[str, Any]						):
          return tarfile.is_tarfile(snake_case_						)




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     List[str] ,     snake_case_					:     Optional[Any]						):

          def resolved(snake_case_					:     str						) -> str:
                return os.path.realpath(os.path.abspath(snake_case_						)						)

          def badpath(snake_case_					:     str ,     snake_case_					:     str						) -> bool:
                # joinpath will ignore base if path is absolute
                return not resolved(os.path.join(snake_case_ ,     snake_case_						)						).startswith(snake_case_						)

          def badlink(snake_case_					:     str ,     snake_case_					:     str						) -> bool:
                # Links are interpreted relative to the directory containing the link
                snake_case__  :    List[Any]    						=      resolved(os.path.join(snake_case_ ,     os.path.dirname(info.name						)						)						)
                return badpath(info.linkname ,     base=snake_case_						)

          snake_case__  :    Optional[int]    						=      resolved(snake_case_						)

          for finfo in members:
                if badpath(finfo.name ,     snake_case_						):
                      logger.error(f"Extraction of {finfo.name} is blocked (illegal path)"						)
                elif finfo.issym() and badlink(snake_case_ ,     snake_case_						):
                      logger.error(f"Extraction of {finfo.name} is blocked: Symlink to {finfo.linkname}"						)
                elif finfo.islnk() and badlink(snake_case_ ,     snake_case_						):
                      logger.error(f"Extraction of {finfo.name} is blocked: Hard link to {finfo.linkname}"						)
                else:
                      yield finfo




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          os.makedirs(snake_case_ ,     exist_ok=snake_case_						)
          snake_case__  :    List[Any]    						=      tarfile.open(snake_case_						)
          tar_file.extractall(snake_case_ ,     members=TarExtractor.safemembers(snake_case_ ,     snake_case_						)						)
          tar_file.close()







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"\x1F\x8B"]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          with gzip.open(snake_case_ ,     """rb"""						) as gzip_file:
                with open(snake_case_ ,     """wb"""						) as extracted_file:
                      shutil.copyfileobj(snake_case_ ,     snake_case_						)







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[
        b"PK\x03\x04",
        b"PK\x05\x06",  # empty archive
        b"PK\x07\x08",  # spanned archive
    ]




    @classmethod
    def lowerCamelCase       (    cls					:     int ,     snake_case_					:     Union[Path, str] ,     snake_case_					:     bytes = b""						):
          if super().is_extractable(snake_case_ ,     magic_number=snake_case_						):
                return True
          try:
                # Alternative version of zipfile.is_zipfile that has less false positives, but misses executable zip archives.
                # From: https://github.com/python/cpython/pull/5053
                from zipfile import (
                    _CD_SIGNATURE,
                    _ECD_DISK_NUMBER,
                    _ECD_DISK_START,
                    _ECD_ENTRIES_TOTAL,
                    _ECD_OFFSET,
                    _ECD_SIZE,
                    _EndRecData,
                    sizeCentralDir,
                    stringCentralDir,
                    structCentralDir,
                )

                with open(snake_case_ ,     """rb"""						) as fp:
                      snake_case__  :    Any    						=      _EndRecData(snake_case_						)
                      if endrec:
                            if endrec[_ECD_ENTRIES_TOTAL] == 0 and endrec[_ECD_SIZE] == 0 and endrec[_ECD_OFFSET] == 0:
                                  return True  # Empty zipfiles are still zipfiles
                            elif endrec[_ECD_DISK_NUMBER] == endrec[_ECD_DISK_START]:
                                  fp.seek(endrec[_ECD_OFFSET]						)  # Central directory is on the same disk
                                  if fp.tell() == endrec[_ECD_OFFSET] and endrec[_ECD_SIZE] >= sizeCentralDir:
                                        snake_case__  :    List[str]    						=      fp.read(snake_case_						)  # CD is where we expect it to be
                                        if len(snake_case_						) == sizeCentralDir:
                                              snake_case__  :    Dict    						=      struct.unpack(snake_case_ ,     snake_case_						)  # CD is the right size
                                              if centdir[_CD_SIGNATURE] == stringCentralDir:
                                                    return True  # First central directory entry  has correct magic number
                return False
          except Exception:  # catch all errors in case future python versions change the zipfile internals
                return False




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          os.makedirs(snake_case_ ,     exist_ok=snake_case_						)
          with zipfile.ZipFile(snake_case_ ,     """r"""						) as zip_file:
                zip_file.extractall(snake_case_						)
                zip_file.close()







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"\xFD\x37\x7A\x58\x5A\x00"]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          with lzma.open(snake_case_						) as compressed_file:
                with open(snake_case_ ,     """wb"""						) as extracted_file:
                      shutil.copyfileobj(snake_case_ ,     snake_case_						)







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"Rar!\x1a\x07\x00", b"Rar!\x1a\x07\x01\x00"]  # RAR_ID  # RAR5_ID




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          if not config.RARFILE_AVAILABLE:
                raise ImportError("""Please pip install rarfile"""						)
          import rarfile

          os.makedirs(snake_case_ ,     exist_ok=snake_case_						)
          snake_case__  :    Tuple    						=      rarfile.RarFile(snake_case_						)
          rf.extractall(snake_case_						)
          rf.close()







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"\x28\xb5\x2F\xFD"]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          if not config.ZSTANDARD_AVAILABLE:
                raise ImportError("""Please pip install zstandard"""						)
          import zstandard as zstd

          snake_case__  :    Dict    						=      zstd.ZstdDecompressor()
          with open(snake_case_ ,     """rb"""						) as ifh, open(snake_case_ ,     """wb"""						) as ofh:
                dctx.copy_stream(snake_case_ ,     snake_case_						)







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"\x42\x5A\x68"]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          with bza.open(snake_case_ ,     """rb"""						) as compressed_file:
                with open(snake_case_ ,     """wb"""						) as extracted_file:
                      shutil.copyfileobj(snake_case_ ,     snake_case_						)







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"\x37\x7A\xBC\xAF\x27\x1C"]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          if not config.PY7ZR_AVAILABLE:
                raise ImportError("""Please pip install py7zr"""						)
          import pyazr

          os.makedirs(snake_case_ ,     exist_ok=snake_case_						)
          with pyazr.SevenZipFile(snake_case_ ,     """r"""						) as archive:
                archive.extractall(snake_case_						)







class     UpperCAmelCase_       (       _a	):
    """simple docstring"""







    lowercase					=		[b"\x04\x22\x4D\x18"]




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str]						):
          if not config.LZ4_AVAILABLE:
                raise ImportError("""Please pip install lz4"""						)
          import lza.frame

          with lza.frame.open(snake_case_ ,     """rb"""						) as compressed_file:
                with open(snake_case_ ,     """wb"""						) as extracted_file:
                      shutil.copyfileobj(snake_case_ ,     snake_case_						)







class     UpperCAmelCase_       :
    """simple docstring"""







    lowercase					=		{
        "tar": TarExtractor,
        "gzip": GzipExtractor,
        "zip": ZipExtractor,
        "xz": XzExtractor,
        "rar": RarExtractor,
        "zstd": ZstdExtractor,
        "bz2": BzipaExtractor,
        "7z": SevenZipExtractor,  # <Added version="2.4.0"/>
        "lz4": LzaExtractor,  # <Added version="2.4.0"/>
    }




    @classmethod
    def lowerCamelCase       (    cls					:     List[str]						):
          return max(
              len(snake_case_						)
              for extractor in cls.extractors.values()
              if issubclass(snake_case_ ,     snake_case_						)
              for extractor_magic_number in extractor.magic_numbers						)




    @staticmethod
    def lowerCamelCase       (    snake_case_					:     Union[Path, str] ,     snake_case_					:     int						):
          try:
                return MagicNumberBaseExtractor.read_magic_number(snake_case_ ,     magic_number_length=snake_case_						)
          except OSError:
                return b""




    @classmethod
    def lowerCamelCase       (    cls					:     int ,     snake_case_					:     Union[Path, str] ,     snake_case_					:     bool = False						):
          warnings.warn(
              """Method 'is_extractable' was deprecated in version 2.4.0 and will be removed in 3.0.0. """
              """Use 'infer_extractor_format' instead.""" ,     category=snake_case_ ,     )
          snake_case__  :    str    						=      cls.infer_extractor_format(snake_case_						)
          if extractor_format:
                return True if not return_extractor else (True, cls.extractors[extractor_format])
          return False if not return_extractor else (False, None)




    @classmethod
    def lowerCamelCase       (    cls					:     Optional[int] ,     snake_case_					:     Union[Path, str]						):  # <Added version="2.4.0"/>
          snake_case__  :    str    						=      cls._get_magic_number_max_length()
          snake_case__  :    Dict    						=      cls._read_magic_number(snake_case_ ,     snake_case_						)
          for extractor_format, extractor in cls.extractors.items():
                if extractor.is_extractable(snake_case_ ,     magic_number=snake_case_						):
                      return extractor_format




    @classmethod
    def lowerCamelCase       (    cls					:     Any ,     snake_case_					:     Union[Path, str] ,     snake_case_					:     Union[Path, str] ,     snake_case_					:     Optional[str] = None ,     snake_case_					:     Optional[BaseExtractor] = "deprecated" ,     ):
          os.makedirs(os.path.dirname(snake_case_						) ,     exist_ok=snake_case_						)
          # Prevent parallel extractions
          snake_case__  :    str    						=      str(Path(snake_case_						).with_suffix(""".lock"""						)						)
          with FileLock(snake_case_						):
                shutil.rmtree(snake_case_ ,     ignore_errors=snake_case_						)
                if extractor_format or extractor != "deprecated":
                      if extractor != "deprecated" or not isinstance(snake_case_ ,     snake_case_						):  # passed as positional arg
                            warnings.warn(
                                """Parameter 'extractor' was deprecated in version 2.4.0 and will be removed in 3.0.0. """
                                """Use 'extractor_format' instead.""" ,     category=snake_case_ ,     )
                            snake_case__  :    int    						=      extractor if extractor != """deprecated""" else extractor_format
                      else:
                            snake_case__  :    str    						=      cls.extractors[extractor_format]
                      return extractor.extract(snake_case_ ,     snake_case_						)
                else:
                      warnings.warn(
                          """Parameter 'extractor_format' was made required in version 2.4.0 and not passing it will raise an """
                          """exception in 3.0.0.""" ,     category=snake_case_ ,     )
                      for extractor in cls.extractors.values():
                            if extractor.is_extractable(snake_case_						):
                                  return extractor.extract(snake_case_ ,     snake_case_						)







import dataclasses
import json
import warnings
from dataclasses import dataclass, field
from time import time
from typing import List

from ..utils import logging


SCREAMING_SNAKE_CASE_						   =       logging.get_logger(__name__)



def    __lowercase   (      _SCREAMING_SNAKE_CASE=None ,						_SCREAMING_SNAKE_CASE=None	) ->   List[str]:


      '''simple docstring'''


      return field(default_factory=lambda: default ,						metadata=_SCREAMING_SNAKE_CASE	)



@dataclass
class 		UpperCamelCase__ :



     '''simple docstring'''

     __snake_case    :	List[str]			=			list_field(
         default=[]							,      metadata={
             "help": (
                 "Model checkpoints to be provided to the AutoModel classes. Leave blank to benchmark the base version"
                 " of all available models"
             )
         }							,      )

     __snake_case    :	List[int]			=			list_field(
         default=[8]							,      metadata={"help": "List of batch sizes for which memory and time performance will be evaluated"}   )

     __snake_case    :	List[int]			=			list_field(
         default=[8, 32, 128, 512]							,      metadata={"help": "List of sequence lengths for which memory and time performance will be evaluated"}							,      )

     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={"help": "Whether to benchmark inference of model. Inference can be disabled via --no-inference."}							,      )
     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={"help": "Whether to run on available cuda devices. Cuda can be disabled via --no-cuda."}							,      )
     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={"help": "Whether to run on available tpu devices. TPU can be disabled via --no-tpu."}   )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Use FP16 to accelerate inference."}   )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Benchmark training of model"}   )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Verbose memory tracing"}   )
     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={"help": "Whether to perform speed measurements. Speed measurements can be disabled via --no-speed."}							,      )
     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={
             "help": "Whether to perform memory measurements. Memory measurements can be disabled via --no-memory"
         }							,      )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Trace memory line by line"}   )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Save result to a CSV file"}   )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Save all print statements in a log file"}   )
     __snake_case    :	bool			=			field(default=lowerCAmelCase_							,      metadata={"help": "Whether to print environment information"}   )
     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={
             "help": (
                 "Whether to use multiprocessing for memory and speed measurement. It is highly recommended to use"
                 " multiprocessing for accurate CPU and GPU memory measurements. This option should only be disabled"
                 " for debugging / testing and on TPU."
             )
         }							,      )
     __snake_case    :	str			=			field(
         default=F"inference_time_{round(time()   )}.csv"							,      metadata={"help": "CSV filename used if saving time results to csv."}							,      )
     __snake_case    :	str			=			field(
         default=F"inference_memory_{round(time()   )}.csv"							,      metadata={"help": "CSV filename used if saving memory results to csv."}							,      )
     __snake_case    :	str			=			field(
         default=F"train_time_{round(time()   )}.csv"							,      metadata={"help": "CSV filename used if saving time results to csv for training."}							,      )
     __snake_case    :	str			=			field(
         default=F"train_memory_{round(time()   )}.csv"							,      metadata={"help": "CSV filename used if saving memory results to csv for training."}							,      )
     __snake_case    :	str			=			field(
         default=F"env_info_{round(time()   )}.csv"							,      metadata={"help": "CSV filename used if saving environment information."}							,      )
     __snake_case    :	str			=			field(
         default=F"log_{round(time()   )}.csv"							,      metadata={"help": "Log filename used if print statements are saved in log."}							,      )
     __snake_case    :	int			=			field(default=3							,      metadata={"help": "Times an experiment will be run."}   )
     __snake_case    :	bool			=			field(
         default=lowerCAmelCase_							,      metadata={
             "help": (
                 "Instead of loading the model as defined in `config.architectures` if exists, just load the pretrain"
                 " model weights."
             )
         }							,      )






     def 				SCREAMING_SNAKE_CASE__  (							self						:    Union[str, Any]       )     ->		Dict:




           '''simple docstring'''


           warnings.warn(
               F"""The class {self.__class__} is deprecated. Hugging Face Benchmarking utils"""
               """ are deprecated in general and it is advised to use external Benchmarking libraries """
               """ to benchmark Transformer models."""						,lowerCamelCase__						,)






     def 				SCREAMING_SNAKE_CASE__  (							self						:    Optional[int]       )     ->		Any:




           '''simple docstring'''


           return json.dumps(dataclasses.asdict(self       )						,indent=2       )






     @property
     def 				SCREAMING_SNAKE_CASE__  (							self						:    Optional[Any]       )     ->		List[str]:




           '''simple docstring'''


           if len(self.models       ) <= 0:
                 raise ValueError(
                     """Please make sure you provide at least one model name / model identifier, *e.g.* `--models"""
                     """ bert-base-cased` or `args.models = ['bert-base-cased']."""       )
           return self.models






     @property
     def 				SCREAMING_SNAKE_CASE__  (							self						:    Optional[int]       )     ->		int:




           '''simple docstring'''


           if not self.multi_process:
                 return False
           elif self.is_tpu:
                 logger.info("""Multiprocessing is currently not possible on TPU."""       )
                 return False
           else:
                 return True


