we don't need contextlib here

This commit is contained in:
Robert Bendun 2024-04-18 00:30:25 +02:00
parent 16682a4ebd
commit e24033bc9e

2
bob.py
View File

@ -2,7 +2,6 @@ import argparse
import socket import socket
from otp import bob from otp import bob
from net import send, send_point, recv, recv_point from net import send, send_point, recv, recv_point
import contextlib
import sys import sys
def main(): def main():
@ -15,7 +14,6 @@ def main():
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as alice: with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as alice:
alice.connect((args.ip, args.port)) alice.connect((args.ip, args.port))
with contextlib.suppress(StopIteration):
b = bob(args.c > 0) b = bob(args.c > 0)
b.send(None) b.send(None)
send_point(alice, b.send(recv_point(alice))) send_point(alice, b.send(recv_point(alice)))