0

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

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

在我使用“字典中的键”这个表达之前，一切似乎都很好。这是一个示例会话：

>>> 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

我究竟做错了什么？

4

2 回答 2

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
于 2014-10-25T03:51:02.757 回答
1

首先，始终继承自object. 以后可以省去很多麻烦。其次，您需要使用__contains__. 第三，在使用__contains__or__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

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

于 2014-10-25T04:09:21.097 回答