main.py

# coding=utf8

import sys

from PyQt5 import uic, QtWidgets, QtCore
from PyQt5.QtCore import QSize
from PyQt5.QtGui import QIcon, QPixmap
from PyQt5.QtWidgets import *
from PyQt5.QtWidgets import QMessageBox
from PyQt5 import QtGui
import urllib.request
import psycopg2 as pg2
from src.getbarcode import read_barcode

mainLayout = uic.loadUiType("ui/mainWindow.ui")[0]
signinLayout = uic.loadUiType("ui/signIn.ui")[0]
initialLayout = uic.loadUiType("ui/initialLayout.ui")[0]

if sys.platform.lower() == 'darwin':
    conn = pg2.connect(host="localhost", database="projectDB", user="postgres", password="0000", port="5433")
elif sys.platform == 'win32':
    conn = pg2.connect(host="localhost", database="projectDB", user="postgres", password="1234", port="5432")
else:
    conn = pg2.connect(host="localhost", database="projectDB", user="postgres", password="1234", port="5432")

cur = conn.cursor()

global session # login session
global searcharr
searcharr=[]

global sessionInfo
sessionInfo=[] # allergy, vegiterrian

class initWindow(QMainWindow, initialLayout):
    def __init__(self) :
        super(initWindow, self).__init__(parent=None)
        self.setupUi(self)

        self.setWindowTitle('Initial Window')

        self.show()

        self.inputPS_text.setEchoMode(QtWidgets.QLineEdit.Password)
        self.signinBtn.clicked.connect(self.opensigninFunction)
        self.loginBtn.clicked.connect(self.loginFunction)

    def opensigninFunction(self):
        signinWindow(self)

    def loginmessegeFuntion(self):
        msg = QMessageBox()
        msg.setText("반갑습니다!! ")
        msg.exec_()

    def loginFunction(self):
        global session, sessionInfo
        id = self.inputID_text.text()
        pw = self.inputPS_text.text()

        if id != "" and pw!="": # null check
            # check login info is right
            loginQ = "SELECT * FROM usertable where userid= '"+id+"' AND password= '" +pw +"';"
            cur.execute(loginQ)
            userResult = cur.fetchall()

            print(userResult)

            if(len(userResult)== 0):
                msg = QMessageBox()
                msg.setIcon(QMessageBox.Critical)
                msg.setText("아이디 비밀번호를 확인해주세요!!")
                msg.setWindowTitle("Error")
                msg.exec_()

                self.inputID_text.setText("")
                self.inputPS_text.setText("")
            else:
                self.loginmessegeFuntion()
                session=id
                userall= []
                newstr = userResult[0][4].replace("{", "").replace("}", "")

                userall = newstr.split(',')

                sessionInfo=[userall, userResult[0][5]]
                print(sessionInfo)
                mainWindow(self)
                self.inputID_text.setText("")
                self.inputPS_text.setText("")

        else:
            msg = QMessageBox()
            msg.setIcon(QMessageBox.Critical)
            msg.setText("아이디 비밀번호를 입력해주세요!! ")
            msg.setWindowTitle("Error")
            msg.exec_()
        # if it's right open main window


class mainWindow(QMainWindow, mainLayout):
    global session, listview
    global searchArr
    global itemAllergy, itemRawmtrl

    def  __init__(self, parent=None):
        super(mainWindow, self).__init__(parent)
        self.setupUi(self)
        self.setWindowTitle('Main Winow')

        self.show()
        self.pushButton_2.clicked.connect(self.read_barcode)

        self.showID.setText(session)
        self.logoutBtn.clicked.connect(self.logoutFunction)

        self.searchBtn.clicked.connect(self.searchFunction)
        self.resultList.itemClicked.connect(self.showResultFunction)

        self.searchBBtn.clicked.connect(self.searchBarcodeFunction)

        # self.alterAlist.currentIndexChanged.connect()
        # self.alterVlist.currentIndexChanged.connect()


    def showResultFunction(self):
        global searcharr, listview
        global itemAllergy, itemRawmtrl

        if listview.currentItem().text()== "검색 결과가 없습니다. ":
            return
        else:
            index = listview.selectionModel().currentIndex().row()

            url = searcharr[index][3]
            data = urllib.request.urlopen(url).read()
            pixmap = QPixmap()
            pixmap.loadFromData(data)

            pixmap.scaled(self.productImg.width(), self.productImg.height(), QtCore.Qt.KeepAspectRatio)

            self.productImg.setPixmap(pixmap) # url 제품 사진
            self.productImg.setScaledContents(True)
            self.productImg.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)

            self.productName.setText("제품명: "+searcharr[index][0])
            self.productAllergy.setText("알러지 유발 물질:" + searcharr[index][2])
            self.productNutrient.setText("영양정보: " + searcharr[index][4])
            self.productmtrl.setText("원재료: "+searcharr[index][1])

            itemAllergy = searcharr[index][2]
            itemRawmtrl = searcharr[index][1]

            self.showpersonalResult(index)

    def showpersonalResult(self, index):
        global itemAllergy, itemRawmtrl, searcharr

        # veg section
        # 0. 사용자가 채식주의자가 아닌 경우  > skip
        # 1. 사용자가 먹을 수 있는 경우 > yes.png
        # 2. 사용자가 먹을 수 없는 경우 > no.png > 대체 식품 (검색 결과 list > select)
        if(sessionInfo[1]== "nan"): # non veg
            self.resultVText.setText(session+ "님은 해당 사항이 없습니다. ")  # 베지터리언 restrict show
        else:
            # check veg type and search mtrls
            # 선택한 제품의 채식주의 True, False 받아오기.
            psearchQ = "SELECT rawmtrl FROM vegproduct where prdlstReportNo = '" + str(searcharr[index][5]) + "' and  " + sessionInfo[1] + " = 'True' ;"
            cur.execute(psearchQ)
            personalResult = cur.fetchall()
            print("veg result: " + str(personalResult))

            if(len(personalResult)==0):
                qPixmapVar = QPixmap()
                qPixmapVar.load("ui/yes.png")
                qPixmapVar = qPixmapVar.scaled(81, 71)
                self.resultVImg.setPixmap(qPixmapVar)
                self.resultVText.setText(sessionInfo[1] + "은 먹을 수 있는 식품입니다! ")  # 베지터리언 restrict show
                self.alterVlist.clear()
            else:
                qPixmapVar = QPixmap()
                qPixmapVar.load("ui/no.png")
                qPixmapVar = qPixmapVar.scaled(81, 71)
                self.resultVImg.setPixmap(qPixmapVar)
                resStr=""
                for i in range(0, len(personalResult)):
                    resStr += str(personalResult[i]).replace("(", '').replace(')', '').replace(',', '') +" "

                self.resultVText.setText(sessionInfo[1]+"은 먹을 수 없는 "+ resStr+ "가(이) 들어있어요! ") # 베지터리언restrict show

                self.alterVlist.clear()
                # 대체 식품 검색 후 추가
                for i in range(0, len(searcharr)):
                    asearchQ = "SELECT * FROM vegproduct where prdlstReportNo = '" + str(searcharr[i][5]) + "' and "+sessionInfo[1]+" = 'True' ;"
                    cur.execute(asearchQ)
                    alterlist = cur.fetchall()

                    if (len(alterlist) == 0):
                        self.alterVlist.addItem(searcharr[i][0]);



        # allergy section
        # 0. 사용자가 해당하는 알러지 식품이 없는 경우 > skip
        # 1. 알러지 유발 물질 x, 교차 반응 물질 x > yes.png
        # 2. 알러지 유발 물질 x, 교차 반응 물질 o > no.png > 대체 식품 (검색 결과 list > select)
        # 3. 알러지 유발 물질 o > no.png > 대체 식품 추가 (검색 결과 list > select)

        if(len(sessionInfo[0]) == 0 ):
            self.resultAText.setText(session + "님은 해당 사항이 없습니다. ")  # 알러지 해당 사항이 없을 경우
        else:
            allarray = []

            # 선택한 제품의 알러지 일치 유무 받아오기.
            for a in range (0, len(sessionInfo[0])):
                psearchQ = "SELECT * FROM allergyproduct where prdlstReportNo = '" + str(searcharr[index][5]) + "' and allergy = '" + sessionInfo[0][a]+ "';"
                cur.execute(psearchQ)
                personalResult = cur.fetchall()

                # cross-react check
                for z in personalResult:
                    reultArr = str(z).split(',')
                    if(reultArr[3]==' True'):
                        allarray.append([sessionInfo[0][a], True, reultArr[1]])
                    elif(reultArr[3]==' False'):
                        allarray.append([sessionInfo[0][a], False, None])
            print("allarray: "+ str(allarray))

            # 결과 show
            if len(allarray) == 0 :
                qPixmapVar = QPixmap()
                qPixmapVar.load("ui/yes.png")
                qPixmapVar = qPixmapVar.scaled(81, 71)
                self.resultAImg.setPixmap(qPixmapVar)
                self.resultAText.setText(session +"님이 해당되는 알러지 유발 물질이 들어있지 않습니다!! ")
                self.alterAlist.clear()
            else:
                qPixmapVar = QPixmap()
                qPixmapVar.load("ui/no.png")
                qPixmapVar = qPixmapVar.scaled(81, 71)
                self.resultAImg.setPixmap(qPixmapVar)

                # 반복문 - 출력
                resultStr =""
                for k in allarray:
                    if k[1] == False:
                        resultStr += "(알러지 유발 물질)" + k[0] + " "
                    elif k[1] == True:
                        resultStr += "( " + k[0]+" 과의 교차반응 물질)" + k[2] +" "

                self.resultAText.setText(session + "님이 해당되는 "+ resultStr + "가(이) 들어습니다! ")

                self.alterAlist.clear()
                # 대체 식품 검색 후 추가
                for listdata in searcharr:
                    for allPer in sessionInfo[0]:
                        asearchQ = "SELECT * FROM allergyproduct where prdlstReportNo = '" + str(listdata[5]) + "' and allergy = '" + allPer + "';"
                        cur.execute(asearchQ)
                        alterlist = cur.fetchall()

                        print("alter"+str(alterlist))
                        if(len(alterlist)==0):
                            self.alterAlist.addItem(listdata[0])

    def searchBarcodeFunction(self):
        global listview, searcharr

        searchtext = self.searchBTxt.text()

        searchQ = "SELECT prdlstname, rawmtrl, allergy, imgurl1, nutrient, prdlstreportno FROM foodinfo where barcode like '%" + searchtext + "%';"
        cur.execute(searchQ)
        searchResult = cur.fetchall()

        listview = self.resultList
        listview.clear()
        listview.setIconSize(QSize(60, 60))
        listview.setVerticalScrollMode(QAbstractItemView.ScrollPerPixel);

        if len(searchResult) == 0:
            listview.addItem('검색 결과가 없습니다. ')
        else:
            searcharr.clear()
            for i in range(0, len(searchResult)):
                searcharr.append(searchResult[i])

                listview.addItem(searchResult[i][0])

                url = searchResult[i][3]
                data = urllib.request.urlopen(url).read()
                pixmap = QPixmap()
                pixmap.loadFromData(data)
                icon = QIcon(pixmap)
                listview.item(i).setIcon(icon)

    def searchFunction(self):
        global listview, searcharr

        searchtext=self.searchTxt.text()

        searchQ = "SELECT prdlstname, rawmtrl, allergy, imgurl1, nutrient, prdlstreportno  FROM foodinfo where prdlstname like '%" + searchtext + "%';"
        cur.execute(searchQ)
        searchResult = cur.fetchall()

        listview = self.resultList
        listview.clear()
        listview.setIconSize(QSize(60, 60))
        listview.setVerticalScrollMode(QAbstractItemView.ScrollPerPixel);

        if len(searchResult)== 0:
            listview.addItem('검색 결과가 없습니다. ')
        else:
            searcharr.clear()
            for i in range (0, len(searchResult)):
                searcharr.append(searchResult[i])

                listview.addItem(searchResult[i][0])

                url = searchResult[i][3]
                data = urllib.request.urlopen(url).read()
                pixmap = QPixmap()
                pixmap.loadFromData(data)
                icon = QIcon(pixmap)
                listview.item(i).setIcon(icon)

    def read_barcode(self):
        barcodes = read_barcode()
        self.searchBTxt.setText('{}'.format(barcodes))

    def logoutFunction(self):
        global session
        session = ''
        sessionInfo.clear()
        self.close()

class signinWindow(QMainWindow, signinLayout):
    global ischecked
    ischecked = False # check overlap id

    def __init__(self, parent=None):
        super(signinWindow, self).__init__(parent)
        self.setupUi(self)
        self.setWindowTitle('Sign In')

        self.show()

        self.setPass_text.setEchoMode(QtWidgets.QLineEdit.Password)
        self.checkOverlapBtn.clicked.connect(self.checkidFunction)
        self.signinBtnBox.accepted.connect(self.signinFunction)
        self.signinBtnBox.rejected.connect(self.closeFunction)

    def closeFunction(self):
        self.close()

    def signinFunction(self):
        global isfilled
        # check fully filled
        isfilled = False

        id = self.setID_text.text()
        pw = self.setPass_text.text()

        age = self.getAge_box.value()
        veg = self.vegCombo.currentText()

        if id and pw and age!=0:
            isfilled = True
        else:
            isfilled = False

        # get veg
        # 'vegan', 'lactoVeg', 'ovoVeg', 'lactoOvoVeg', 'pescoVeg', 'polloVeg', 'flex'
        if veg == '해당없음':
            veg='nan'
        elif veg == '비건':
            veg='vegan'
        elif veg == "락토 베지테리언":
            veg='lactoVeg'
        elif veg == "오보 베지테리언":
            veg='ovoVeg'
        elif veg == "락토 오보 베지테리언":
            veg='lactoOvoVeg'
        elif veg == "페스코 베지테리언":
            veg='pescoVeg'
        elif veg == "폴로 베지테리언":
            veg='polloVeg'

        # get gender
        if self.femaleBtn.isChecked():
            gender = 'f'
            isfilled = True
        elif self.maleBtn.isChecked():
            gender = 'm'
            isfilled = True
        else:
            self.fillerrormeassage()
            isfilled = False

        # get allergey
        allegy=[]
        if self.none_check.isChecked():
            print('no allergy')
        else:
            if self.bean_check.isChecked():
                allegy.append(self.bean_check.text())
            if self.beef_check.isChecked():
                allegy.append(self.beef_check.text())
            if self.buckwheat_check.isChecked():
                allegy.append(self.buckwheat_check.text())
            if self.egg_check.isChecked():
                allegy.append(self.egg_check.text())
            if self.milk_check.isChecked():
                allegy.append(self.milk_check.text())
            if self.peanut_check.isChecked():
                allegy.append(self.peanut_check.text())
            if self.wheat_check.isChecked():
                allegy.append(self.wheat_check.text())
            if self.crab_check.isChecked():
                allegy.append(self.crab_check.text())
            if self.mackerel_check.isChecked():
                allegy.append(self.mackerel_check.text())
            if self.peach_check.isChecked():
                allegy.append(self.peach_check.text())
            if self.pork_check.isChecked():
                allegy.append(self.pork_check.text())
            if self.shrimp_check.isChecked():
                allegy.append(self.shrimp_check.text())
            if self.squid_check.isChecked():
                allegy.append(self.squid_check.text())
            if self.tomato_check.isChecked():
                allegy.append(self.tomato_check.text())
            if self.acid_check.isChecked():
                allegy.append(self.acid_check.text())
            if self.calm_check.isChecked():
                allegy.append(self.calm_check.text())
            if self.chicken_check.isChecked():
                allegy.append(self.chicken_check.text())
            if self.kiwi_check.isChecked():
                allegy.append(self.kiwi_check.text())
            if self.pinenut_check.isChecked():
                allegy.append(self.pinenut_check.text())
            if self.sesame_check.isChecked():
                allegy.append(self.sesame_check.text())
            if self.walnut_check.isChecked():
                allegy.append(self.walnut_check.text())

        # parse allegy to string(enum type)
        allstr = str(allegy)
        allstr = allstr.replace("[", "{")
        allstr = allstr.replace("]", "}")
        allstr = allstr.replace("'", "")

        if isfilled and ischecked :
            adduserQ = "INSERT INTO usertable values('" + id + "', '" + pw + "', '" + gender + "', " + str(age) + ", '" + allstr + "', '" + veg + "')"
            print(adduserQ)
            cur.execute(adduserQ)
            conn.commit()
            self.signinmeassage()
            self.close()
        else:
            self.fillerrormeassage()

    def signinmeassage(self):
        msg = QMessageBox()
        msg.setText("회원 가입이 완료되었습니다!")
        msg.exec_()

    def fillerrormeassage(self):
        msg = QMessageBox()
        msg.setIcon(QMessageBox.Critical)
        msg.setText("양식을 전부 작성해 주세요!!")
        msg.exec_()

    def checkidFunction(self):
        global ischecked
        getid = self.setID_text.text()
        # check login info is right

        loginQ = "SELECT * FROM usertable where userid= '" + getid + "';"
        cur.execute(loginQ)
        userResult = cur.fetchall()

        if (len(userResult) == 0):
            msg = QMessageBox()
            msg.setText("사용 가능한 아이디 입니다! ")
            msg.exec_()
            ischecked = True

        else:
            msg = QMessageBox()
            msg.setIcon(QMessageBox.Critical)
            msg.setText("중복된 아이디 입니다! ")
            msg.setWindowTitle("Error")
            msg.exec_()

            self.setID_text.setText("")


class initDB():
    createUserQ="CREATE TABLE IF NOT EXISTS UserTable (userID TEXT, password TEXT, gender gen, age INT, allergies allergy[], vName veg, primary key(userID));"

    cur.execute(createUserQ, createUserQ)
    conn.commit()

if __name__ == "__main__" :
    app = QApplication(sys.argv)
    initDB()
    myWindow = initWindow()
    myWindow.show()
    app.exec_()