mirror of
https://github.com/iluvcapra/wavinfo.git
synced 2025-12-31 08:50:41 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
from . import test_wave_parsing
|
|
|
|
from io import BytesIO
|
|
from typing import Generator
|
|
import zipfile as zf
|
|
import os.path
|
|
import os
|
|
from contextlib import contextmanager
|
|
|
|
# class TestFileLoader:
|
|
# """
|
|
# This guy manages the test_files archive.
|
|
# """
|
|
|
|
# def __init__(self, test_file_arch_name = 'archive.zip') -> None:
|
|
# self.base_path = os.path.join(os.path.dirname(__file__), "test_files")
|
|
# self.test_file_arch_name = test_file_arch_name
|
|
# self._gather_test_files_into_archive()
|
|
|
|
# @property
|
|
# def arch_path(self):
|
|
# return os.path.join(self.base_path, self.test_file_arch_name)
|
|
|
|
# @contextmanager
|
|
# def open(self, name) -> Generator[BytesIO]:
|
|
# z = zf.ZipFile(self.arch_path, 'r')
|
|
# member = z.open(name, 'r')
|
|
# try:
|
|
# yield member
|
|
# finally:
|
|
# zf.close()
|
|
|
|
# def _gather_test_files_into_archive(self):
|
|
# with zf.ZipFile(self.arch_path, 'a') as zip:
|
|
# for root, _, files in os.walk(self.base_path):
|
|
# for name in files:
|
|
# if root == self.base_path and name == self.test_file_arch_name:
|
|
# continue
|
|
# else:
|
|
# p = os.path.join(root, name)
|
|
# zip.write(p)
|
|
# os.unlink(p)
|
|
|
|
|