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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
from trac.core import *
from trac.db.schema import Table, Column, Index
from trac.env import IEnvironmentSetupParticipant
from trac.db.api import DatabaseManager
# Database version identifier for upgrades.
db_version = 1
# Database schema
schema = {
'code_comments': Table('code_comments', key=('id', 'version'))[
Column('id', auto_increment=True),
Column('version', type='int'),
Column('text'),
Column('path'),
Column('revision', type='int'),
Column('line', type='int'),
Column('author'),
Column('time', type='int'),
Index(['path']),
Index(['author']),
],
}
def to_sql(env, table):
""" Convenience function to get the to_sql for the active connector."""
dc = DatabaseManager(env)._get_connector()[0]
return dc.to_sql(table)
def create_tables(env, db):
cursor = db.cursor()
for table_name in schema:
for stmt in to_sql(env, schema[table_name]):
cursor.execute(stmt)
cursor.execute("INSERT into system values ('code_comments_schema_version', %s)",
str(db_version))
# Upgrades
def upgrade_from_1_to_2(env, db):
pass
upgrade_map = {
2: upgrade_from_1_to_2
}
class CodeCommentsSetup(Component):
"""Component that deals with database setup and upgrades."""
implements(IEnvironmentSetupParticipant)
def environment_created(self):
"""Called when a new Trac environment is created."""
pass
def environment_needs_upgrade(self, db):
"""Called when Trac checks whether the environment needs to be upgraded.
Returns `True` if upgrade is needed, `False` otherwise."""
return self._get_version(db) != db_version
def upgrade_environment(self, db):
"""Actually perform an environment upgrade, but don't commit as
that is done by the common upgrade procedure when all plugins are done."""
current_ver = self._get_version(db)
if current_ver == 0:
create_tables(self.env, db)
else:
while current_ver+1 <= db_version:
upgrade_map[current_ver+1](self.env, db)
current_ver += 1
cursor = db.cursor()
cursor.execute("UPDATE system SET value=%s WHERE name='code_comments_schema_version'",
str(db_version))
def _get_version(self, db):
cursor = db.cursor()
try:
sql = "SELECT value FROM system WHERE name='code_comments_schema_version'"
cursor.execute(sql)
for row in cursor:
return int(row[0])
return 0
except:
return 0
|
