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 | ||
3c43cfc9 | 18 | from sqlalchemy.ext.declarative import declarative_base |
9f264942 | 19 | from sqlalchemy.orm import scoped_session, sessionmaker, object_session |
92edc74e E |
20 | from sqlalchemy.orm.query import Query |
21 | from sqlalchemy.sql.expression import desc | |
22 | from mediagoblin.db.sql.fake import DESCENDING | |
7b194a79 E |
23 | |
24 | ||
92edc74e E |
25 | def _get_query_model(query): |
26 | cols = query.column_descriptions | |
27 | assert len(cols) == 1, "These functions work only on simple queries" | |
28 | return cols[0]["type"] | |
29 | ||
30 | ||
31 | class GMGQuery(Query): | |
32 | def sort(self, key, direction): | |
33 | key_col = getattr(_get_query_model(self), key) | |
34 | if direction is DESCENDING: | |
35 | key_col = desc(key_col) | |
36 | return self.order_by(key_col) | |
37 | ||
38 | def skip(self, amount): | |
39 | return self.offset(amount) | |
40 | ||
41 | ||
42 | Session = scoped_session(sessionmaker(query_cls=GMGQuery)) | |
7b194a79 E |
43 | |
44 | ||
26089828 E |
45 | def _fix_query_dict(query_dict): |
46 | if '_id' in query_dict: | |
47 | query_dict['id'] = query_dict.pop('_id') | |
48 | ||
49 | ||
7b194a79 E |
50 | class GMGTableBase(object): |
51 | query = Session.query_property() | |
52 | ||
53 | @classmethod | |
54 | def find(cls, query_dict={}): | |
26089828 | 55 | _fix_query_dict(query_dict) |
7b194a79 E |
56 | return cls.query.filter_by(**query_dict) |
57 | ||
58 | @classmethod | |
59 | def find_one(cls, query_dict={}): | |
26089828 | 60 | _fix_query_dict(query_dict) |
7b194a79 | 61 | return cls.query.filter_by(**query_dict).first() |
26089828 E |
62 | |
63 | @classmethod | |
64 | def one(cls, query_dict): | |
4305580e | 65 | return cls.find(query_dict).one() |
03c22862 E |
66 | |
67 | def get(self, key): | |
68 | return getattr(self, key) | |
9f264942 | 69 | |
c60bbe07 E |
70 | def setdefault(self, key, defaultvalue): |
71 | # The key *has* to exist on sql. | |
72 | return getattr(self, key) | |
73 | ||
fbad3a9f | 74 | def save(self, validate=True): |
9f264942 E |
75 | assert validate |
76 | sess = object_session(self) | |
77 | if sess is None: | |
78 | sess = Session() | |
79 | sess.add(self) | |
80 | sess.commit() | |
3c43cfc9 | 81 | |
329e3903 E |
82 | def delete(self, commit=True): |
83 | """Delete the object and commit the change immediately by default""" | |
c60bbe07 E |
84 | sess = object_session(self) |
85 | assert sess is not None, "Not going to delete detached %r" % self | |
86 | sess.delete(self) | |
329e3903 E |
87 | if commit: |
88 | sess.commit() | |
c60bbe07 | 89 | |
3c43cfc9 E |
90 | |
91 | Base = declarative_base(cls=GMGTableBase) | |
de917303 E |
92 | |
93 | ||
94 | class DictReadAttrProxy(object): | |
95 | """ | |
96 | Maps read accesses to obj['key'] to obj.key | |
97 | and hides all the rest of the obj | |
98 | """ | |
99 | def __init__(self, proxied_obj): | |
100 | self.proxied_obj = proxied_obj | |
101 | ||
102 | def __getitem__(self, key): | |
103 | try: | |
104 | return getattr(self.proxied_obj, key) | |
105 | except AttributeError: | |
106 | raise KeyError("%r is not an attribute on %r" | |
107 | % (key, self.proxied_obj)) |