-
Notifications
You must be signed in to change notification settings - Fork 0
/
migrate.py
64 lines (52 loc) · 1.39 KB
/
migrate.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
57
58
59
60
61
62
63
from pony.orm import *
from functools import wraps
from db import db
class SchemaMigration(db.Entity):
version = PrimaryKey(int)
db.generate_mapping(create_tables=True)
def migration(version):
def decorator(func):
@wraps(func)
def wrapper():
if SchemaMigration.get(version=version) is None:
func()
SchemaMigration(version=version)
else:
pass
return wrapper
return decorator
@db_session
@migration(1)
def initial_version():
db.execute("""
CREATE TABLE Person (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
email TEXT
)
""")
db.execute("""
CREATE TABLE Role (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT UNIQUE NOT NULL
)
""")
db.execute("""
CREATE TABLE Assignment (
id INTEGER PRIMARY KEY AUTOINCREMENT,
person INTEGER NOT NULL REFERENCES Person (id),
role INTEGER NOT NULL REFERENCES Role (id),
year INTEGER NOT NULL,
semester INTEGER NOT NULL
)
""")
db.execute("""
CREATE INDEX idx_assignment__person ON Assignment (person)
""")
db.execute("""
CREATE INDEX idx_assignment__role ON Assignment (role)
""")
def migrate():
initial_version()
if __name__ == "__main__":
migrate()