start.py

#!/usr/bin/env python
import json
import sqlite3

import sys

from flask import Flask, send_from_directory, request, g, _app_ctx_stack
import flask
from datapopulator import DataPopulator
from dbmigrator import DbMigrator
from responseBuilder import ResponseBuilder

# configuration
DATABASE = 'test.db'  # default database file

# harApp: Human Activity Recognition Flask Application
harApp = Flask(__name__, static_folder="static")
""":type : Flask"""

# Set configuration to grab the ALL CAPS variables from this file
harApp.config.from_object(__name__)

def get_db():
    """Opens a new database connection if there is none yet for the
    current application context.
    """
    top = _app_ctx_stack.top
    if not hasattr(top, 'sqlite_db'):
        sqlite_db = sqlite3.connect(harApp.config['DATABASE'])
        sqlite_db.row_factory = sqlite3.Row
        top.sqlite_db = sqlite_db

    return top.sqlite_db

starting_migrations = [
    """
        CREATE TABLE sessions (id INTEGER PRIMARY KEY NOT NULL, activity INTEGER, user INTEGER);
    """,
    """
        CREATE TABLE raw_samples (
        id INTEGER PRIMARY KEY NOT NULL,
        session_id INTEGER,
        timestamp REAL,
        heartrate REAL,
        handAccX  REAL,
        handAccY  REAL,
        handAccZ  REAL,
        handGyrX  REAL,
        handGyrY  REAL,
        handGyrZ  REAL,
        handMagX  REAL,
        handMagY  REAL,
        handMagZ  REAL,
        chestAccX  REAL,
        chestAccY  REAL,
        chestAccZ  REAL,
        chestGyrX  REAL,
        chestGyrY  REAL,
        chestGyrZ  REAL,
        chestMagX  REAL,
        chestMagY  REAL,
        chestMagZ  REAL,
        ankleAccX  REAL,
        ankleAccY  REAL,
        ankleAccZ  REAL,
        ankleGyrX  REAL,
        ankleGyrY  REAL,
        ankleGyrZ  REAL,
        ankleMagX  REAL,
        ankleMagY  REAL,
        ankleMagZ  REAL,
        FOREIGN KEY(session_id) REFERENCES sessions(id)
        );
    """
]

@harApp.route("/")
def hello():
    return "Hello World!"

@harApp.route("/view")
def starterView():
    return send_from_directory('static/', "view.html")

@harApp.route("/test/view")
def testView():
    return send_from_directory('static/', "view.html")

@harApp.route("/data")
def getData():
    responseBuilder = ResponseBuilder(request.args)
    response = responseBuilder.build()
    return json.dumps(response)

@harApp.route("/test/data")
def getTestData():
    with open('static/testresponse.js') as file:
        return json.dumps(json.load(file))


@harApp.before_request
def set_app_db():
    g.db = get_db()

@harApp.teardown_request
def close_app_db(exception):
    db = getattr(g, 'db', None)
    if db is not None:
        db.close()

def usage():
    print "Usage: "
    print "start.py [populatedb] [<db filename>]"
    print "where the <db filename> is the name of the database to use. The populatedb argument populates the database" \
          "with with the PAMAP dataset. Omit it to simply run the Flask web server."

# To populate the DB: python start.py populatedb
if __name__ == "__main__":
    dbFilename = None
    if (len(sys.argv) > 3):
        usage()
        exit()
    if (len(sys.argv) < 2):
        harApp.run(host='0.0.0.0')
    elif (sys.argv[1] == "populatedb"):
        if len(sys.argv) == 3:
            dbFilename = sys.argv[2]
        else:
            dbFilename = "test.db"
        dataMigrator = DbMigrator(dbFilename)
        dataMigrator.migrate(starting_migrations)
        dataPopulator = DataPopulator()
        dataPopulator.parseDataAndPersistIntoDb(dbFilename)
    else:
        usage()
        exit()