扭曲 - 运行 在一个线程中

Question

新 python 扭曲的新手

我的团队希望我在单独的线程中制作一些现有代码 运行。

我想出了一个虚构的例子：

from twisted.internet import threads, reactor
from twisted.internet.defer import inlineCallbacks
from time import sleep


class SomeClass(object):
    def __init__(self):
        self.working = False

    def set_working(self, is_working):
        self.working = is_working
        print 'Flag set to {}'.format(is_working)

    @inlineCallbacks
    def do_worker_thread(self):

        # I want to make this call on the main thread
        self.set_working(True)

        # I want to do all this garbage on a separate thread and keep trucking on the main thread
        # This mimics some calls in the real code. There is a call to deferToThread and a try
        # except block there.
        def thread_proc():
            try:
                for i in range(0, 100):
                    print 'Step %d starting'.format(i)
                    self.execute_step(i)
            except Exception:
                print 'An exception happened'
        reactor.callInThread(thread_proc)

        # When the worker thread is done, I want to call back 'on_thread_work_done'

    def execute_step(self, num):
        sleep(num)
        print 'Worker thread: %d'.format(num)

    def on_thread_work_done(self):
        """I want to be called back when the worker thread is done"""
        self.set_working(False)

    @inlineCallbacks
    def do_main_thread(self):
        for c in range(ord('a'), ord('z')+1):
            sleep(c)
            print 'Main thread: {}'.format(c)


if __name__ == "__main__":
    someClass = SomeClass()
    result = someClass.do_worker_thread()
    result.addCallback(someClass.do_main_thread())

    reactor.run()

do_worker_thread 上的内容当前 运行 在主线程上。我在那里发表了评论，我希望它在一个单独的线程中 运行。重要的是 do_worker_thread returns 立即。

我希望输出类似于：

Flag set to True
Step 0 starting
Main thread: a
Worker thread: 0
Worker thread: 1
Main thread: b
Worker thread: 2
Main thread: c
...

如何更改 do_worker_thread 中的内容，以便我的 set_working 调用在主线程上进行，并且在工作线程完成其工作之前不会将其设置为 False？

Answer 1

您可以使用 callFromThread or blockingCallFromThread。

Answer 2

尝试一下。它使用 callFromThread() 来安排主线程上的工作。

from twisted.internet import threads, reactor
from twisted.internet.defer import inlineCallbacks, returnValue
from time import sleep


class SomeClass(object):
    def __init__(self):
        self.working = False

    def set_working(self, is_working):
        self.working = is_working
        print 'Flag set to {}'.format(is_working)

    @inlineCallbacks
    def do_worker_thread(self):

        # I want to make this call on the main thread
        self.set_working(True)

        # I want to do all this garbage on a separate thread and keep trucking on the main thread
        # This mimics some calls in the real code. There is a call to deferToThread and a try
        # except block there.
        def thread_proc():
            try:
                for i in xrange(0, 10):
                    print 'Step {} starting'.format(i)
                    self.execute_step(i)
            except Exception:
                print 'An exception happened'

        yield threads.deferToThread(thread_proc)

        # When the worker thread is done, I want to call back 'on_thread_work_done'
        self.on_thread_work_done()

        returnValue(17)

    def execute_step(self, num):
        sleep(1)
        print 'Worker thread: {}'.format(num)

    def on_thread_work_done(self):
        """I want to be called back when the worker thread is done"""
        self.set_working(False)

    def do_main_thread(self):
        for i in [chr(x) for x in range(ord('a'), ord('z')+1)]:
            print 'Main thread: {}'.format(i)
            sleep(1)

    def thread_done(self, result):
        print 'Thread done: {}'.format(result)

if __name__ == "__main__":
    someClass = SomeClass()
    # Schedule the threaded work
    result = someClass.do_worker_thread().addCallback(someClass.thread_done)
    # Schedule the main thread work
    reactor.callFromThread(someClass.do_main_thread)
    reactor.run()