initialsetup.py

#!/usr/bin/env python

import pygtk
pygtk.require('2.0')
import gtk
import json
import requests
import twisted
import sys
import os
import collections

url = "http://aquarimeter.rocks/api/v1/"
class Aquarium_Setup:

    #checks fields are filled in on form
    def check_register(self,widget):
        fill = self.validate_fields()
        long_pwd = self.check_length(self.valid_Password.get_text())
        temp_valid = self.check_temp()

        if(not fill):
            self.error_fill()
        elif(not long_pwd):
            self.error_password()
        elif(not temp_valid):
            self.error_temp_range()
        else:
            self.register()
            
    #checks required fields is filled before registering
    def validate_fields(self):
        if(not self.valid_Email.get_text() or
           not self.valid_Password.get_text() or
           not self.first_Name.get_text() or
           not self.last_Name.get_text()):
            return False
        else:
            return True

    #checks password is long enough
    def check_length(self,password):
        if(len(password) < 10):
            return False
        else:
            return True

    #checks temperature values are valid 
    def check_temp(self):
        idealT = self.ideal_spin.get_value()
        minT = self.min_spin.get_value()
        maxT = self.max_spin.get_value()
        if(minT <= idealT and idealT<= maxT):
            return True
        else:
            return False
        
    #pop up message box asking to fill all fields
    def error_fill(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_WARNING, buttons=gtk.BUTTONS_OK)
        info.set_property('title', 'Missing information')
        info.set_property('text', 'Please fill in all fields')
        info.run()
        info.destroy()

    #pop up message for short password
    def error_password(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_WARNING, buttons=gtk.BUTTONS_OK)
        info.set_property('title', 'Password too short')
        info.set_property('text', 'Password must be at least 10 characters long')
        info.run()
        info.destroy()

    #pop up message for bad temperature set
    def error_temp_range(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_WARNING, buttons=gtk.BUTTONS_OK)
        info.set_property('title', 'Wrong temperature range')
        info.set_property('text', 'Ideal temperature must be between Min and Max temperature.\n'
                          'Min temperature must be less than or equal to Max temperature')
        info.run()
        info.destroy()

    #registers with entered information from gui
    def register(self):
        #json payload send happens here
        #storing email and password so no need to get again
        email = self.valid_Email.get_text()
        password = self.valid_Password.get_text()
        firstnm = self.first_Name.get_text()
        lastnm = self.last_Name.get_text()
        register_data = {"user":{"email":email,
                                 "password":password,
                                 "first_name":self.first_Name.get_text(),
                                 "last_name":self.last_Name.get_text()}}

        reg_data = json.dumps(register_data)
        url_reg = "http://aquarimeter.rocks:5000/api/v1/register"
        
        reg = requests.post(url_reg,reg_data)
        
        if reg.status_code == 200:
            self.login(email,password)
        elif reg.status_code == 422:
            self.error_register_422()
        else:
            self.error_register()
            
    #pop up message for registeration error code == 422
    def error_register_422(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_ERROR, buttons=gtk.BUTTONS_CLOSE)
        info.set_property('title', 'Error 422')
        info.set_property('text', 'Something went wrong with validation')
        info.run()
        info.destroy()

    #pop up message for any other status code except 200 and 422
    def error_register(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_ERROR, buttons=gtk.BUTTONS_CLOSE)
        info.set_property('title', 'Error with validation')
        info.set_property('text', 'Site is suffering internel '
                          'conflict or url is no longer valid')
        info.run()
        info.destroy()

    #login with password and email
    def login(self,email,password):
        #login  happens here
        login_data = {"user":{"email":email,
                                 "password":password}}
        
        log_data = json.dumps(login_data)
        url_log = "http://aquarimeter.rocks:5000/api/v1/login"
        log = requests.post(url_log, data = url_log)
        
        if log.status_code == 200:
            #auth_token, ideal_temp, min_temp and max_temp will be passed
            #to aquarimete program
            auth_token = log[auth_token]
            idealT = self.ideal_spin.get_value()
            minT = self.min_spin.get_value()
            maxT = self.max_spin.get_value()
            comand = "sudo aquarimeter.py" + auth_token + idealT + minT + maxT
            os.system(comand)
            self.successful_login()
            
        elif log.status_code == 401:
            self.error_login_401()
        else:
            self.error_login()
            
    #successful login starts aquarimeter and closes initial setup
    def successful_login(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_INFO, buttons=gtk.BUTTONS_CLOSE)
        info.set_property('title', 'Error with login')
        info.set_property('text', 'Unknown error has occured url might be bad')
        info.run()
        info.destroy()
            
    #login is invalid. status code 401
    def error_login_401(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_ERROR, buttons=gtk.BUTTONS_CLOSE)
        info.set_property('title', 'Error 401')
        info.set_property('text', 'username/password is incorrect. ')
        info.run()
        info.destroy()

    #any other status code covered here
    def error_login(self):
        info = gtk.MessageDialog(type=gtk.MESSAGE_ERROR, buttons=gtk.BUTTONS_CLOSE)
        info.set_property('title', 'Error with login')
        info.set_property('text', 'Unknown error has occured url might be bad')
        info.run()
        info.destroy()
       
    #main gui window
    def __init__(self):
        self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.window.set_title("Aquarimeter Intial Setup")
        self.window.connect("destroy", lambda wid: gtk.main_quit())
        
        main_Box = gtk.VBox(False, 5) #all items will be arranged vertically in here
        self.window.add(main_Box)
        self.window.set_border_width(10)

        label = gtk.Label("Please enter all intial values as indicated. Your"
                          "settings will be saved in\n Aquarimeter Web, the"
                          " web application which helps you view the status "
                          "of your\n aquarium. It can be at http://aquarium."
                          "oconnor.ninja/")
        main_Box.pack_start(label)
        #create fields for registeration information to be entered
        reg_frame = gtk.Frame("Login Details for Aquarimeter Web") 

        #labels for information needed
        info_Box = gtk.HBox(False,5)
        vbox1 = gtk.VBox(False,5)
        info_Box.add(vbox1)

        label = gtk.Label("E-mail")
        vbox1.add(label)
        label = gtk.Label("Password")
        vbox1.add(label)
        label = gtk.Label("First Name")
        vbox1.add(label)
        label = gtk.Label("Last Name")
        vbox1.add(label)

        #fields to enter information
        vbox2 = gtk.VBox(False,5)
        info_Box.add(vbox2)
        
        self.valid_Email = gtk.Entry()
        vbox2.add(self.valid_Email)
        
        self.valid_Password = gtk.Entry()
        vbox2.add(self.valid_Password)
        
        self.first_Name = gtk.Entry()
        vbox2.add(self.first_Name)
        
        self.last_Name = gtk.Entry()
        vbox2.add(self.last_Name)
        
        reg_frame.add(info_Box)
        main_Box.pack_start(reg_frame, False, False, 0)

        #labels for varias temperature settings
        temp_frame = gtk.Frame("Temperature set")
        vbox3 = gtk.VBox(False,5)
        hbox3 = gtk.HBox(False,5)
        hbox3.add(vbox3)
        
        label = gtk.Label("Intial Temperature")
        vbox3.add(label)
        label = gtk.Label("Min Temperature")
        vbox3.add(label)
        label = gtk.Label("Max Temperature")
        vbox3.add(label)

        #temperature selection setting
        vbox4 = gtk.VBox(False,5)
        hbox3.add(vbox4)

        adj1 = gtk.Adjustment(70.0, 55.0, 85.0, 1.0, 5.0, 0.0)
        self.ideal_spin = gtk.SpinButton(adj1, 0, 0)
        vbox4.add(self.ideal_spin)

        adj2 = gtk.Adjustment(70.0, 55.0, 85.0, 1.0, 5.0, 0.0)
        self.min_spin = gtk.SpinButton(adj2, 0, 0)
        vbox4.add(self.min_spin)
        
        adj3 = gtk.Adjustment(70.0, 55.0, 85.0, 1.0, 5.0, 0.0)
        self.max_spin = gtk.SpinButton(adj3, 0, 0)
        vbox4.add(self.max_spin)

        #temperature units setting
        vbox5 = gtk.VBox(False,0)
        hbox3.add(vbox5)

        label = gtk.Label("Fahrenheit")
        vbox5.add(label)
        
        temp_frame.add(hbox3)
        main_Box.pack_start(temp_frame, False, False, 0)

        #cancel and submit button section
        hbox10 = gtk.HBox(False,2)
        main_Box.pack_start(hbox10, False,False,0)
        
        button = gtk.Button("Submit and Register")
        button.connect("clicked", self.check_register)
        hbox10.pack_end(button,False,False,2)
        
        button = gtk.Button("cancel")
        button.connect_object("clicked",gtk.Widget.destroy, self.window)
        hbox10.pack_end(button,False,False,2)

        #show everything in a window
        self.window.show_all()

    def main(self):
        gtk.main()


if __name__ == "__main__":
    Aqua_Setup = Aquarium_Setup()
    Aqua_Setup.main()