move subscribe_event into api_base

rqalpha/__init__.py

@@ -21,7 +21,7 @@
 import pkgutil
 
 from rqalpha.__main__ import cli
-from rqalpha.api.api_base import export_as_api
+from rqalpha.api.api_base import export_as_api, subscribe_event
 
 __all__ = [
     '__version__',
@@ -141,15 +141,3 @@ def run_func(**kwargs):
     config = parse_config(config, user_funcs=user_funcs)
     clear_all_cached_functions()
     return main.run(config, user_funcs=user_funcs)
-
-
-def subscribe_event(event_type, handler):
-    import types
-    from rqalpha.events import EVENT
-    from rqalpha.environment import Environment
-
-    assert isinstance(handler, types.FunctionType)
-    assert isinstance(event_type, EVENT)
-
-    env = Environment.get_instance()
-    env.event_bus.add_listener(event_type, handler)

rqalpha/api/api_base.py

@@ -19,6 +19,7 @@
 import datetime
 import inspect
 import sys
+import types
 from collections import Iterable
 from functools import wraps
 from types import FunctionType
@@ -928,3 +929,13 @@ def get_position(order_book_id, direction):
         raise RuntimeError(_("Booking has not been set, please check your broker configuration."))
 
     return booking.get_position(order_book_id, direction)
+
+
+@export_as_api
+@apply_rules(
+    verify_that('event_type').is_instance_of(EVENT),
+    verify_that('handler').is_instance_of(types.FunctionType)
+)
+def subscribe_event(event_type, handler):
+    env = Environment.get_instance()
+    env.event_bus.add_listener(event_type, handler)

tests/test_s_tick_size.py

@@ -1,13 +1,11 @@
-from rqalpha import subscribe_event, events
-
 SLIPPAGE = 10
 price = 0
 stock = "000001.XSHE"
 
 
 def init(context):
     context.count = 0
-    subscribe_event(events.EVENT.TRADE, on_trade)
+    subscribe_event(EVENT.TRADE, on_trade)
 
 
 def on_trade(event):