python: Use vendored copy of secp256k1lab

Co-authored-by: Sebastian Falbesoner <[email protected]>

Author: real-or-random

python/.ruff.toml

@@ -1,4 +1,3 @@
 [format]
-# Exclude the file taken from Bitcoin Core for now, in case we want to
-# contribute our changes back to upstream.
-exclude = ["secp256k1lab/secp256k1.py"]
+# Exclude vendored package.
+exclude = ["secp256k1lab/*"]

python/chilldkg_ref/__init__.py

@@ -1 +1,8 @@
+from pathlib import Path
+import sys
+
+
 __all__ = ["chilldkg"]
+
+# Prefer the vendored copy of secp256k1lab.
+sys.path.insert(0, str(Path(__file__).parent / "../secp256k1lab/src"))

python/mypy.ini

@@ -0,0 +1,4 @@
+[mypy]
+# Include path to vendored copy of secp256k1lab, in order to
+# avoid "import-not-found" errors in mypy's `--strict` mode
+mypy_path = $MYPY_CONFIG_FILE_DIR/secp256k1lab/src

python/tests.py

@@ -7,9 +7,6 @@
 from typing import Tuple, List, Optional
 from secrets import token_bytes as random_bytes
 
-from secp256k1lab.secp256k1 import GE, G, Scalar
-from secp256k1lab.keys import pubkey_gen_plain
-
 from chilldkg_ref.util import (
     FaultyParticipantOrCoordinatorError,
     FaultyCoordinatorError,
@@ -21,6 +18,11 @@
 import chilldkg_ref.encpedpop as encpedpop
 import chilldkg_ref.chilldkg as chilldkg
 
+# Import from secp256k1lab after the chilldkg_ref imports because the latter
+# modifies sys.path to make sure the vendored copy of secp256k1lab is found.
+from secp256k1lab.secp256k1 import GE, G, Scalar
+from secp256k1lab.keys import pubkey_gen_plain
+
 from example import simulate_chilldkg_full as simulate_chilldkg_full_example