from tarfile import TarFile, CompressionError, ReadError

class TarZstFile(TarFile):

  OPEN_METH = {
    **TarFile.OPEN_METH,
    'zst': 'zstopen'
  }

  @classmethod
  def zstopen(cls, name, mode='r', fileobj=None, level_or_option=None, zstd_dict=None, **kwargs):
    """Open zstd compressed tar archive name for reading or writing.
       Appending is not allowed.
    """
    if mode not in ("r", "w", "x"):
      raise ValueError("mode must be 'r', 'w' or 'x'")

    try:
      from pyzstd import ZstdFile, ZstdError
    except ImportError:
      raise CompressionError("pyzstd module is not available") from None

    fileobj = ZstdFile(fileobj or name, mode, level_or_option=level_or_option, zstd_dict=zstd_dict)

    try:
      tar = cls.taropen(name, mode, fileobj, **kwargs)
    except (ZstdError, EOFError) as exception:
      fileobj.close()
      if mode == 'r':
        raise ReadError('not a zstd file') from exception
      raise
    except:
      fileobj.close()
      raise

    tar._extfileobj = False
    return tar

with TarZstFile.open('archive.tar.zst', 'r') as tarzst:
  for tarinfo in tarzst:
    print(f'[{tarinfo.name}]')