lab x

Labs/lab-SQL/Lab-SQLite/app/forms.py

@@ -5,5 +5,4 @@
 
 class CustomerForm(Form):
     company = StringField('company', validators=[DataRequired()])
-    email = EmailField('email', validators=[DataRequired()])
-
+    email = EmailField('email', validators=[DataRequired()])

Labs/lab-SQL/Lab-SQLite/app/models.py

@@ -1,7 +1,16 @@
 import sqlite3 as sql
 
-def insert_customer():
+def insert_customer(company,email):
     # SQL statement to insert into database goes here
+    with sql.connect('app.db') as con:
+    	cur = con.cursor()
+    	cur.execute('INSERT INTO customers (company,email) VALUES (?,?)',(company,email))
+    	con.commit()
 
 def retrieve_customers():
     # SQL statement to query database goes here
+    with sql.connect('app.db') as con:
+    	con.row_factory = sql.Row
+    	cur = con.cursor()
+    	result = cur.execute('SELECT * FROM customers').fetchall()
+    return result

Labs/lab-SQL/Lab-SQLite/app/templates/customer.html

@@ -11,6 +11,14 @@ <h2>Add Customer to Our Database</h2>
       <div class="divider"></div>
       <div class="section">
         <!--add input fields-->
+        <p>
+          Company name:<br>
+          {{form.company(size=120)}}<br>
+        </p>
+        <p>
+          Customer email:<br>
+          {{form.email(size=120)}}<br>
+        </p>
       </div>
       <div class="divider"></div>
       <div class="section">
@@ -20,4 +28,4 @@ <h2>Add Customer to Our Database</h2>
     </form>
   </div>
 </div>
-{% endblock %}
+{% endblock %}

Labs/lab-SQL/Lab-SQLite/app/templates/home.html

@@ -17,8 +17,14 @@ <h3>These are all of our awesome customers:</h3>
             </tr>
           </thead>
 <!-- code to display all user -->
+            {% for customer in customers %}
+            <tr>
+              <td> {{customer['company']}} </td>
+              <td> {{customer['email']}} </td>
+            </tr>
+            {% endfor %}
         </table>
       </div>
     </div>
   </div>
-{% endblock %}
+{% endblock %}

Labs/lab-SQL/Lab-SQLite/app/views.py

@@ -1,6 +1,7 @@
 from flask import render_template, redirect, request
 from app import app, models, db
 from .forms import CustomerForm
+#from models import *
 # Access the models file to use SQL functions
 
 
@@ -14,11 +15,15 @@ def create_customer():
     if form.validate_on_submit():
         # Get data from the form
         # Send data from form to Database
+        company = form.company.data
+        email = form.email.data
+        models.insert_customer(company,email)
         return redirect('/customers')
     return render_template('customer.html', form=form)
 
 @app.route('/customers')
 def display_customer():
     #Retreive data from database to display
+    customers = models.retrieve_customers()
     return render_template('home.html',
                             customers=customers)

Labs/lab-SQL/Lab-SQLite/schema.sql

@@ -1,2 +1,9 @@
 -- Insert code to create Database Schema
 -- This will create your .db database file for use
+
+drop table if exists customers;
+create table customers (
+	customer_id integer primary key,
+	company text not null,
+	email text non null
+);