can read otp
This commit is contained in:
@@ -1,35 +1,38 @@
|
||||
import datetime
|
||||
import logging
|
||||
import re
|
||||
import sys
|
||||
import time
|
||||
|
||||
import serial
|
||||
from gsmmodem import GsmModem
|
||||
from gsmmodem.modem import SentSms
|
||||
|
||||
from commandor import Commandor
|
||||
from excel_reader import ExcelReader
|
||||
from pojo.serial_modem import SerialModem
|
||||
from utils.logging import init_logger
|
||||
|
||||
PORT = "/dev/tty.usbmodem11101"
|
||||
BAUDRATE = 115200
|
||||
|
||||
|
||||
# ser = serial.Serial(PORT, BAUDRATE, timeout=1)
|
||||
OTP_TIMEOUT = 120
|
||||
sms_received = False
|
||||
|
||||
|
||||
def get_devices_ports() -> list:
|
||||
return ["/dev/tty.usbmodem1121101",
|
||||
"/dev/tty.usbmodem1121103",
|
||||
"/dev/tty.usbmodem1121105",
|
||||
"/dev/tty.usbmodem1121107",
|
||||
"/dev/tty.usbmodem1121201",
|
||||
"/dev/tty.usbmodem1121203",
|
||||
"/dev/tty.usbmodem1121205",
|
||||
"/dev/tty.usbmodem1121207",
|
||||
"/dev/tty.usbmodem1121301",
|
||||
return ["/dev/tty.usbmodem121101",
|
||||
"/dev/tty.usbmodem121103",
|
||||
"/dev/tty.usbmodem121105",
|
||||
"/dev/tty.usbmodem121107",
|
||||
"/dev/tty.usbmodem121201",
|
||||
"/dev/tty.usbmodem121203",
|
||||
"/dev/tty.usbmodem121205",
|
||||
"/dev/tty.usbmodem121207",
|
||||
"/dev/tty.usbmodem121301",
|
||||
# "/dev/tty.usbmodem1121303",
|
||||
"/dev/tty.usbmodem1121305",
|
||||
"/dev/tty.usbmodem1121307",
|
||||
"/dev/tty.usbmodem1121401",
|
||||
"/dev/tty.usbmodem1121403",
|
||||
"/dev/tty.usbmodem1121405",
|
||||
"/dev/tty.usbmodem1121407"]
|
||||
"/dev/tty.usbmodem121305",
|
||||
"/dev/tty.usbmodem121307",
|
||||
"/dev/tty.usbmodem121401",
|
||||
"/dev/tty.usbmodem121403",
|
||||
"/dev/tty.usbmodem121405",
|
||||
"/dev/tty.usbmodem121407"]
|
||||
|
||||
|
||||
def has_sim(ser) -> bool:
|
||||
@@ -46,7 +49,7 @@ def send_command(cmd: str, ser) -> bytes:
|
||||
print("send command {}".format(cmd))
|
||||
ser.write(cmd.encode())
|
||||
msg = ser.read(100)
|
||||
print(msg)
|
||||
logger.info(msg)
|
||||
return msg
|
||||
|
||||
|
||||
@@ -55,73 +58,72 @@ def get_phone_number():
|
||||
send_command(cmd)
|
||||
|
||||
|
||||
def get_sim_ccid():
|
||||
cmd = "AT+CCID\r"
|
||||
send_command(cmd)
|
||||
|
||||
|
||||
def send_sms(sms_destination: str, msg: str):
|
||||
print('Initializing modem...')
|
||||
modem = GsmModem(PORT, BAUDRATE)
|
||||
modem.connect('0000')
|
||||
# modem.waitForNetworkCoverage(10)
|
||||
print('Sending SMS to: {0}'.format(sms_destination))
|
||||
|
||||
response = modem.sendSms(sms_destination, msg, True)
|
||||
if type(response) == SentSms:
|
||||
print('SMS Delivered.')
|
||||
else:
|
||||
print('SMS Could not be sent')
|
||||
|
||||
modem.close()
|
||||
|
||||
|
||||
def create_modem_for_port(port: str) -> GsmModem:
|
||||
print('Initializing modem... for ' + port)
|
||||
def create_modem_for_port(port: str) -> SerialModem:
|
||||
logger.info('Initializing modem... for ' + port)
|
||||
# Uncomment the following line to see what the modem is doing:
|
||||
init_logger()
|
||||
modem = GsmModem(port, BAUDRATE)
|
||||
modem.connect('0000')
|
||||
number = modem.ownNumber
|
||||
print("The SIM card phone number is:")
|
||||
print(number)
|
||||
logger.info("The SIM card phone number is:")
|
||||
logger.info(number)
|
||||
cmd = "AT+CCID\r"
|
||||
ccid = modem.write(cmd, True)
|
||||
print("The SIM card ccid is:" + " ".join(ccid))
|
||||
return modem
|
||||
response = modem.write(cmd, True)
|
||||
ccid = response[0].split(" ")[1].replace("\"", "")
|
||||
logger.info("The SIM card ccid is:" + ccid)
|
||||
|
||||
serial_modem = SerialModem(modem=modem, ccid=ccid)
|
||||
return serial_modem
|
||||
|
||||
|
||||
def start_to_handle_sms(modem: GsmModem):
|
||||
modem.smsReceivedCallback = handle_sms
|
||||
# modem = GsmModem(PORT, BAUDRATE, smsReceivedCallbackFunc=handle_sms)
|
||||
modem.smsTextMode = False
|
||||
# modem.connect('0000')
|
||||
print('Waiting for SMS message...')
|
||||
try:
|
||||
modem.rxThread.join(
|
||||
2 ** 31) # Specify a (huge) timeout so that it essentially blocks indefinitely, but still receives CTRL+C interrupt signal
|
||||
finally:
|
||||
modem.close()
|
||||
def start_to_handle_sms(serial_modem: SerialModem):
|
||||
serial_modem.modem.smsReceivedCallback = handle_sms
|
||||
global sms_received
|
||||
sms_received = False
|
||||
serial_modem.modem.smsTextMode = False
|
||||
logger.info('Waiting for SMS message, for phone number ' + str(serial_modem.phone_number))
|
||||
listen_at = time.time()
|
||||
while not sms_received:
|
||||
time.sleep(2)
|
||||
# check whether timeout
|
||||
now = time.time()
|
||||
if (listen_at + OTP_TIMEOUT) < now:
|
||||
logger.info("time out for {}, switch to next contact".format(serial_modem.phone_number))
|
||||
return
|
||||
return
|
||||
|
||||
|
||||
def handle_sms(sms):
|
||||
print(u'== SMS message received ==\nFrom: {0}\nTime: {1}\nMessage:\n{2}\n'.format(sms.number, sms.time, sms.text))
|
||||
logger.info(u'== SMS message received ==\nFrom: {0}\nTime: {1}\nMessage:\n{2}\n'.format(sms.number, sms.time, sms.text))
|
||||
global sms_received
|
||||
sms_received = True
|
||||
# extract the otp number
|
||||
logger.info("try to extract the otp")
|
||||
match = re.search(r'\d{6,8}', sms.text)
|
||||
logger.info("otp is " + match.group(0))
|
||||
|
||||
|
||||
def init_modems() -> list:
|
||||
modems = []
|
||||
for port in get_devices_ports():
|
||||
modems.append(create_modem_for_port(port))
|
||||
# read the contact, and contact the 2 objects together
|
||||
excel_reader = ExcelReader()
|
||||
contacts = excel_reader.read_contacts()
|
||||
for modem in modems:
|
||||
contact = [contact for contact in contacts if contact.ccid == modem.ccid][0]
|
||||
modem.phone_number = contact.phone
|
||||
modem.contact = contact
|
||||
return modems
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
# enable verbose logs for all port
|
||||
init_logger()
|
||||
logger = logging.getLogger()
|
||||
logger.addHandler(logging.StreamHandler(stream=sys.stdout))
|
||||
# create modems for all the available port
|
||||
modem_list = []
|
||||
for port in get_devices_ports():
|
||||
modem_list.append(create_modem_for_port(port))
|
||||
|
||||
# cmd = "AT+CMEE=2\r"
|
||||
# send_command(cmd)
|
||||
|
||||
# if has_sim():
|
||||
# # send_sms("+33649614591", "Modem pool test")
|
||||
# modem = show_own_phone_number()
|
||||
# get_sim_ccid()
|
||||
# # start_to_handle_sms(modem)
|
||||
# else:
|
||||
# print("no cart sim")
|
||||
modem_list = init_modems()
|
||||
# create listeners for chaque modem
|
||||
for modem in modem_list:
|
||||
start_to_handle_sms(modem)
|
||||
|
||||
Reference in New Issue
Block a user