|
15 | 15 |
|
16 | 16 | from libp2p import new_host |
17 | 17 | from libp2p.bitswap import BitswapClient |
18 | | -from libp2p.crypto.ed25519 import create_new_key_pair |
19 | 18 | from libp2p.bitswap.cid import cid_to_bytes, format_cid_for_display |
20 | 19 | from libp2p.bitswap.dag import MerkleDag |
| 20 | +from libp2p.crypto.ed25519 import create_new_key_pair |
21 | 21 | from libp2p.peer.peerinfo import info_from_p2p_addr |
22 | 22 | from libp2p.utils.address_validation import ( |
23 | 23 | find_free_port, |
|
40 | 40 | DEFAULT_LISTEN_PORT = 4013 |
41 | 41 |
|
42 | 42 |
|
43 | | -def select_preferred_listen_addr( |
44 | | - addrs: list[Multiaddr], port: int |
45 | | -) -> Multiaddr: |
| 43 | +def select_preferred_listen_addr(addrs: list[Multiaddr], port: int) -> Multiaddr: |
46 | 44 | """Pick a stable, local-friendly address for copy/paste commands.""" |
47 | 45 | preferred_v4 = f"/ip4/127.0.0.1/tcp/{port}" |
48 | 46 | for addr in addrs: |
@@ -102,7 +100,7 @@ async def run_provider(file_path: str, port: int = 0, seed: str | None = None): |
102 | 100 | # Convert seed string to bytes (must be 32 bytes for Ed25519) |
103 | 101 | seed_bytes = hashlib.sha256(seed.encode()).digest() |
104 | 102 | key_pair = create_new_key_pair(seed=seed_bytes) |
105 | | - logger.info(f"Using deterministic peer ID from seed") |
| 103 | + logger.info("Using deterministic peer ID from seed") |
106 | 104 |
|
107 | 105 | host = new_host(key_pair=key_pair) |
108 | 106 |
|
@@ -164,9 +162,7 @@ def progress_callback(current: int, total: int, status: str): |
164 | 162 | # Prefer a deterministic local address for copy/paste commands. |
165 | 163 | transport_addrs = host.get_transport_addrs() |
166 | 164 | provider_addr = select_preferred_listen_addr(transport_addrs, port) |
167 | | - provider_addr = provider_addr.encapsulate( |
168 | | - Multiaddr(f"/p2p/{host.get_id()}") |
169 | | - ) |
| 165 | + provider_addr = provider_addr.encapsulate(Multiaddr(f"/p2p/{host.get_id()}")) |
170 | 166 | root_cid_text = format_cid_for_display(root_cid) |
171 | 167 | logger.info(f"Root CID: {root_cid_text}") |
172 | 168 | logger.info("") |
@@ -237,7 +233,7 @@ async def run_client( |
237 | 233 | # Convert seed string to bytes (must be 32 bytes for Ed25519) |
238 | 234 | seed_bytes = hashlib.sha256(seed.encode()).digest() |
239 | 235 | key_pair = create_new_key_pair(seed=seed_bytes) |
240 | | - logger.info(f"Using deterministic peer ID from seed") |
| 236 | + logger.info("Using deterministic peer ID from seed") |
241 | 237 |
|
242 | 238 | host = new_host(key_pair=key_pair) |
243 | 239 |
|
@@ -379,10 +375,7 @@ def parse_args(): |
379 | 375 | "--port", |
380 | 376 | type=int, |
381 | 377 | default=DEFAULT_LISTEN_PORT, |
382 | | - help=( |
383 | | - "Port to listen on (default: 4012). " |
384 | | - "Use 0 to auto-select a random port." |
385 | | - ), |
| 378 | + help=("Port to listen on (default: 4012). Use 0 to auto-select a random port."), |
386 | 379 | ) |
387 | 380 | parser.add_argument( |
388 | 381 | "--file", |
@@ -416,7 +409,10 @@ def parse_args(): |
416 | 409 | parser.add_argument( |
417 | 410 | "--seed", |
418 | 411 | type=str, |
419 | | - help="Seed string for deterministic peer ID generation (same seed = same peer ID)", |
| 412 | + help=( |
| 413 | + "Seed string for deterministic peer ID generation " |
| 414 | + "(same seed = same peer ID)" |
| 415 | + ), |
420 | 416 | ) |
421 | 417 |
|
422 | 418 | args = parser.parse_args() |
@@ -450,7 +446,9 @@ def main(): |
450 | 446 | if args.mode == "provider": |
451 | 447 | trio.run(run_provider, args.file, args.port, args.seed) |
452 | 448 | elif args.mode == "client": |
453 | | - trio.run(run_client, args.provider, args.cid, args.output, args.port, args.seed) |
| 449 | + trio.run( |
| 450 | + run_client, args.provider, args.cid, args.output, args.port, args.seed |
| 451 | + ) |
454 | 452 | except Exception as e: |
455 | 453 | logger.critical(f"Script failed: {e}", exc_info=True) |
456 | 454 | sys.exit(1) |
|
0 commit comments