Question

我正在为shelve模块编写一个包装类，我打算像字典一样使用它。这是代码：

import shelve

class MyShelve:

    def __init__(self, filename='myshelve.db'):
        self.s = shelve.open(filename)

    def __del__(self):
        self.s.close()

    def __repr__(self):
        return repr(self.s)

    def __getitem__(self, k):
        return self.s.get(k, None)

    def __setitem__(self, k, v):
        self.s[k] = v

在我使用dict＆＃34;中的表达式＆＃34;键之前，一切似乎都运行良好。这是一个示例会话：

>>> d = {'1': 'One', '2': 'Two'}
>>> d
{'1': 'One', '2': 'Two'}
>>> '1' in d
True
>>> from myshelve import MyShelve
>>> s = MyShelve()
>>> s['1'] = 'One'
>>> s['2'] = 'Two'
>>> s
{'1': 'One', '2': 'Two'}
>>> '1' in s.s
True
>>> '1' in s
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "myshelve.py", line 15, in __getitem__
    return self.s.get(k, None)
  File "/usr/lib64/python2.7/shelve.py", line 113, in get
    if key in self.dict:
  File "/usr/lib64/python2.7/_abcoll.py", line 369, in __contains__
    self[key]
  File "/usr/lib64/python2.7/bsddb/__init__.py", line 270, in __getitem__
    return _DeadlockWrap(lambda: self.db[key])  # self.db[key]
  File "/usr/lib64/python2.7/bsddb/dbutils.py", line 68, in DeadlockWrap
    return function(*_args, **_kwargs)
  File "/usr/lib64/python2.7/bsddb/__init__.py", line 270, in <lambda>
    return _DeadlockWrap(lambda: self.db[key])  # self.db[key]
TypeError: Integer keys only allowed for Recno and Queue DB's

我做错了什么？

Answer 1

尝试使用__contain__方法，并自定义案例。它是“in”操作符调用的内容。

class MyShelve:

    def __init__(self,filename='myshelve.db'):
        self.s = shelve.open(filename)

    def __del__(self):
        self.s.close()

    def __repr__(self):
        return repr(self.s)

    def __getitem__(self, k):
        return self.s.get(k, None)

    def __setitem__(self, k, v):
        self.s[k] = v
    def __contains__(self, m):
            return True if (m in self.s.values() or self.s.keys)  else False

    def close(self):
        self.s.close()

    def clear(self):
        self.s.clear()

    def items(self):
         return self._shelve.iteritems()

s= MyShelve()

s['1'] = 'One'
s['2'] = 'two'
print '1' in s #Output: True

Answer 2

首先，始终从object继承。它将在以后为您节省很多麻烦。其次，您需要使用__contains__。第三，在使用__contains__或__getitem__或任何 dunder方法时，请确保使用异常，即try-except块。以下是您要查找的内容示例：

class MyShelve(object):

    def __init__(self, filename='myshelve.db'):
        self.s = shelve.open(filename)

    def __del__(self):
        self.s.close()

    def __repr__(self):
        return repr(self.s)

    def __getitem__(self, item):
        return self.s.get(item, False)

    def __contains__(self, item):
        try:
            return item in self.s
        except TypeError:
            return False

    def __setitem__(self, k, v):
        self.s[k] = v

演示：

In[3]: from shelving import MyShelve
In[4]: s = MyShelve()
In[5]: s['1'] = 'One'
In[6]: s['2'] = 'Two'
In[7]: '1' in s
Out[7]: True

请注意，如果没有异常块，表达式3 in s将评估为：

Traceback (most recent call last):
(...)
TypeError: gdbm key must be string, not int

事后看来，最好使用具有预配置值的函数，而不是在这种情况下使用类，因为你将覆盖很多东西，这将浪费你的时间。此外，它更容易使用带有返回文件对象的函数的上下文管理器，而不是具有封装文件对象的类（因为当你打开一个搁架时，实际上是创建了一个文件）。

在Python中使用带有搁置模块的repr

2 个答案:

在Python中使用带有搁置模块的__repr__

2 个答案:

在Python中使用带有搁置模块的repr