HYTOS

# coding: utf-8

""" This is AppWebService module """

# import urllib.request

# import urllib.parse

import json

import requests

import cv2

import sys

import base64

import numpy as np

from AppDocData import AppDocData

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

class AppWebService:

    """ This is AppDatabase class """

    def __init__(self, url=None):

        if not url:

            configs = AppDocData.instance().getConfigs('Engine', 'Address')

            self._url = configs[0].value if configs else None

        else:

            self._url = url

        #self._url = 'http://192.168.0.21:80'

    def test_connection(self):

        try:

            if not self._url:

                return False

            url = self._url + 'Test_Connection'

            response = requests.get(url)

            if response.status_code == 200:

                return True

            else:

                mb = QMessageBox()

                mb.setIcon(QMessageBox.Critical)

                mb.setWindowTitle('Error')

                mb.setText('status code({}) : {}'.format(str(response.status_code), response.text))

                mb.exec_()

                return False

        except Exception as ex:#requests.exceptions.RequestException as ex:

            mb = QMessageBox()

            mb.setIcon(QMessageBox.Critical)

            mb.setWindowTitle('Error')

            mb.setText('{}'.format(ex))

            mb.exec_()

            return False

    def request_symbol_box(self, img):

        # send uncroped image

        try:

            if not self.test_connection():

                return []

            symbol_box = '/recognition/symbol_box'

            _, bts = cv2.imencode('.png', img)

            bts = bts.tostring()

            response = requests.post(self._url + symbol_box, data=bts)

            return response.json()['symbol_box'] #

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                           sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

            return []

    def request_text_box(self, img):

        # send uncroped image, test code

        try:

            if not self.test_connection():

                return []

            text_box = '/recognition/text_box'

            _, bts = cv2.imencode('.png', img)

            bts = bts.tostring()

            response = requests.post(self._url + text_box, data=bts)

            return response.json()['text_box']

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                           sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

            return []

    def request_text_box_tile(self, img_infos):

        """ main code """

        try:

            if not self.test_connection():

                return []

            text_box = '/recognition/stream_text_box'

            imgs = [info[4] for info in img_infos]

            str_imgs = []

            for img in imgs:

                _, bts = cv2.imencode('.png', img)

                # bts = bts.tostring()

                bts = base64.b64encode(bts).decode()

                str_imgs.append(bts)

            response = requests.post(self._url + text_box, data=json.dumps(str_imgs))

            box_list = response.json()['text_box_list']

            # for debug

            if len(img_infos) != len(box_list):

                print('check return values')

                return

            for index in range(len(img_infos)):

                img_infos[index].append(box_list[index])

            return img_infos

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                           sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

            return []

    '''

    def request(self, url):

    response = urllib.request.urlopen(url)

    byte_data = response.read()

    text_data = byte_data.decode('utf-8')

    return text_data

    '''