-
Notifications
You must be signed in to change notification settings - Fork 2
/
create_db.py
56 lines (48 loc) · 1.32 KB
/
create_db.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import sqlite3, config
connection = sqlite3.connect(config.DB_PATH)
cursor = connection.cursor()
cursor.execute("""
CREATE TABLE IF NOT EXISTS stock (
id INTEGER PRIMARY KEY,
symbol TEXT NOT NULL UNIQUE,
name TEXT NOT NULL,
exchange TEXT NOT NULL,
shortable BOOLEAN NOT NULL
)
""")
cursor.execute("""
CREATE TABLE IF NOT EXISTS stock_price (
id INTEGER PRIMARY KEY,
stock_id INTEGER,
date NOT NULL,
open NOT NULL,
high NOT NULL,
low NOT NULL,
close NOT NULL,
volume NOT NULL,
sma_20,
sma_50,
rsi_14,
FOREIGN KEY (stock_id) REFERENCES stock (id)
)
""")
cursor.execute("""
CREATE TABLE IF NOT EXISTS strategy (
id INTEGER PRIMARY KEY,
name NOT NULL
)
""")
cursor.execute("""
CREATE TABLE IF NOT EXISTS stock_strategy (
stock_id INTEGER NOT NULL,
strategy_id INTEGER NOT NULL,
FOREIGN KEY (stock_id) REFERENCES stock(id)
FOREIGN KEY (strategy_id) REFERENCES strategy(id)
)
""")
strategies = ['opening_range_breakout', 'opening_range_breakdown']
for strat in strategies:
cursor.execute("""
INSERT INTO strategy (name) VALUES (?)
""", (strat,))
connection.commit()