iokit 0.0.1__tar.gz

iokit-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Vladislav A. Proskurov
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

iokit-0.0.1/PKG-INFO

@@ -0,0 +1,17 @@
+Metadata-Version: 2.1
+Name: iokit
+Version: 0.0.1
+Summary: Input Output Kit
+Author-email: "Vladislav A. Proskurov" <rilshok@pm.me>
+License: MIT
+Project-URL: Homepage, https://github.com/rilshok/iokit
+Project-URL: Repository, https://github.com/rilshok/iokit
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: humanize>=4.9.0
+Requires-Dist: typing-extensions>=4.8.0
+Requires-Dist: pytz>=2024.1
+
+# iok
+input / output kit

iokit-0.0.1/README.md

@@ -0,0 +1,2 @@
+# iok
+input / output kit

iokit-0.0.1/pyproject.toml

@@ -0,0 +1,38 @@
+[project]
+name = "iokit"
+description = "Input Output Kit"
+readme = "README.md"
+requires-python = ">=3.10"
+license = {text = "MIT"}
+dynamic = ["version"]
+authors = [
+    {name = "Vladislav A. Proskurov", email = "rilshok@pm.me"},
+]
+dependencies = [
+    "humanize>=4.9.0",
+    "typing-extensions>=4.8.0",
+    "pytz>=2024.1",
+]
+
+[tool.setuptools.dynamic]
+version = {attr = "iokit.__version__"}
+
+[project.urls]
+Homepage = "https://github.com/rilshok/iokit"
+Repository = "https://github.com/rilshok/iokit"
+
+[tool.mypy]
+strict = true
+
+[tool.ruff]
+line-length = 100
+
+[tool.ruff.format]
+docstring-code-format = true
+
+[tool.vulture]
+make_whitelist = true
+sort_by_size = true
+verbose = true
+min_confidence = 100
+paths = ["src/iokit"]

iokit-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

iokit-0.0.1/src/iokit/__init__.py

@@ -0,0 +1,9 @@
+__all__ = [
+    "State",
+    "Txt",
+    "Gzip",
+]
+__version__ = "0.0.1"
+
+from .extensions import Gzip, Txt
+from .state import State

iokit-0.0.1/src/iokit/extensions/__init__.py

@@ -0,0 +1,7 @@
+__all__ = [
+    "Txt",
+    "Gzip",
+]
+
+from .gz import Gzip
+from .txt import Txt

iokit-0.0.1/src/iokit/extensions/gz.py

@@ -0,0 +1,20 @@
+import gzip
+from io import BytesIO
+from typing import Any
+
+from iokit.state import State
+
+
+class Gzip(State, suffix="gz"):
+    def __init__(self, state: State, compression: int = 1, **kwargs: Any):
+        data = BytesIO()
+        gzip_file = gzip.GzipFile(fileobj=data, mode="wb", compresslevel=compression, mtime=0)
+        with gzip_file as gzip_buffer:
+            gzip_buffer.write(state.data.getvalue())
+        super().__init__(data=data, name=state.name, **kwargs)
+
+    def load(self) -> State:
+        gzip_file = gzip.GzipFile(fileobj=self.data, mode="rb")
+        with gzip_file as file:
+            data = file.read()
+        return State(data=data, name=str(self.name).removesuffix(".gz")).cast()

iokit-0.0.1/src/iokit/extensions/txt.py

@@ -0,0 +1,13 @@
+from typing import Any
+
+from iokit.state import State
+
+
+class Txt(State, suffix="txt"):
+    def __init__(self, data: str, **kwargs: Any):
+        if not isinstance(data, str):  # type: ignore
+            raise TypeError(f"Expected str, got {type(data).__name__}")
+        super().__init__(data=data.encode("utf-8"), **kwargs)
+
+    def load(self) -> str:
+        return self.data.getvalue().decode("utf-8")

iokit-0.0.1/src/iokit/state.py

@@ -0,0 +1,133 @@
+from datetime import datetime
+from io import BytesIO
+from typing import Any
+
+import pytz
+from humanize import naturalsize
+from typing_extensions import Self
+
+Payload = BytesIO | bytes
+
+
+def now() -> datetime:
+    return datetime.now(pytz.utc)
+
+
+class StateName:
+    def __init__(self, name: str):
+        self._name = name
+
+    @property
+    def stem(self) -> str:
+        split = self._name.split(sep=".", maxsplit=1)
+        return split[0]
+
+    @stem.setter
+    def stem(self, new: str) -> None:
+        suffix = ".".join(self.suffixes)
+        self._name = f"{new}.{suffix}"
+
+    @property
+    def suffix(self) -> str:
+        split = self._name.rsplit(sep=".", maxsplit=1)
+        if len(split) == 2:
+            return f"{split[-1]}"
+        raise ValueError(f"State name '{self._name}' does not have a suffix")
+
+    @property
+    def suffixes(self) -> tuple[str, ...]:
+        split = self._name.split(sep=".")[1:]
+        return tuple(f"{s}" for s in split)
+
+    def __str__(self) -> str:
+        return self._name
+
+    def __repr__(self) -> str:
+        return self._name
+
+    @classmethod
+    def make(cls, stem: "str | StateName", suffix: str) -> Self:
+        if suffix:
+            return cls(f"{stem}.{suffix}")
+        return cls(str(stem))
+
+
+class State:
+    _suffix: str = ""
+    _suffixes: tuple[str, ...] = ("",)
+
+    def __init__(self, data: Payload, name: str | StateName = "", time: datetime | None = None):
+        self._data = BytesIO(data) if isinstance(data, bytes) else data
+        self._name = StateName.make(name, self._suffix)
+        self._time = time or now()
+
+    def __init_subclass__(
+        cls,
+        suffix: str | None = None,
+        suffixes: tuple[str, ...] | None = None,
+    ) -> None:
+        if suffix is None and suffixes is not None:
+            if len(suffixes) == 0:
+                raise ValueError("State subclasses must define at least one suffix")
+            suffix = suffixes[0]
+        if suffix is not None and suffixes is None:
+            suffixes = (suffix,)
+
+        if suffix is not None and suffixes is not None:
+            if suffix not in suffixes:
+                suffixes = (suffix, *suffixes)
+
+        if suffix is None or suffixes is None:
+            raise ValueError("State subclasses must define a suffix or suffixes")
+
+        cls._suffix = suffix
+        cls._suffixes = suffixes
+
+    @property
+    def name(self) -> StateName:
+        return self._name
+
+    @name.setter
+    def name(self, value: str | StateName) -> None:
+        if isinstance(value, str):
+            value = StateName(value)
+        self._name = value
+
+    @property
+    def time(self) -> datetime:
+        return self._time
+
+    @time.setter
+    def time(self, value: datetime) -> None:
+        self._time = value
+
+    @property
+    def data(self) -> BytesIO:
+        self._data.seek(0)
+        return BytesIO(self._data.getvalue())
+
+    @property
+    def size(self) -> int:
+        return self._data.getbuffer().nbytes
+
+    def __repr__(self) -> str:
+        size = naturalsize(self.size, gnu=True)
+        return f"{self.name} ({size})"
+
+    def cast(self) -> "State":
+        suffix = self.name.suffix
+        for klass in State.__subclasses__():
+            if suffix in getattr(klass, "_suffixes"):
+                break
+        else:
+            raise ValueError(f"Unknown state suffix {suffix}")
+        state = klass.__new__(klass)
+        setattr(state, "_data", self.data)
+        setattr(state, "_name", self.name)
+        setattr(state, "_time", self.time)
+        return state
+
+    def load(self) -> Any:
+        if not self.name.suffix:
+            return self.data.getvalue()
+        return self.cast().load()

iokit-0.0.1/src/iokit.egg-info/PKG-INFO

@@ -0,0 +1,17 @@
+Metadata-Version: 2.1
+Name: iokit
+Version: 0.0.1
+Summary: Input Output Kit
+Author-email: "Vladislav A. Proskurov" <rilshok@pm.me>
+License: MIT
+Project-URL: Homepage, https://github.com/rilshok/iokit
+Project-URL: Repository, https://github.com/rilshok/iokit
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: humanize>=4.9.0
+Requires-Dist: typing-extensions>=4.8.0
+Requires-Dist: pytz>=2024.1
+
+# iok
+input / output kit

iokit-0.0.1/src/iokit.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+pyproject.toml
+src/iokit/__init__.py
+src/iokit/state.py
+src/iokit.egg-info/PKG-INFO
+src/iokit.egg-info/SOURCES.txt
+src/iokit.egg-info/dependency_links.txt
+src/iokit.egg-info/requires.txt
+src/iokit.egg-info/top_level.txt
+src/iokit/extensions/__init__.py
+src/iokit/extensions/gz.py
+src/iokit/extensions/txt.py

iokit-0.0.1/src/iokit.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

iokit-0.0.1/src/iokit.egg-info/requires.txt

@@ -0,0 +1,3 @@
+humanize>=4.9.0
+typing-extensions>=4.8.0
+pytz>=2024.1

iokit-0.0.1/src/iokit.egg-info/top_level.txt

@@ -0,0 +1 @@
+iokit