1 files changed, 5 insertions, 18 deletions
diff --git a/ATRI/__init__.py b/ATRI/__init__.py
index 4458ad8..e5bc2f4 100644
--- a/ATRI/__init__.py
+++ b/ATRI/__init__.py
@@ -1,19 +1,3 @@
-#!/usr/bin/env python3
-# -*- coding:utf-8 -*-
-'''
-File: __init__.py
-Created Date: 2021-03-07 11:22:06
-Author: Kyomotoi
-Email: [email protected]
-License: GPLv3
-Project: https://github.com/Kyomotoi/ATRI
---------
-Last Modified: Sunday, 7th March 2021 3:01:51 pm
-Modified By: Kyomotoi ([email protected])
---------
-Copyright (c) 2021 Kyomotoi
-'''
-
 from time import sleep
 
 import nonebot as nb
@@ -30,10 +14,13 @@ def asgi():
     return nb.get_asgi()
 
 
+def driver():
+    return nb.get_driver()
+
+
 def init():
     nb.init(**RUNTIME_CONFIG)
-    driver = nb.get_driver()
-    driver.register_adapter("cqhttp", ATRIBot)
+    driver().register_adapter("cqhttp", ATRIBot)
     nb.load_plugins('ATRI/plugins')
     logger.info(f"Now running: {__version__}")
     sleep(3)