如何使用上下文管理器以原子方式（退出时）在列表中附加值？

Question

首先，我是 Python 新手。

我正在尝试编写自己的上下文管理器，它将在退出时将值附加到我的列表中。 我创建了我的列表的副本，但似乎这不能正常工作。

这是我的代码：

import time
import copy

a = [1,2,3]


class contextmanager():
    def __init__(self, var):
        self.abc = var

    def current(self):

        b.append(98325)



        return b

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('Current list: {}'.format(self.current()))

with contextmanager(a) as t:
    time.sleep(1)
    b = copy.deepcopy(a)

    print("Changed list: {}".format(t.current()))
    time.sleep(2) 

我想在退出时添加值，例如：

当前 - [1,2,3]

附加 - [4,5,6]

退出前 - [1,2,3]

退出 - [1,2,3,4,5,6]

但事实并非如此，我明白了：

当前 - [1,2,3,4,5,6]

附加 - [4,5,6]

退出 - [1,2,3,4,5,6,4,5,6]

我究竟做错了什么？ 我该怎么做才能解决它？ 如果你告诉我我需要在我的代码中改变什么，我会很高兴。

我似乎还不太了解 Python.. 感谢您的帮助。

Answer 1

这不是您假设使用上下文管理器的方式。 首先 OO 原则建议进行适当的封装，这意味着用户不必知道实现细节。

我会提出以下设计：

• 上下文管理器将列表作为其创建参数
• 它提供了两种方法：
  • append(val)准备将 val 附加到列表并返回自身以允许链接
  • append appending()返回要追加的项目列表
• 仅当未引发异常时才附加准备好的项目

可能的实现：

class contextmanager():
    def __init__(self, var):
        self.orig = var
        self.cur = []

    def append(self, val):
        self.cur.append(val)
        return self

    def appending(self):
        return self.cur

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type is None:
            self.orig.extend(self.cur)

用法示例：

a = [1,2,3]

print("initial", a)
with contextmanager(a) as t:
    t.append(4)
    print("appending", t.appending())
    t.append(5).append(6)
    print("appending", t.appending())

print("first pass", a)

try:
    with contextmanager(a) as t:
        t.append(7)
        t.append(8)
        print("appending", t.appending())
        raise Exception()
        t.append(9)
        print("appending", t.appending())
except Exception as e:
    print("Exception", e)

print("final", a)

这使：

initial [1, 2, 3]
appending [4]
appending [4, 5, 6]
first pass [1, 2, 3, 4, 5, 6]
appending [7, 8]
Exception 
final [1, 2, 3, 4, 5, 6]

Answer 2

你的代码被追加两次。 你的第一个追加是在你的 Contextmanager 的语句块中（这一行： print("Changed list: {}".format(t.current())) ）

您的第二个 append 位于__exit__方法中，该方法在 Contextmanager 完成其语句块后调用。

您的代码工作如下：

创建一个并填充 1,2,3
上下文管理器开始工作：
-> 深拷贝 a 到 b
-> 首先通过t.current()追加
-> __exit__结束 -> 在第二个追加的位置调用__exit__ 。

希望它可以帮助您更好地理解它。