mp_deye_connector.py

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

import socket
import ubinascii
from machine import WDT

from mp_deye_config import DeyeConfig

class DeyeConnector:

    def __init__(self, config: DeyeConfig):
        self.log_level = config.log_level
        self.wdt_enable = config.wdt_enable
        self.config = config.logger

    def send_request(self, req_frame):
        for res in socket.getaddrinfo(self.config.ip_address, self.config.port, socket.AF_INET, socket.SOCK_STREAM):
            family, socktype, proto, canonname, sockadress = res
            if self.wdt_enable: wdt = WDT()
            try:
                client_socket = socket.socket(family, socktype, proto)
                client_socket.settimeout(10)
                if self.wdt_enable: wdt.feed()
                client_socket.connect(sockadress)
            except: 
                if self.log_level <= 30: print("WARN: Could not open socket on IP ", self.config.ip_address)
                client_socket.close()
                return

            if self.log_level <= 10: print("DEBUG: Request frame: ", ubinascii.hexlify(req_frame))
            if self.wdt_enable: wdt.feed()
            client_socket.sendall(req_frame)

            attempts = 5
            while (attempts > 0):
                if self.wdt_enable: wdt.feed()
                attempts = attempts - 1
                try:
                    data = client_socket.recv(1024)
                    try:
                        data
                    except:
                        if self.log_level <= 30: print("WARN: No data received")
                    if self.log_level <= 10: print("DEBUG: Response frame: ", ubinascii.hexlify(data))
                    if self.wdt_enable: wdt.feed()
                    client_socket.close()
                    return data
                except:
                    if self.log_level <= 30: print("WARN: Connection timeout/error (send_request)")

        if self.wdt_enable: wdt.feed()
        client_socket.close()
        
        return bytearray()