-
-
Notifications
You must be signed in to change notification settings - Fork 36
/
console.py
92 lines (77 loc) · 3.45 KB
/
console.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import asyncio
import os
import sys
import traceback
from prompt_toolkit import PromptSession
from prompt_toolkit.history import FileHistory
from core.config import Config
from core.logger import Logger
from core.path import assets_path
from core.utils.info import Info
if not Config('db_path', cfg_type=str):
raise AttributeError('Wait! You need to fill a valid database address into the config.toml "db_path" field\n'
'Example: \ndb_path = "sqlite:///database/save.db"\n'
'(Also you can fill in the above example directly,'
' bot will automatically create a SQLite database in the "./database/save.db")')
from bot import init_bot
from core.bot import init_async
from core.builtins import PrivateAssets, EnableDirtyWordCheck, Url
from core.console.info import *
from core.console.message import MessageSession
from core.extra.scheduler import load_extra_schedulers
from core.parser.message import parser
from core.types import MsgInfo, Session
from core.database import BotDBUtil, session
from core.database.tables import DBVersion
query_dbver = session.query(DBVersion).first()
if not query_dbver:
session.add_all([DBVersion(value=str(BotDBUtil.database_version))])
session.commit()
query_dbver = session.query(DBVersion).first()
if (current_ver := int(query_dbver.value)) < (target_ver := BotDBUtil.database_version):
print(f'Updating database from {current_ver} to {target_ver}...')
from core.database.update import update_database
update_database()
print('Database updated successfully! Please restart the program.')
sys.exit()
EnableDirtyWordCheck.status = True
Url.disable_mm = True
PrivateAssets.set(os.path.join(assets_path, 'private', 'console'))
console_history_path = os.path.join(PrivateAssets.path, '.console_history')
if os.path.exists(console_history_path):
os.remove(console_history_path)
async def console_scheduler():
load_extra_schedulers()
await init_async()
async def console_command():
try:
session = PromptSession(history=FileHistory(console_history_path))
while True:
m = await asyncio.to_thread(session.prompt, '> ')
await send_command(m)
except Exception:
Logger.error(traceback.format_exc())
async def send_command(msg):
Logger.info('-------Start-------')
returns = await parser(MessageSession(target=MsgInfo(target_id=f'{target_name}|0',
sender_id=f'{sender_name}|0',
sender_name='Console',
target_from=target_name,
sender_from=sender_name,
client_name=client_name,
message_id=0,
reply_id=None),
session=Session(message=msg, target=f'{target_name}|0', sender=f'{sender_name}|0')))
Logger.info('----Process end----')
return returns
if __name__ == '__main__':
init_bot()
Info.client_name = client_name
loop = asyncio.new_event_loop()
loop.run_until_complete(console_scheduler())
try:
loop.run_until_complete(console_command())
except (KeyboardInterrupt, SystemExit):
print('Exited.')
finally:
loop.close()