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
|
#This file is part of Tryton. The COPYRIGHT file at the top level of
#this repository contains the full copyright notices and license terms.
from trytond.model import ModelStorage
class ModelSingleton(ModelStorage):
"""
Define a singleton model in Tryton.
"""
def get_singleton_id(self, cursor, user, context=None):
'''
Return the id of the unique record if there is one.
:param cursor: the database cursor
:param user: the user id
:param context: the context
'''
singleton_ids = super(ModelSingleton, self).search(cursor, user, [],
limit=1, context=context)
if singleton_ids:
return singleton_ids[0]
return False
def create(self, cursor, user, values, context=None):
singleton_id = self.get_singleton_id(cursor, user, context=context)
if singleton_id:
self.write(cursor, user, singleton_id, values, context=context)
else:
singleton_id = super(ModelSingleton, self).create(cursor,
user, values, context=context)
return singleton_id
def read(self, cursor, user, ids, fields_names=None, context=None):
singleton_id = self.get_singleton_id(cursor, user, context=context)
if not singleton_id:
res = self.default_get(cursor, user, fields_names, context=context,
with_rec_name=False)
if not isinstance(ids, (int, long)):
res['id'] = ids[0]
res = [res]
else:
res['id'] = ids
return res
if isinstance(ids, (int, long)):
ids2 = singleton_id
else:
ids2 = [singleton_id]
res = super(ModelSingleton, self).read(cursor, user, ids2,
fields_names=fields_names, context=context)
if isinstance(ids, (int, long)):
res['id'] = ids
else:
res[0]['id'] = ids[0]
return res
def write(self, cursor, user, ids, values, context=None):
singleton_id = self.get_singleton_id(cursor, user, context=context)
if not singleton_id:
return self.create(cursor, user, values, context=context)
if isinstance(ids, (int, long)):
ids = singleton_id
else:
ids = [singleton_id]
return super(ModelSingleton, self).write(cursor, user, ids, values,
context=context)
def delete(self, cursor, user, ids, context=None):
singleton_id = self.get_singleton_id(cursor, user, context=context)
if not singleton_id:
return True
if isinstance(ids, (int, long)):
ids = singleton_id
else:
ids = [singleton_id]
return super(ModelSingleton, self).delete(cursor, user, ids,
context=context)
def copy(self, cursor, user, ids, default=None, context=None):
if default:
self.write(cursor, user, ids, default, context=context)
return ids
def search(self, cursor, user, domain, offset=0, limit=None, order=None,
context=None, count=False):
res = super(ModelSingleton, self).search(cursor, user, domain,
offset=offset, limit=limit, order=order, context=context,
count=count)
if not res:
if count:
return 1
return [1]
return res
def default_get(self, cursor, user, fields_names, context=None,
with_rec_name=True):
res = super(ModelSingleton, self).default_get(cursor, user,
fields_names, context=context, with_rec_name=with_rec_name)
singleton_id = self.get_singleton_id(cursor, user, context=context)
if singleton_id:
if with_rec_name:
fields_names = fields_names[:]
for field in fields_names[:]:
if self._columns[field]._type in ('many2one',):
fields_names.append(field + '.rec_name')
res = self.read(cursor, user, singleton_id,
fields_names=fields_names, context=context)
for field in (x for x in res.keys() if x not in fields_names):
del res[field]
return res
|
