Onionr/src/communicator/daemoneventhooks/__init__.py

80 lines
2.6 KiB
Python
Raw Normal View History

2020-01-02 02:07:34 +00:00
"""Onionr - Private P2P Communication.
Hooks to handle daemon events
"""
from threading import Thread
from .removefrominsertqueue import remove_from_insert_queue
2020-01-02 02:07:34 +00:00
from typing import TYPE_CHECKING
from gevent import sleep
2020-01-04 12:13:10 +00:00
from communicatorutils.uploadblocks import mixmate
from communicatorutils import restarttor
2020-01-04 12:13:10 +00:00
2020-01-02 02:07:34 +00:00
if TYPE_CHECKING:
from toomanyobjs import TooMany
from communicator import OnionrCommunicatorDaemon
from httpapi.daemoneventsapi import DaemonEventsBP
from onionrtypes import BlockHash
2020-01-04 12:13:10 +00:00
from apiservers import PublicAPI
2020-01-02 02:07:34 +00:00
"""
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
def daemon_event_handlers(shared_state: 'TooMany'):
def _get_inst(class_name: str):
while True:
try:
return shared_state.get_by_string(class_name)
except KeyError:
sleep(0.2)
comm_inst = _get_inst('OnionrCommunicatorDaemon')
2020-01-04 12:13:10 +00:00
public_api: 'PublicAPI' = _get_inst('PublicAPI')
2020-01-02 02:07:34 +00:00
events_api: 'DaemonEventsBP' = _get_inst('DaemonEventsBP')
def remove_from_insert_queue_wrapper(block_hash: 'BlockHash'):
remove_from_insert_queue(comm_inst, block_hash)
2020-01-04 12:13:10 +00:00
return "removed"
def print_test(text=''):
print("It works!", text)
return f"It works! {text}"
2020-01-04 12:13:10 +00:00
def upload_event(block: 'BlockHash' = ''):
if not block:
raise ValueError
public_api.hideBlocks.append(block)
try:
mixmate.block_mixer(comm_inst.blocksToUpload, block)
except ValueError:
pass
return "removed"
def restart_tor():
restarttor.restart(comm_inst)
comm_inst.offlinePeers = []
def test_runtime():
Thread(target=comm_inst.shared_state.get_by_string(
"OnionrRunTestManager").run_tests).start()
events_api.register_listener(remove_from_insert_queue_wrapper)
events_api.register_listener(print_test)
2020-01-04 12:13:10 +00:00
events_api.register_listener(upload_event)
events_api.register_listener(test_runtime)
2020-01-02 02:07:34 +00:00