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
|
# Copyright (C) 2007 by the Free Software Foundation, Inc.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
# USA.
__all__ = [
'Address',
'Language',
'MailingList',
'Message',
'Pendings',
'Preferences',
'User',
'Version',
]
import os
import sys
import elixir
from sqlalchemy import create_engine
from string import Template
from urlparse import urlparse
import Mailman.Version
from Mailman import constants
from Mailman.Errors import SchemaVersionMismatchError
from Mailman.configuration import config
# This /must/ be set before any Elixir classes are defined (i.e. imported).
# This tells Elixir to use the short table names (i.e. the class name) instead
# of a mangled full class path.
elixir.options_defaults['shortnames'] = True
from Mailman.database.model.address import Address
from Mailman.database.model.language import Language
from Mailman.database.model.mailinglist import MailingList
from Mailman.database.model.member import Member
from Mailman.database.model.message import Message
from Mailman.database.model.pending import Pendings
from Mailman.database.model.preferences import Preferences
from Mailman.database.model.requests import Requests
from Mailman.database.model.user import User
from Mailman.database.model.version import Version
def initialize(debug):
# Calculate the engine url
url = Template(config.SQLALCHEMY_ENGINE_URL).safe_substitute(config.paths)
# XXX By design of SQLite, database file creation does not honor
# umask. See their ticket #1193:
# http://www.sqlite.org/cvstrac/tktview?tn=1193,31
#
# This sucks for us because the mailman.db file /must/ be group writable,
# however even though we guarantee our umask is 002 here, it still gets
# created without the necessary g+w permission, due to SQLite's policy.
# This should only affect SQLite engines because its the only one that
# creates a little file on the local file system. This kludges around
# their bug by "touch"ing the database file before SQLite has any chance
# to create it, thus honoring the umask and ensuring the right
# permissions. We only try to do this for SQLite engines, and yes, we
# could have chmod'd the file after the fact, but half dozen and all...
touch(url)
engine = create_engine(url)
engine.echo = (config.SQLALCHEMY_ECHO if debug is None else debug)
elixir.metadata.bind = engine
elixir.setup_all()
elixir.create_all()
# Validate schema version.
v = Version.get_by(component='schema')
if not v:
# Database has not yet been initialized
v = Version(component='schema',
version=Mailman.Version.DATABASE_SCHEMA_VERSION)
elixir.session.flush()
elif v.version <> Mailman.Version.DATABASE_SCHEMA_VERSION:
# XXX Update schema
raise SchemaVersionMismatchError(v.version)
def touch(url):
parts = urlparse(url)
if parts.scheme <> 'sqlite':
return
path = os.path.normpath(parts.path)
fd = os.open(path, os.O_WRONLY | os.O_NONBLOCK | os.O_CREAT, 0666)
# Ignore errors
if fd > 0:
os.close(fd)
def _reset():
for entity in elixir.entities:
for row in entity.query.filter_by().all():
row.delete()
|
