Sneed-Group/pypush-plus-plus
1
0
Fork 0
mirror of https://github.com/Sneed-Group/pypush-plus-plus synced 2024-12-25 20:32:29 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix a bit

Browse source
This commit is contained in:
JJTech0130 2023-09-02 19:20:46 -04:00
parent e72cfbf72c
commit b5e9a1c4db
No known key found for this signature in database
GPG key ID: 23C92EBCCF8F93D6
2 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
demo.py
View file

@ -281,4 +281,9 @@ if __name__ == "__main__":
parser.add_argument("--gateway", type=str, help="Override the gateway phone number") parser.add_argument("--gateway", type=str, help="Override the gateway phone number")
args = parser.parse_args() args = parser.parse_args()
if not args.pdu.startswith("REG-RESP"):
print("Invalid REG-RESP PDU")
exit(1)
trio.run(main, args) trio.run(main, args)

8
sms_registration.py
View file

@ -21,7 +21,7 @@ def register(push_token: bytes, no_parse = False, gateway = GATEWAY) -> tuple[st
if no_parse: if no_parse:
print("Now do the next part and rerun with --pdu") print("Now do the next part and rerun with --pdu")
exit() exit()
parse_pdu(r, req_id) parse_pdu(r.text, req_id)
# if r.text.split("?")[0] != "REG-RESP": # if r.text.split("?")[0] != "REG-RESP":
# raise Exception(f"Failed to register: {r.text}") # raise Exception(f"Failed to register: {r.text}")
@ -37,10 +37,10 @@ def register(push_token: bytes, no_parse = False, gateway = GATEWAY) -> tuple[st
# return resp_params["n"], signature # return resp_params["n"], signature
def parse_pdu(r: str, req_id: int | None = None): def parse_pdu(r: str, req_id: int | None = None):
if r.text.split("?")[0] != "REG-RESP": if r.split("?")[0] != "REG-RESP":
raise Exception(f"Failed to register: {r.text}") raise Exception(f"Failed to register: {r}")
#assert r.text.split("?")[0] == "REG-RESP" #assert r.text.split("?")[0] == "REG-RESP"
resp_params = r.text.split("?")[1] resp_params = r.split("?")[1]
resp_params = resp_params.split(";") resp_params = resp_params.split(";")
resp_params = {param.split("=")[0]: param.split("=")[1] for param in resp_params} resp_params = {param.split("=")[0]: param.split("=")[1] for param in resp_params}
assert resp_params["v"] == "3" assert resp_params["v"] == "3"