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
|
# -*- coding: utf-8 -*-
from __future__ import absolute_import
from . import get_db_type
SQL = {
"sqlite": ["""
CREATE TABLE "category" (
id INTEGER NOT NULL,
name VARCHAR(255) NOT NULL,
PRIMARY KEY (id)
);""",
'ALTER TABLE "thread" ADD COLUMN category_id INTEGER;',
'CREATE UNIQUE INDEX "ix_category_name" ON "category" (name);',
],
"postgres": ["""
CREATE TABLE "category" (
id INTEGER NOT NULL,
name VARCHAR(255) NOT NULL,
PRIMARY KEY (id)
);""", """
CREATE SEQUENCE category_id_seq
START WITH 1
INCREMENT BY 1
NO MAXVALUE
NO MINVALUE
CACHE 1
;""",
"ALTER SEQUENCE category_id_seq OWNED BY category.id;",
"ALTER TABLE ONLY category ALTER COLUMN id SET DEFAULT nextval('category_id_seq'::regclass);",
'ALTER TABLE "thread" ADD COLUMN category_id INTEGER;',
'ALTER TABLE "thread" ADD FOREIGN KEY (category_id) REFERENCES category(id);',
'CREATE UNIQUE INDEX "ix_category_name" ON "category" USING btree (name);',
],
"mysql": ["""
CREATE TABLE `category` (
id INTEGER NOT NULL AUTO_INCREMENT,
name VARCHAR(255) NOT NULL,
PRIMARY KEY (id)
);""",
'ALTER TABLE `thread` ADD COLUMN category_id INTEGER;',
'ALTER TABLE `thread` ADD FOREIGN KEY (category_id) REFERENCES category(id);',
'CREATE UNIQUE INDEX `ix_category_name` ON `category` (name);',
],
}
def apply(store):
"""Add the subject_prefix column and delete the description column"""
dbtype = get_db_type(store)
for statement in SQL[dbtype]:
store.execute(statement)
store.commit()
|