1#-------------------------------------------------------------------------
2#  This file contains real Python object wrappers for DB and DBEnv
3#  C "objects" that can be usefully subclassed.  The previous SWIG
4#  based interface allowed this thanks to SWIG's shadow classes.
5#   --  Gregory P. Smith
6#-------------------------------------------------------------------------
7#
8# (C) Copyright 2001  Autonomous Zone Industries
9#
10# License:  This is free software.  You may use this software for any
11#           purpose including modification/redistribution, so long as
12#           this header remains intact and that you do not claim any
13#           rights of ownership or authorship of this software.  This
14#           software has been tested, but no warranty is expressed or
15#           implied.
16#
17
18#
19# TODO it would be *really nice* to have an automatic shadow class populator
20# so that new methods don't need to be added  here manually after being
21# added to _berkeleydb.c.
22#
23
24import sys
25from collections.abc import MutableMapping
26from . import db
27
28class DBEnv:
29    def __init__(self, *args, **kwargs):
30        self._cobj = db.DBEnv(*args, **kwargs)
31
32    def close(self, *args, **kwargs):
33        return self._cobj.close(*args, **kwargs)
34    def open(self, *args, **kwargs):
35        return self._cobj.open(*args, **kwargs)
36    def remove(self, *args, **kwargs):
37        return self._cobj.remove(*args, **kwargs)
38    def set_shm_key(self, *args, **kwargs):
39        return self._cobj.set_shm_key(*args, **kwargs)
40    def set_cachesize(self, *args, **kwargs):
41        return self._cobj.set_cachesize(*args, **kwargs)
42    def set_data_dir(self, *args, **kwargs):
43        return self._cobj.set_data_dir(*args, **kwargs)
44    def set_flags(self, *args, **kwargs):
45        return self._cobj.set_flags(*args, **kwargs)
46    def set_lg_bsize(self, *args, **kwargs):
47        return self._cobj.set_lg_bsize(*args, **kwargs)
48    def set_lg_dir(self, *args, **kwargs):
49        return self._cobj.set_lg_dir(*args, **kwargs)
50    def set_lg_max(self, *args, **kwargs):
51        return self._cobj.set_lg_max(*args, **kwargs)
52    def set_lk_detect(self, *args, **kwargs):
53        return self._cobj.set_lk_detect(*args, **kwargs)
54    def set_lk_max_locks(self, *args, **kwargs):
55        return self._cobj.set_lk_max_locks(*args, **kwargs)
56    def set_lk_max_lockers(self, *args, **kwargs):
57        return self._cobj.set_lk_max_lockers(*args, **kwargs)
58    def set_lk_max_objects(self, *args, **kwargs):
59        return self._cobj.set_lk_max_objects(*args, **kwargs)
60    def set_mp_mmapsize(self, *args, **kwargs):
61        return self._cobj.set_mp_mmapsize(*args, **kwargs)
62    def set_timeout(self, *args, **kwargs):
63        return self._cobj.set_timeout(*args, **kwargs)
64    def set_tmp_dir(self, *args, **kwargs):
65        return self._cobj.set_tmp_dir(*args, **kwargs)
66    def txn_begin(self, *args, **kwargs):
67        return self._cobj.txn_begin(*args, **kwargs)
68    def txn_checkpoint(self, *args, **kwargs):
69        return self._cobj.txn_checkpoint(*args, **kwargs)
70    def txn_stat(self, *args, **kwargs):
71        return self._cobj.txn_stat(*args, **kwargs)
72    def set_tx_max(self, *args, **kwargs):
73        return self._cobj.set_tx_max(*args, **kwargs)
74    def set_tx_timestamp(self, *args, **kwargs):
75        return self._cobj.set_tx_timestamp(*args, **kwargs)
76    def lock_detect(self, *args, **kwargs):
77        return self._cobj.lock_detect(*args, **kwargs)
78    def lock_get(self, *args, **kwargs):
79        return self._cobj.lock_get(*args, **kwargs)
80    def lock_id(self, *args, **kwargs):
81        return self._cobj.lock_id(*args, **kwargs)
82    def lock_put(self, *args, **kwargs):
83        return self._cobj.lock_put(*args, **kwargs)
84    def lock_stat(self, *args, **kwargs):
85        return self._cobj.lock_stat(*args, **kwargs)
86    def log_archive(self, *args, **kwargs):
87        return self._cobj.log_archive(*args, **kwargs)
88
89    def set_get_returns_none(self, *args, **kwargs):
90        return self._cobj.set_get_returns_none(*args, **kwargs)
91
92    def log_stat(self, *args, **kwargs):
93        return self._cobj.log_stat(*args, **kwargs)
94
95    def dbremove(self, *args, **kwargs):
96        return self._cobj.dbremove(*args, **kwargs)
97    def dbrename(self, *args, **kwargs):
98        return self._cobj.dbrename(*args, **kwargs)
99    def set_encrypt(self, *args, **kwargs):
100        return self._cobj.set_encrypt(*args, **kwargs)
101
102    def fileid_reset(self, *args, **kwargs):
103        return self._cobj.fileid_reset(*args, **kwargs)
104
105    def lsn_reset(self, *args, **kwargs):
106        return self._cobj.lsn_reset(*args, **kwargs)
107
108
109class DB(MutableMapping):
110    def __init__(self, dbenv, *args, **kwargs):
111        # give it the proper DBEnv C object that its expecting
112        self._cobj = db.DB(*((dbenv._cobj,) + args), **kwargs)
113
114    # TODO are there other dict methods that need to be overridden?
115    def __len__(self):
116        return len(self._cobj)
117    def __getitem__(self, arg):
118        return self._cobj[arg]
119    def __setitem__(self, key, value):
120        self._cobj[key] = value
121    def __delitem__(self, arg):
122        del self._cobj[arg]
123
124    def __iter__(self) :
125        return self._cobj.__iter__()
126
127    def append(self, *args, **kwargs):
128        return self._cobj.append(*args, **kwargs)
129    def associate(self, *args, **kwargs):
130        return self._cobj.associate(*args, **kwargs)
131    def close(self, *args, **kwargs):
132        return self._cobj.close(*args, **kwargs)
133    def consume(self, *args, **kwargs):
134        return self._cobj.consume(*args, **kwargs)
135    def consume_wait(self, *args, **kwargs):
136        return self._cobj.consume_wait(*args, **kwargs)
137    def cursor(self, *args, **kwargs):
138        return self._cobj.cursor(*args, **kwargs)
139    def delete(self, *args, **kwargs):
140        return self._cobj.delete(*args, **kwargs)
141    def fd(self, *args, **kwargs):
142        return self._cobj.fd(*args, **kwargs)
143    def get(self, *args, **kwargs):
144        return self._cobj.get(*args, **kwargs)
145    def pget(self, *args, **kwargs):
146        return self._cobj.pget(*args, **kwargs)
147    def get_both(self, *args, **kwargs):
148        return self._cobj.get_both(*args, **kwargs)
149    def get_byteswapped(self, *args, **kwargs):
150        return self._cobj.get_byteswapped(*args, **kwargs)
151    def get_size(self, *args, **kwargs):
152        return self._cobj.get_size(*args, **kwargs)
153    def get_type(self, *args, **kwargs):
154        return self._cobj.get_type(*args, **kwargs)
155    def join(self, *args, **kwargs):
156        return self._cobj.join(*args, **kwargs)
157    def key_range(self, *args, **kwargs):
158        return self._cobj.key_range(*args, **kwargs)
159    def has_key(self, *args, **kwargs):
160        return self._cobj.has_key(*args, **kwargs)
161    def items(self, *args, **kwargs):
162        return self._cobj.items(*args, **kwargs)
163    def keys(self, *args, **kwargs):
164        return self._cobj.keys(*args, **kwargs)
165    def open(self, *args, **kwargs):
166        return self._cobj.open(*args, **kwargs)
167    def put(self, *args, **kwargs):
168        return self._cobj.put(*args, **kwargs)
169    def remove(self, *args, **kwargs):
170        return self._cobj.remove(*args, **kwargs)
171    def rename(self, *args, **kwargs):
172        return self._cobj.rename(*args, **kwargs)
173    def set_bt_minkey(self, *args, **kwargs):
174        return self._cobj.set_bt_minkey(*args, **kwargs)
175    def set_bt_compare(self, *args, **kwargs):
176        return self._cobj.set_bt_compare(*args, **kwargs)
177    def set_cachesize(self, *args, **kwargs):
178        return self._cobj.set_cachesize(*args, **kwargs)
179    def set_dup_compare(self, *args, **kwargs) :
180        return self._cobj.set_dup_compare(*args, **kwargs)
181    def set_flags(self, *args, **kwargs):
182        return self._cobj.set_flags(*args, **kwargs)
183    def set_h_ffactor(self, *args, **kwargs):
184        return self._cobj.set_h_ffactor(*args, **kwargs)
185    def set_h_nelem(self, *args, **kwargs):
186        return self._cobj.set_h_nelem(*args, **kwargs)
187    def set_lorder(self, *args, **kwargs):
188        return self._cobj.set_lorder(*args, **kwargs)
189    def set_pagesize(self, *args, **kwargs):
190        return self._cobj.set_pagesize(*args, **kwargs)
191    def set_re_delim(self, *args, **kwargs):
192        return self._cobj.set_re_delim(*args, **kwargs)
193    def set_re_len(self, *args, **kwargs):
194        return self._cobj.set_re_len(*args, **kwargs)
195    def set_re_pad(self, *args, **kwargs):
196        return self._cobj.set_re_pad(*args, **kwargs)
197    def set_re_source(self, *args, **kwargs):
198        return self._cobj.set_re_source(*args, **kwargs)
199    def set_q_extentsize(self, *args, **kwargs):
200        return self._cobj.set_q_extentsize(*args, **kwargs)
201    def stat(self, *args, **kwargs):
202        return self._cobj.stat(*args, **kwargs)
203    def sync(self, *args, **kwargs):
204        return self._cobj.sync(*args, **kwargs)
205    def type(self, *args, **kwargs):
206        return self._cobj.type(*args, **kwargs)
207    def upgrade(self, *args, **kwargs):
208        return self._cobj.upgrade(*args, **kwargs)
209    def values(self, *args, **kwargs):
210        return self._cobj.values(*args, **kwargs)
211    def verify(self, *args, **kwargs):
212        return self._cobj.verify(*args, **kwargs)
213    def set_get_returns_none(self, *args, **kwargs):
214        return self._cobj.set_get_returns_none(*args, **kwargs)
215
216    def set_encrypt(self, *args, **kwargs):
217        return self._cobj.set_encrypt(*args, **kwargs)
218
219
220class DBSequence:
221    def __init__(self, *args, **kwargs):
222        self._cobj = db.DBSequence(*args, **kwargs)
223
224    def close(self, *args, **kwargs):
225        return self._cobj.close(*args, **kwargs)
226    def get(self, *args, **kwargs):
227        return self._cobj.get(*args, **kwargs)
228    def get_dbp(self, *args, **kwargs):
229        return self._cobj.get_dbp(*args, **kwargs)
230    def get_key(self, *args, **kwargs):
231        return self._cobj.get_key(*args, **kwargs)
232    def init_value(self, *args, **kwargs):
233        return self._cobj.init_value(*args, **kwargs)
234    def open(self, *args, **kwargs):
235        return self._cobj.open(*args, **kwargs)
236    def remove(self, *args, **kwargs):
237        return self._cobj.remove(*args, **kwargs)
238    def stat(self, *args, **kwargs):
239        return self._cobj.stat(*args, **kwargs)
240    def set_cachesize(self, *args, **kwargs):
241        return self._cobj.set_cachesize(*args, **kwargs)
242    def set_flags(self, *args, **kwargs):
243        return self._cobj.set_flags(*args, **kwargs)
244    def set_range(self, *args, **kwargs):
245        return self._cobj.set_range(*args, **kwargs)
246    def get_cachesize(self, *args, **kwargs):
247        return self._cobj.get_cachesize(*args, **kwargs)
248    def get_flags(self, *args, **kwargs):
249        return self._cobj.get_flags(*args, **kwargs)
250    def get_range(self, *args, **kwargs):
251        return self._cobj.get_range(*args, **kwargs)
252