pgdrive.py

import psycopg2
from uuid import getnode as get_mac
import tkinter as tk
import time
ACTION_di = 17
ADD_CNT_di = 5
DEC_CNT_di = 27
RESET_CNT_di = 18
INC_OP_CNT_di = 12
LBT = 0
station_name = ""
LBD = 0
mac = get_mac()
pittsteel = psycopg2.connect("dbname=PittSteel host=192.168.2.3 user=caleb password=tori")

cur = pittsteel.cursor()
station_id = -1
 
def setStationId(station,callmain):
    global station_id
    station_id = station
    if station_id == -1:
        cur.execute("""SELECT station_id,station_name FROM psproductivity.station WHERE station_mac = %s;""",(mac,))
        rows = cur.fetchall()
        station_id = rows[0][0]
        
    callmain()

def register(callmain):
    global mac
    global cur
    
    cur.execute("""SELECT station_id,station_name FROM psproductivity.station WHERE station_mac = %s;""",(mac,))
    rows = cur.fetchall()
    def enterName(rwin,name,time,dist):
        global pittsteel
        global LBT
        global LBD
        cur.execute("""INSERT INTO psproductivity.station (station_name,station_mac,station_lbt,station_lbd,action,addcnt,deccnt,resetcnt,incop) VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s)""",(name,mac,time,dist,17,5,27,18,12))
        pittsteel.commit()
        cur.execute("""SELECT station_lbt,station_lbd FROM psproductivity.station WHERE station_mac = %s;""",(mac,))
        rows2 = cur.fetchall()
        LBT = rows2[0][0]
        LBD = rows2[0][1]
        nonlocal callmain
        rwin.destroy()
        setStationId(-1,callmain)
        

    if(len(rows) == 0):
        question = tk.Tk()
        question.title("Configure")
        tk.Label(question, text = 'Enter Station Name').pack()
        a = tk.Entry(question)
        a.pack()
        tk.Label(question, text = 'Enter Look Back Time').pack()
        b = tk.Entry(question)
        b.pack()
        tk.Label(question, text = 'Enter Look Back Operations').pack()
        c = tk.Entry(question)
        c.pack()
        tk.Button(question, text='Submit', command = lambda:enterName(question,a.get(),b.get(),c.get())).pack()
        question.mainloop()
    else :
        global station_name
        station_name = rows[0][1]
        cur.execute("""SELECT station_lbt,station_lbd,action,addcnt,deccnt,resetcnt,incop FROM psproductivity.station WHERE station_mac = %s;""",(mac,))
        global LBT
        global LBD
        global ACTION_di
        global ADD_CNT_di
        global DEC_CNT_di
        global RESET_CNT_di
        global INC_OP_CNT_di
        
        rows2 = cur.fetchall()
        ACTION_di = rows2[0][2]
        ADD_CNT_di = rows2[0][3]
        DEC_CNT_di = rows2[0][4]
        RESET_CNT_di = rows2[0][5]
        INC_OP_CNT_di = rows2[0][6]
        LBT = rows2[0][0]
        LBD = rows2[0][1]
        setStationId(rows[0][0],callmain)


def insertActivity(actType, actTime):
    global station_id
    global pittsteel

    cur.execute("""INSERT INTO psproductivity.activity (activity_type,activity_time,activity_station_id) VALUES(%s,to_timestamp(%s), %s)""",(actType, actTime ,station_id))
    pittsteel.commit()

def insertprodtakt(takt, takttime):
    global station_id
    global pittsteel
    
    cur.execute("""INSERT INTO psproductivity.prodtakt (prodtakt_start,prodtakt_takt,prodtakt_station_id) VALUES(to_timestamp(%s), %s, %s)""",(takttime,takt,station_id))
    pittsteel.commit()

def getSched(treeview):
    global station_id
    cur.execute("""
    SELECT pswosched_id, wo_number, item_descrip1, pswosched_submit || '/' || pswosched_request
    FROM psproductivity.pswosched
    JOIN wo ON(wo_id = pswosched_woitem_id)
    JOIN itemsite ON (wo_itemsite_id = itemsite_id)
    JOIN item ON (itemsite_item_id = item_id)
    WHERE pswosched_station_id = %s
    ORDER BY pswosched_priority
    """,(station_id,))

    for row in cur:
        treeview.insert('', 'end', iid=row[0], values=(row[1], row[2], row[3]))

def updateWork(woitem_id, qty):
    global station_id
    

def launchConfig(lastRecord):
    cur.execute("""
    SELECT activity_type, activity_time FROM psproductivity.activity
    WHERE activity_station_id = %s ORDER BY activity_time DESC LIMIT 1""",(station_id,))
    rrunning = False
    rStartTime = 0

    for row in cur:
        if(row[0] == 'Start'):
            ttime  = lastRecord()
            if(time.time() - ttime < LBT * 60): # It hasnt been too long just pick up the last running time and set it to running 
                rrunning =True
                rStartTime = row[1]
            else:
                insertActivity('Stop',ttime) # Otherwise give us a nice stop for the last time it was runnning
        break

    return rrunning, rStartTime