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

10
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,11 +14,10 @@ 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))) response = b.send(recv(alice))
response = b.send(recv(alice))
sys.stdout.buffer.write(response) sys.stdout.buffer.write(response)
if __name__ == "__main__": if __name__ == "__main__":