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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2012 Steffen Hoffmann <hoff.st@web.de>
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution.
from trac.core import Component, TracError, implements
from trac.db.api import DatabaseManager
from trac.env import IEnvironmentSetupParticipant
from tractags import db_default
from tractags.api import _
class TagSetup(Component):
"""Plugin setup and upgrade handler."""
implements(IEnvironmentSetupParticipant)
# IEnvironmentSetupParticipant methods
def environment_created(self):
pass
def environment_needs_upgrade(self, db):
schema_ver = self.get_schema_version(db)
if schema_ver == db_default.schema_version:
return False
elif schema_ver > db_default.schema_version:
raise TracError(_("""A newer plugin version has been installed
before, but downgrading is unsupported."""))
self.log.info("TracTags database schema version is %d, should be %d"
% (schema_ver, db_default.schema_version))
return True
def upgrade_environment(self, db):
"""Each schema version should have its own upgrade module, named
upgrades/dbN.py, where 'N' is the version number (int).
"""
db_mgr = DatabaseManager(self.env)
schema_ver = self.get_schema_version(db)
cursor = db.cursor()
# Is this a new installation?
if not schema_ver:
# Perform a single-step install: Create plugin schema and
# insert default data into the database.
connector = db_mgr._get_connector()[0]
for table in db_default.schema:
for stmt in connector.to_sql(table):
cursor.execute(stmt)
for table, cols, vals in db_default.get_data(db):
cursor.executemany("INSERT INTO %s (%s) VALUES (%s)" % (table,
','.join(cols),
','.join(['%s' for c in cols])), vals)
else:
# Perform incremental upgrades.
for i in range(schema_ver + 1, db_default.schema_version + 1):
name = 'db%i' % i
try:
upgrades = __import__('tractags.upgrades', globals(),
locals(), [name])
script = getattr(upgrades, name)
except AttributeError:
raise TracError(_("""
No upgrade module for version %(num)i (%(version)s.py)
""", num=i, version=name))
script.do_upgrade(self.env, i, cursor)
cursor.execute("""
UPDATE system
SET value=%s
WHERE name='tags_version'
""", (db_default.schema_version,))
self.log.info("Upgraded TracTags db schema from version %d to %d"
% (schema_ver, db_default.schema_version))
db.commit()
# Internal methods
def get_schema_version(self, db=None):
"""Return the current schema version for this plugin."""
if not db:
db = self.env.get_db_cnx()
cursor = db.cursor()
cursor.execute("""
SELECT value
FROM system
WHERE name='tags_version'
""")
row = cursor.fetchone()
if not (row and int(row[0]) > 2):
# Care for pre-tags-0.7 installations.
dburi = self.config.get('trac', 'database')
tables = self._get_tables(dburi, cursor)
if 'tags' in tables:
self.env.log.debug("TracTags needs to register schema version")
return 2
if 'wiki_namespace' in tables:
self.env.log.debug("TracTags needs to migrate old data")
return 1
# This is a new installation.
return 0
# The expected outcome for any up-to-date installation.
return row and int(row[0]) or 0
def _get_tables(self, dburi, cursor):
"""Code from TracMigratePlugin by Jun Omae (see tracmigrate.admin)."""
if dburi.startswith('sqlite:'):
sql = """
SELECT name
FROM sqlite_master
WHERE type='table'
AND NOT name='sqlite_sequence'
"""
elif dburi.startswith('postgres:'):
sql = """
SELECT tablename
FROM pg_tables
WHERE schemaname = ANY (current_schemas(false))
"""
elif dburi.startswith('mysql:'):
sql = "SHOW TABLES"
else:
raise TracError('Unsupported database type "%s"'
% dburi.split(':')[0])
cursor.execute(sql)
return sorted([row[0] for row in cursor])
|
