From 61e6b08b661a0c1477d000d9628ced4181be098f Mon Sep 17 00:00:00 2001 From: JJTech0130 Date: Mon, 28 Aug 2023 08:13:20 -0400 Subject: [PATCH] add alive option --- demo.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/demo.py b/demo.py index 2d20ca8..943d708 100644 --- a/demo.py +++ b/demo.py @@ -138,18 +138,12 @@ async def main(args: argparse.Namespace): print("Re-registering...") users = register(conn, users) + print(f"Done?") - # You CANNOT turn around and re-register like this: - # It will BREAK the tie between phone number and Apple ID - - # import emulated.nac - - # vd = emulated.nac.generate_validation_data() - # vd = b64encode(vd).decode() - - # users = ids.register(conn, [users[1]], vd) - - print(f"Done? {users}") + if args.alive: + logging.getLogger("apns").setLevel(logging.DEBUG) + while True: + await trio.sleep(20) @@ -270,6 +264,7 @@ async def output_task(im: imessage.iMessageUser): if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument("--reregister", action="store_true", help="Force re-registration") + parser.add_argument("--alive", action="store_true", help="Keep the connection alive") parser.add_argument("--client-data", action="store_true", help="Publish client data (only necessary for actually sending/receiving messages)") args = parser.parse_args() trio.run(main, args) \ No newline at end of file