bytencrypt 0.1.0__tar.gz

bytencrypt-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Teraflop AI
+
+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.

bytencrypt-0.1.0/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: bytencrypt
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: infisicalsdk>=1.0.17
+Requires-Dist: libtorrent>=2.1.1
+Requires-Dist: pynacl>=1.6.2
+Requires-Dist: tqdm>=4.70.0
+Dynamic: license-file
+
+ruff check --select I --fix . && ruff format .

bytencrypt-0.1.0/README.md

@@ -0,0 +1 @@
+ruff check --select I --fix . && ruff format .

bytencrypt-0.1.0/pyproject.toml

@@ -0,0 +1,12 @@
+[project]
+name = "bytencrypt"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "infisicalsdk>=1.0.17",
+    "libtorrent>=2.1.1",
+    "pynacl>=1.6.2",
+    "tqdm>=4.70.0",
+]

bytencrypt-0.1.0/setup.cfg

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

bytencrypt-0.1.0/src/bytencrypt.egg-info/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: bytencrypt
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: infisicalsdk>=1.0.17
+Requires-Dist: libtorrent>=2.1.1
+Requires-Dist: pynacl>=1.6.2
+Requires-Dist: tqdm>=4.70.0
+Dynamic: license-file
+
+ruff check --select I --fix . && ruff format .

bytencrypt-0.1.0/src/bytencrypt.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+pyproject.toml
+src/downloader.py
+src/encryption.py
+src/torrent.py
+src/vault.py
+src/bytencrypt.egg-info/PKG-INFO
+src/bytencrypt.egg-info/SOURCES.txt
+src/bytencrypt.egg-info/dependency_links.txt
+src/bytencrypt.egg-info/requires.txt
+src/bytencrypt.egg-info/top_level.txt
+tests/test_encrypt_decrypt.py

bytencrypt-0.1.0/src/bytencrypt.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

bytencrypt-0.1.0/src/bytencrypt.egg-info/requires.txt

@@ -0,0 +1,4 @@
+infisicalsdk>=1.0.17
+libtorrent>=2.1.1
+pynacl>=1.6.2
+tqdm>=4.70.0

bytencrypt-0.1.0/src/bytencrypt.egg-info/top_level.txt

@@ -0,0 +1,4 @@
+downloader
+encryption
+torrent
+vault

bytencrypt-0.1.0/src/downloader.py

@@ -0,0 +1,29 @@
+import getpass
+import os
+import tempfile
+
+from encryption import FileCrypt
+from torrent import Torrent
+
+
+class Downloader:
+    def __init__(self, out="downloads", pw=None, stage=None, remove=True, **settings):
+        self.out, self.remove = out, remove
+        self.stage = stage or os.path.join(
+            "/dev/shm" if os.path.isdir("/dev/shm") else tempfile.gettempdir(),
+            str(os.getuid()),
+        )
+        os.makedirs(self.stage, 0o700, exist_ok=True)
+        self.fc = FileCrypt(pw or getpass.getpass("password: "))
+        self.torrent = Torrent(self.stage, **settings)
+
+    def get(self, *srcs):
+        handles = [self.torrent.add(s) for s in srcs]
+        done = []
+        for h in handles:
+            self.torrent.wait(h)
+            for f in self.torrent.files(h):
+                dst = os.path.join(self.out, os.path.relpath(f, self.stage)) + ".enc"
+                os.makedirs(os.path.dirname(dst), exist_ok=True)
+                done.append(self.fc.encrypt(f, out=dst, remove=self.remove))
+        return done

bytencrypt-0.1.0/src/encryption.py

@@ -0,0 +1,59 @@
+import os
+
+import nacl.pwhash
+import nacl.secret
+import nacl.utils
+
+
+class FileCrypt:
+    CHUNK = 1 << 20
+    A = nacl.secret.Aead
+    KDF = nacl.pwhash.argon2id
+    BLOCK = A.NONCE_SIZE + CHUNK + A.MACBYTES
+
+    def __init__(self, pw):
+        self.pw = pw.encode() if isinstance(pw, str) else pw
+
+    def _aead(self, salt):
+        key = self.KDF.kdf(
+            self.A.KEY_SIZE,
+            self.pw,
+            salt,
+            self.KDF.OPSLIMIT_MODERATE,
+            self.KDF.MEMLIMIT_MODERATE,
+        )
+        return self.A(key)
+
+    @staticmethod
+    def _aad(i, last):
+        return i.to_bytes(8, "big") + bytes([last])
+
+    def encrypt(self, path, out=None, remove=True):
+        out = out or path + ".enc"
+        salt = nacl.utils.random(self.KDF.SALTBYTES)
+        aead = self._aead(salt)
+        with open(path, "rb") as f, open(out, "wb") as o:
+            o.write(salt)
+            i, last = 0, False
+            while not last:
+                chunk = f.read(self.CHUNK)
+                last = len(chunk) < self.CHUNK
+                o.write(aead.encrypt(chunk, aad=self._aad(i, last)))
+                i += 1
+        if remove:
+            os.remove(path)
+        return out
+
+    def decrypt(self, path, out=None, remove=True):
+        out = out or path.removesuffix(".enc")
+        with open(path, "rb") as f, open(out, "wb") as o:
+            aead = self._aead(f.read(self.KDF.SALTBYTES))
+            i, last = 0, False
+            while not last:
+                block = f.read(self.BLOCK)
+                last = len(block) < self.BLOCK
+                o.write(aead.decrypt(block, aad=self._aad(i, last)))
+                i += 1
+        if remove:
+            os.remove(path)
+        return out

bytencrypt-0.1.0/src/torrent.py

@@ -0,0 +1,51 @@
+import os
+import time
+
+import libtorrent as lt
+from tqdm import tqdm
+
+
+class Torrent:
+    SETTINGS = {
+        "listen_interfaces": "0.0.0.0:6881",
+        "out_enc_policy": lt.enc_policy.forced,
+        "in_enc_policy": lt.enc_policy.forced,
+        "allowed_enc_level": lt.enc_level.rc4,
+    }
+
+    def __init__(self, save="downloads", **settings):
+        self.save = save
+        os.makedirs(save, exist_ok=True)
+        self.ses = lt.session({**self.SETTINGS, **settings})
+
+    def add(self, src):
+        if src.startswith("magnet:"):
+            p = lt.parse_magnet_uri(src)
+        else:
+            p = lt.add_torrent_params()
+            p.ti = lt.torrent_info(src)
+        p.save_path = self.save
+        return self.ses.add_torrent(p)
+
+    @staticmethod
+    def wait(h, interval=1):
+        s = h.status()
+        with tqdm(total=s.total_wanted, unit="B", unit_scale=True, desc=s.name) as bar:
+            while not s.is_seeding:
+                time.sleep(interval)
+                s = h.status()
+                bar.total = s.total_wanted
+                bar.update(s.total_wanted_done - bar.n)
+                bar.set_postfix(peers=s.num_peers, refresh=False)
+
+    def files(self, h):
+        fs = h.torrent_file().files()
+        paths = (
+            os.path.join(self.save, fs.file_path(i)) for i in range(fs.num_files())
+        )
+        return [p for p in paths if os.path.isfile(p)]
+
+    def download(self, src):
+        h = self.add(src)
+        self.wait(h)
+        return self.files(h)

bytencrypt-0.1.0/src/vault.py

@@ -0,0 +1,32 @@
+import os
+
+from infisical_sdk import InfisicalSDKClient
+
+
+class Vault:
+    def __init__(
+        self,
+        project_id=None,
+        env="prod",
+        path="/",
+        host=None,
+        client_id=None,
+        client_secret=None,
+    ):
+        self.project_id = project_id or os.environ["INFISICAL_PROJECT_ID"]
+        self.env, self.path = env, path
+        self.client = InfisicalSDKClient(
+            host=host or os.environ.get("INFISICAL_HOST", "https://app.infisical.com")
+        )
+        self.client.auth.universal_auth.login(
+            client_id or os.environ["INFISICAL_CLIENT_ID"],
+            client_secret or os.environ["INFISICAL_CLIENT_SECRET"],
+        )
+
+    def get(self, name):
+        return self.client.secrets.get_secret_by_name(
+            secret_name=name,
+            project_id=self.project_id,
+            environment_slug=self.env,
+            secret_path=self.path,
+        ).secretValue

bytencrypt-0.1.0/tests/test_encrypt_decrypt.py

@@ -0,0 +1,5 @@
+from encryption import FileCrypt
+
+fc = FileCrypt("password")
+fc.encrypt("tests/2505.09388v1.pdf")
+fc.decrypt("tests/2505.09388v1.pdf.enc", remove=False)