Merge branch 'main' of github.com:waketzheng/carstino

Author: waketzheng

pip_conf.py

@@ -30,8 +30,8 @@
 """
 
 __author__ = "waketzheng@gmail.com"
-__updated_at__ = "2025.11.22"
-__version__ = "0.8.7"
+__updated_at__ = "2025.11.26"
+__version__ = "0.8.9"
 import contextlib
 import functools
 import os
@@ -608,7 +608,7 @@ def check_installed(self):
 
 
 class UvMirror(Mirror):
-    GITHUB_PROXY = "https://hub.gitmirror.com/"
+    GITHUB_PROXY = "https://hk.gh-proxy.org/"
     PYTHON_DOWNLOAD_URL = (
         "https://github.com/astral-sh/python-build-standalone/releases/download"
     )
@@ -622,10 +622,39 @@ def allow_insecure(url):
             return ""
         return '\nallow-insecure-host=["{}"]'.format(parse_host(url))
 
+    @classmethod
+    def _get_python_mirror(cls, default):
+        # type: (str) -> str
+        choices = {"default": default}
+        for name in ("PIP_CONF_PYTHON_MIRROR", "UV_PYTHON_INSTALL_MIRROR"):
+            value = os.getenv(name)
+            if value and value not in choices.values():
+                choices[name] = value
+        if len(choices) == 1:
+            return default
+        tip = "Which url do you want to set for python install mirror? (Leave blank to use default)"
+        items = list(choices.items())
+        choices_text = "\n".join(
+            [
+                "{i}. {key} ({value})".format(i=i, key=key, value=value)
+                for i, (key, value) in enumerate(items)
+            ]
+        )
+        a = input(tip + "\n" + choices_text + "\n").strip()
+        if a:
+            try:
+                url = items[int(a)][1]
+            except (TypeError, ValueError, IndexError):
+                print("Invalid choice, default will be used.")
+            else:
+                return url
+        return default
+
     @classmethod
     def python_install_mirror(cls):
         # type: () -> str
-        return cls.TEMPLATE.format(cls.GITHUB_PROXY + cls.PYTHON_DOWNLOAD_URL)
+        default = cls.GITHUB_PROXY + cls.PYTHON_DOWNLOAD_URL
+        return cls.TEMPLATE.format(cls._get_python_mirror(default))
 
     def build_content(self, url=None, extra_index=None):
         # type: (Optional[str], Optional[str]) -> str
@@ -660,8 +689,8 @@ def set(self, set_python_mirror=False):
         config_toml_path = os.path.join(dirpath, filename)
         text = self.build_content()
         if os.path.exists(config_toml_path):
-            with open(config_toml_path) as f:
-                content = f.read().strip()
+            with open(config_toml_path, "rb") as f:
+                content = f.read().strip().decode("utf-8")
             if text in content:
                 print("uv mirror set as expected. Skip!")
                 return None
@@ -752,8 +781,8 @@ def fix_v1_6_error(cls):
             print("WARNING: plugin file not found {}".format(file))
             return
         s = "semver"
-        with open(file) as f:
-            text = f.read()
+        with open(file, "rb") as f:
+            text = f.read().decode("utf-8")
         if s in text:
             text = text.replace(s, "constraints")
             with open(file, "w") as f:
@@ -838,8 +867,8 @@ def set(self):
         item = "[plugins.pypi_mirror]"
         text = item + '{}url = "{}"'.format(os.linesep, self.url)
         if os.path.exists(config_toml_path):
-            with open(config_toml_path) as f:
-                content = f.read().strip()
+            with open(config_toml_path, "rb") as f:
+                content = f.read().strip().decode("utf-8")
             if text in content:
                 print("poetry mirror set as expected. Skip!")
                 return None
@@ -904,8 +933,8 @@ def init_pip_conf(
     text = TEMPLATE.format(url, parse_host(url))
     conf_file = get_conf_path(is_windows, at_etc)
     if os.path.exists(conf_file):
-        with open(conf_file) as fp:
-            s = fp.read()
+        with open(conf_file, "rb") as fp:
+            s = fp.read().decode("utf-8")
         if text in s:
             print("Pip source already be configured as expected.\nSkip!")
             return None
@@ -936,8 +965,8 @@ def can_set_global():
 
 def read_lines(filename):
     # type: (str) -> list[str]
-    with open(filename) as f:
-        s = f.read()
+    with open(filename, "rb") as f:
+        s = f.read().decode("utf-8")
     return s.splitlines()
 
 

tests/test_box.py

@@ -0,0 +1,9 @@
+from waketbox import DigitGame
+
+
+def test_box():
+    game = DigitGame("helloworld")
+    raw = "32roqworisdonasdfjqiqirqieru938345ev !@#$%^&***()_+\n\t\r\n"
+    pad = game.black(raw)
+    assert len(pad) <= len(raw) * 4 + game.token_length
+    assert game.white(pad) == raw

waketbox.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+"""
+Waket Box
+
+Usage::
+    FIX_ZWJ=xxx python <me>.py <path-to-file>.txt
+
+"""
+
+from __future__ import annotations
+
+import base64
+import os
+import random
+import string
+import sys
+import zlib
+from functools import cached_property
+from pathlib import Path
+
+
+class DigitGame:
+    def __init__(self, token: str) -> None:
+        self._token = token
+
+    @property
+    def token(self) -> str:
+        return self._token
+
+    @cached_property
+    def token_length(self) -> int:
+        return len(self.token)
+
+    def is_padded(self, text: str) -> bool:
+        if len(text) % self.token_length != 1:
+            return False
+        return text[-1] in string.ascii_lowercase[: self.token_length]
+
+    def black(self, raw: str) -> str:
+        b1 = zlib.compress(raw.encode())
+        b2 = base64.urlsafe_b64encode(b1)
+        s1 = b2.strip(b"=").hex()
+        len1 = len(s1)
+        multi = int(len1 // self.token_length) + bool(len1 % self.token_length)
+        tokens = self.token * multi
+        delta = len(tokens) - len1
+        last_char = string.ascii_lowercase[delta]
+        if delta:
+            choices = string.digits + "abcdef"
+            s1 += "".join(random.choice(choices) for _ in range(delta))
+        chars = [chr(int(a, 16) ^ ord(b)) for a, b in zip(s1, tokens, strict=True)]
+        return "".join(chars) + last_char
+
+    def white(self, pad: str) -> str:
+        last_char = pad[-1]
+        delta = string.ascii_lowercase.index(last_char)
+        pad = pad[:-1]
+        tokens = self.token * (len(pad) // self.token_length)
+        unpad = [ord(p) ^ ord(t) for p, t in zip(pad, tokens, strict=True)]
+        if delta:
+            unpad = unpad[:-delta]
+        s1 = "".join(hex(i)[2:] for i in unpad)
+        b2 = bytes.fromhex(s1)
+        len1 = len(b2)
+        mod = len1 % 4
+        if mod:
+            b2 += b"=" * (4 - mod)
+        b1 = base64.urlsafe_b64decode(b2)
+        return zlib.decompress(b1).decode()
+
+
+def main():
+    if len(sys.argv) < 2:
+        print(__doc__.replace("<me>", Path(__file__).stem))
+        return
+    file = Path(sys.argv[1])
+    text = file.read_text()
+    box = DigitGame(os.environ["FIX_ZWJ"])
+    if "--pad" not in sys.argv and box.is_padded(text):
+        raw = box.white(text)
+        new_file = file.with_name(file.name + ".raw")
+        new_file.write_text(raw)
+        print(f"Save raw to {new_file}")
+    else:
+        padded = box.black(text)
+        new_file = file.with_name(file.name + ".pad")
+        new_file.write_text(padded)
+        print(f"Create pad to {new_file}")
+
+
+if __name__ == "__main__":
+    main()