add wait for sms py

wait_for_sms.py

@@ -0,0 +1,136 @@
+import datetime
+import logging
+import sys
+import time
+from typing import Union
+
+from gsmmodem import GsmModem
+
+import params
+from commandor_page import CommandorPage
+from logs.AppLogging import init_logger
+from modems.ModemPool import ModemPool
+from modems.card_pool import CardPool
+from params import MODEM_POOL_PORTS, CARD_POOL_PORT, firebase_store_manager, oracle_log_sender
+from pojo.ReserveResultPojo import ReserveResultPojo
+from pojo.serial_modem import SerialModem
+from utils.excel_reader import ExcelHelper
+
+OTP_TIMEOUT = 120
+is_finished = False
+commandor = CommandorPage()
+thread_event = None
+current_gsm_modem = None
+card_pool = CardPool(CARD_POOL_PORT)
+# used to save the current slot position
+current_card_pool_slot = 1
+current_sim_position = 1
+
+
+def get_devices_ports() -> list:
+    return MODEM_POOL_PORTS
+
+def create_modem_for_port(port: str) -> Union[SerialModem, None]:
+    logger.info('Initializing modem... for ' + port)
+    serial_modem = None
+    try:
+        modem = GsmModem(port)
+        return SerialModem(modem=modem)
+    except Exception as ext:
+        logger.error(ext)
+        return serial_modem
+
+
+def timeout_occurred(serial_modem: SerialModem):
+    logger.info("will close timeout modem")
+    serial_modem.modem.close()
+    commandor.reset_air_plan_mode()
+
+
+def start_to_handle_sms(serial_modem: SerialModem):
+    global current_gsm_modem
+    current_gsm_modem = serial_modem.modem
+    current_gsm_modem.deleteMultipleStoredSms(memory="SM")
+    serial_modem.modem.smsReceivedCallback = handle_sms
+    global is_finished
+    is_finished = 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 is_finished:
+        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))
+            # save the contact in timeout
+            timeout_occurred(serial_modem)
+            current_gsm_modem.close()
+            return
+    return
+
+
+def handle_sms(sms):
+    logger.info(
+        u'== SMS message received ==\nFrom: {0}\nTime: {1}\nMessage:\n{2}\n'.format(sms.number, sms.time, sms.text))
+    # extract the otp number
+    params.oracle_log_sender.send_sms_reception_log(sms.number, sms.text)
+
+
+def init_modems() -> list:
+    modems = []
+    for port in get_devices_ports():
+        serial_modem = create_modem_for_port(port)
+        if serial_modem:
+            modems.append(serial_modem)
+    return modems
+
+
+def on_message_received(ch, method, properties, body):
+    print(str(body))
+    print(" [x] Received {} {}".format(body, datetime.datetime.now()))
+    #     parse the received message
+    result = ReserveResultPojo.from_json(body)
+    result.sim_position = current_sim_position
+    result.slot_position = current_card_pool_slot
+    logger.info(result)
+    firebase_store_manager.save(result)
+    oracle_log_sender.send_appoint_result(result)
+    # set the flag to True
+    global is_finished
+    is_finished = True
+
+
+def start_book():
+    slot_number = 1
+    slot_sum = 2
+    for i in reversed(range(slot_number, slot_sum + 1)):
+        card_pool.reset()
+        logger.info("will switch to " + str(i))
+        card_pool.switch_to_slot(i)
+        modem_pool = ModemPool(get_devices_ports())
+        modem_pool.reset_all_modems()
+        modem_list = init_modems()
+        # create listeners for chaque modem
+        # read the contact, and contact the 2 objects together
+        excel_reader = ExcelHelper()
+        contacts = excel_reader.read_contacts()
+        global current_sim_position
+        current_sim_position = 1
+        for modem in modem_list:
+            current_sim_position = current_sim_position + 1
+            try:
+                # get contact for current modem
+                modem.get_ccid()
+                start_to_handle_sms(modem)
+            except Exception as error:
+                print(error)
+                continue
+
+
+if __name__ == '__main__':
+    init_logger()
+    logger = logging.getLogger()
+    logger.addHandler(logging.StreamHandler(stream=sys.stdout))
+    start_book()
+    # card_pool.switch_to_slot(12)