habitmove/loop/tables.py

import sqlite3


def create_database(name):
    return sqlite3.connect(name)


# better way to do the above
# def create_connection(db_file):
#     """create a database connection to the SQLite database
#         specified by db_file
#     :param db_file: database file
#     :return: Connection object or None
#     """
#     conn = None
#     try:
#         conn = sqlite3.connect(db_file)
#         return conn
#     except Error as e:
#         print(e)
#     return conn


def create_tables(db):
    c = db.cursor()
    c.execute(
        """ CREATE TABLE IF NOT EXISTS Habits (
                                        id integer PRIMARY KEY AUTOINCREMENT,
                                        archived integer,
                                        color integer,
                                        description text,
                                        freq_den integer,
                                        freq_num integer,
                                        highlight integer,
                                        name text,
                                        position integer,
                                        reminder_hour integer,
                                        reminder_min integer,
                                        reminder_days integer NOT NULL DEFAULT 127,
                                        type integer NOT NULL DEFAULT 0,
                                        target_type integer NOT NULL DEFAULT 0,
                                        target_value real NOT NULL DEFAULT 0,
                                        unit text NOT NULL DEFAULT "",
                                        question text,
                                        uuid text 
                                    ); """
    )
    c.execute(
        """ CREATE TABLE IF NOT EXISTS Repetitions (
                                        id integer PRIMARY KEY AUTOINCREMENT,
                                        habit integer NOT NULL REFERENCES Habits(id),
                                        timestamp integer NOT NULL,
                                        value integer NOT NULL
                                    ); """
    )


def migrate(name):
    db = create_database(name)
    create_tables(db)
    return db