Smart-Plug/SmartPlug240usec.py at master · herborspice/Smart-Plug · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

import sys,os
import time
import math
#from scipy.interpolate import interp1d
#import numpy as np
import sqlite3
#import matplotlib.pyplot as plt
#from scipy import integrate

#from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
#from matplotlib.backends.backend_qt4agg import NavigationToolbar2QT as NavigationToolbar
#from matplotlib.figure import Figure

#import random
import serial
from tkinter import *
import csv


start= time.time()
conn = sqlite3.connect(':memory:')
c = conn.cursor()
c.execute("""CREATE TABLE plugData(current real, voltage real, time real)""")
c.execute("SELECT * FROM plugData WHERE time != 0") #for degugging prints all from database
#print(c.fetchall())
conn.commit()

ser = serial.Serial('/dev/ttyACM1')  # open serial port
ser.baudrate = 115200
ser.flush()
#print(ser.name)         # check which port was really used

#c = csv.writer(open("Toster_Data.csv", "wb") )
#c.writerow( ["Time","Current", "Voltage"])

window = Tk()
window.title("Smart Plug")
window.geometry('350x200')

#limit = 10
#period = 1000


def relay_open():
    ser.write(b'0')
    #print('open?')

def relay_close():
    ser.write(b'1')
    #print('close?')

def update_data():
 i = 0
 j = 0

 points = 139
 rows_csv = points + 1

 data = list()
 #data.append(['Time', 'Current', 'Voltage'])

 while i <= points:  #change number to be as large as needed

    if ser.inWaiting()> 0:

        inputString = ser.readline().decode('ascii')

        print(inputString)

        Curr = str(inputString.split()[0])
        Volt = str(inputString.split()[1])


        data.append([Curr, Volt])

        #print(data)


        #csv_writer.writerow([ timeBoi, currBoi, voltBoi])
        #print('data in')

        #print(tme)
        #print(current)
        #print(voltage)

        i = i+1

        #time.sleep(.5) #slows down to try to match the data coming from the arduino

    #else:
        #print('No data from Arduino')


 with open('240Vacuum.csv', 'wt') as f:
        csv_writer = csv.writer(f)

        while j <  rows_csv:  #change number to be one more that i
            csv_writer.writerow(data[j])
            print(data[j])
            j=j+1


relay1 = Button(window, text = "Relay Open", command = relay_open)
relay1.grid(column=0, row = 0)

relay2 = Button(window, text = "Relay Close", command = relay_close)
relay2.grid(column=1, row =0)

dataCollect = Button(window, text = "Collect Data", command = update_data)
dataCollect.grid(column=2, row = 0)


window.mainloop()