ettpXRhU

1import sqlite3
2from sqlite3 import Error
3import os
4
5
6def create_connection(db_file):
7    """ create a database connection to the SQLite database
8        specified by db_file
9    :param db_file: database file
10    :return: Connection object or None
11    """
12    conn = None
13    try:
14        conn = sqlite3.connect(db_file)
15        return conn
16    except Error as e:
17        print(e)
18
19    return conn
20
21
22def create_table(conn, create_table_sql):
23    """ create a table from the create_table_sql statement
24    :param conn: Connection object
25    :param create_table_sql: a CREATE TABLE statement
26    :return:
27    """
28    try:
29        c = conn.cursor()
30        c.execute(create_table_sql)
31    except Error as e:
32        print(e)
33
34
35def main():
36    if not os.path.exists('db1'):
37        os.mkdir('db1')
38    database = 'db1/test.db'
39
40    sqlite_create_cvp_table = """  CREATE TABLE IF NOT EXISTS uploads (
41                                        id	TEXT UNIQUE,
42                                        status	NUMERIC,
43                                        isUploaded	INTEGER,
44                                        isProcessed	INTEGER,
45                                        location	TEXT,
46                                        datetime	TEXT,
47                                        model_name	TEXT,
48                                        pbtxt_name TEXT
49                                        );"""
50
51    # create a database connection
52    conn = create_connection(database)
53
54    # create tables
55    if conn is not None:
56        # create projects table
57        create_table(conn, sqlite_create_cvp_table)
58
59    else:
60        print("Error! cannot create the database connection.")
61
62
63if __name__ == '__main__':
64    main()