Commit | Line | Data |
---|---|---|
fbad3a9f | 1 | # GNU MediaGoblin -- federated, autonomous media hosting |
7f4ebeed | 2 | # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS. |
fbad3a9f E |
3 | # |
4 | # This program is free software: you can redistribute it and/or modify | |
5 | # it under the terms of the GNU Affero General Public License as published by | |
6 | # the Free Software Foundation, either version 3 of the License, or | |
7 | # (at your option) any later version. | |
8 | # | |
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU Affero General Public License for more details. | |
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | ||
17 | ||
7b194a79 | 18 | from sqlalchemy import create_engine |
193e0f53 | 19 | import logging |
7b194a79 E |
20 | |
21 | from mediagoblin.db.sql.base import Session | |
22 | from mediagoblin.db.sql.models import Base | |
23 | ||
24 | ||
25 | class DatabaseMaster(object): | |
26 | def __init__(self, engine): | |
27 | self.engine = engine | |
28 | ||
fbad3a9f | 29 | for k, v in Base._decl_class_registry.iteritems(): |
7b194a79 E |
30 | setattr(self, k, v) |
31 | ||
32 | def commit(self): | |
33 | Session.commit() | |
34 | ||
35 | def save(self, obj): | |
36 | Session.add(obj) | |
37 | Session.flush() | |
38 | ||
39 | def reset_after_request(self): | |
2bc8ff0d | 40 | Session.rollback() |
7b194a79 E |
41 | Session.remove() |
42 | ||
43 | ||
44 | def setup_connection_and_db_from_config(app_config): | |
193e0f53 E |
45 | engine = create_engine(app_config['sql_engine']) |
46 | logging.getLogger('sqlalchemy.engine').setLevel(logging.INFO) | |
7b194a79 E |
47 | Session.configure(bind=engine) |
48 | ||
49 | return "dummy conn", DatabaseMaster(engine) | |
415077a7 E |
50 | |
51 | ||
52 | def check_db_migrations_current(db): | |
53 | pass |