yuuki/libs/thread.py

# -*- coding: utf-8 -*-
"""
Yuuki_Libs
(c) 2020 Star Inc.
This Source Code Form is subject to the terms of the Mozilla Public
License, v. 2.0. If a copy of the MPL was not distributed with this
file, You can obtain one at http://mozilla.org/MPL/2.0/.
"""
import multiprocessing
import threading


class Thread:
    def __init__(self):
        self.lock = threading.Lock()

    @staticmethod
    def add(function, args=()):
        added_thread = threading.Thread(name=function.__name__, target=function, args=args)
        added_thread.start()

    @staticmethod
    def get_thread_info():
        print(threading.active_count())
        print(threading.enumerate())
        print(f"{threading.current_thread()} add Threading\n")


class Multiprocess:
    multiprocess_list = {}

    def add(self, function, args=()):
        added_multiprocess = multiprocessing.Process(name=function.__name__, target=function, args=args)
        self.multiprocess_list[function.__name__] = added_multiprocess
        added_multiprocess.start()

    def stop(self, function_name):
        assert function_name in self.multiprocess_list
        self.multiprocess_list[function_name].terminate()