-
Notifications
You must be signed in to change notification settings - Fork 2k
/
system_info.py
78 lines (57 loc) · 1.93 KB
/
system_info.py
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
# encoding: utf-8
'''
The system_info table and SystemInfo mapped class store runtime-editable
configuration options.
For more details, check :doc:`maintaining/configuration`.
'''
from sqlalchemy import types, Column, Table
from six import text_type
import vdm.sqlalchemy
import meta
import core
import domain_object
__all__ = ['system_info_table', 'SystemInfo',
'get_system_info', 'set_system_info']
system_info_table = Table(
'system_info', meta.metadata,
Column('id', types.Integer(), primary_key=True, nullable=False),
Column('key', types.Unicode(100), unique=True, nullable=False),
Column('value', types.UnicodeText),
Column('state', types.UnicodeText, default=core.State.ACTIVE),
)
class SystemInfo(core.StatefulObjectMixin,
domain_object.DomainObject):
def __init__(self, key, value):
super(SystemInfo, self).__init__()
self.key = key
self.value = text_type(value)
meta.mapper(SystemInfo, system_info_table)
def get_system_info(key, default=None):
''' get data from system_info table '''
from sqlalchemy.exc import ProgrammingError
try:
obj = meta.Session.query(SystemInfo).filter_by(key=key).first()
if obj:
return obj.value
except ProgrammingError:
meta.Session.rollback()
return default
def delete_system_info(key, default=None):
''' delete data from system_info table '''
obj = meta.Session.query(SystemInfo).filter_by(key=key).first()
if obj:
meta.Session.delete(obj)
meta.Session.commit()
def set_system_info(key, value):
''' save data in the system_info table '''
obj = None
obj = meta.Session.query(SystemInfo).filter_by(key=key).first()
if obj and obj.value == text_type(value):
return
if not obj:
obj = SystemInfo(key, value)
else:
obj.value = text_type(value)
meta.Session.add(obj)
meta.Session.commit()
return True