Make installable package

patrislav1 · Oct 18, 2023 · 8ffa1f1 · 8ffa1f1
1 parent 4a070f6
commit 8ffa1f1
Show file tree

Hide file tree

Showing 11 changed files with 35 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 __pycache__
 /.venv
+/dist
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,24 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "unisoc_unlock"
+version = "0.0.1"
+authors = [
+  { name="Patrick Huesmann", email="[email protected]" },
+]
+description = "Tool for unlocking a Unisoc/Spreadtrum Android bootloader"
+readme = "README.md"
+requires-python = ">=3.6"
+dependencies = [
+  "libusb1",
+  "pycryptodome"
+]
+
+[project.urls]
+"Homepage" = "https://github.com/pypa/sampleproject"
+
+[project.scripts]
+unisoc_unlock = "unisoc_unlock.cli:main"
+
diff --git a/requirements.txt b/requirements.txt
diff --git a/src/unisoc_unlock/__init__.py b/src/unisoc_unlock/__init__.py
diff --git a/src/unisoc_unlock/__main__.py b/src/unisoc_unlock/__main__.py
@@ -0,0 +1,3 @@
+from .cli import main
+
+main()
diff --git a/src/unisoc_unlock/bundled_adb/__init__.py b/src/unisoc_unlock/bundled_adb/__init__.py
diff --git a/bundled_adb/common.py → src/unisoc_unlock/bundled_adb/common.py b/bundled_adb/common.py → src/unisoc_unlock/bundled_adb/common.py
diff --git a/bundled_adb/fastboot.py → src/unisoc_unlock/bundled_adb/fastboot.py b/bundled_adb/fastboot.py → src/unisoc_unlock/bundled_adb/fastboot.py
diff --git a/bundled_adb/usb_exceptions.py → ...isoc_unlock/bundled_adb/usb_exceptions.py b/bundled_adb/usb_exceptions.py → ...isoc_unlock/bundled_adb/usb_exceptions.py
diff --git a/unisoc-unlock.py → src/unisoc_unlock/cli.py b/unisoc-unlock.py → src/unisoc_unlock/cli.py
@@ -1,7 +1,8 @@
 #!/usr/bin/env python3
 
+import os
 import sys
-from bundled_adb import fastboot, usb_exceptions
+from .bundled_adb import fastboot, usb_exceptions
 from Crypto.Signature import PKCS1_v1_5
 from Crypto.Hash import SHA256
 from Crypto.PublicKey import RSA
@@ -52,7 +53,11 @@ def main():
     print(f'OEM ID: {oem_id.id}')
     id = oem_id.id.ljust(2*64, '0')
     id_raw = base64.b16decode(id, casefold=True)
-    sgn = sign_token(id_raw, 'rsa4096_vbmeta.pem')
+    pemfile = os.path.join(
+        os.path.dirname(__file__),
+        'rsa4096_vbmeta.pem'
+    )
+    sgn = sign_token(id_raw, pemfile)
 
     print('Download signature')
     dev.Download(io.BytesIO(sgn), source_len=len(sgn))

diff --git a/rsa4096_vbmeta.pem → src/unisoc_unlock/rsa4096_vbmeta.pem b/rsa4096_vbmeta.pem → src/unisoc_unlock/rsa4096_vbmeta.pem