mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-02 11:58:21 +08:00
111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf8 -*-
|
|
# cp936
|
|
#
|
|
# The MIT License (MIT)
|
|
#
|
|
# Copyright (c) 2021 hikyuu.org
|
|
#
|
|
# Create on: 2021-03-09
|
|
# Author: fasiondog
|
|
|
|
import cmd
|
|
import functools
|
|
|
|
|
|
def shell_cmd(func):
|
|
@functools.wraps(func)
|
|
def wrapcmd(*args, **kargs):
|
|
try:
|
|
self = args[0]
|
|
self.__class__.lineno += 1
|
|
self.__class__.prompt = '\033[32;40mHKU [\033[0m\033[1;32;40m%s\033[0m\033[32;40m]:\033[0m ' % self.__class__.lineno
|
|
result = func(*args, **kargs)
|
|
print("")
|
|
return result
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
return wrapcmd
|
|
|
|
|
|
class HKUShell(cmd.Cmd):
|
|
intro = """\033[1;32;40m
|
|
ooooo ooooo ooooo oooo oooo oooooo oooo ooooo ooo ooooo ooo
|
|
`888' `888' `888' `888 .8P' `888. .8' `888' `8' `888' `8'
|
|
888 888 888 888 d8' `888. .8' 888 8 888 8
|
|
888ooooo888 888 88888[ `888.8' 888 8 888 8
|
|
888 888 888 888`88b. `888' 888 8 888 8
|
|
888 888 888 888 `88b. 888 `88. .8' `88. .8'
|
|
o888o o888o o888o o888o o888o o888o `YbodP' `YbodP'
|
|
\033[0m
|
|
\033[33m\033[1m Welcome to the hikyuu shell. Type help or ? to list commands.\033[0m
|
|
|
|
"""
|
|
lineno = 1
|
|
prompt = '\033[32;40mHKU [\033[0m\033[1;32;40m%s\033[0m\033[32;40m]:\033[0m ' % lineno
|
|
file = None
|
|
server = None
|
|
|
|
def __del__(self):
|
|
if HKUShell.server is not None:
|
|
HKUShell.server.terminate()
|
|
HKUShell.server.wait()
|
|
HKUShell.server = None
|
|
|
|
@classmethod
|
|
def add_command(cls, func):
|
|
"""
|
|
def func(self, args):
|
|
do something, don't return
|
|
"""
|
|
@functools.wraps(func)
|
|
def wrapcmd(*args, **kargs):
|
|
try:
|
|
self = args[0]
|
|
self.__class__.lineno += 1
|
|
self.__class__.prompt = '\033[32;40mHKU [\033[0m\033[1;32;40m%s\033[0m\033[32;40m]:\033[0m ' % self.__class__.lineno
|
|
result = func(*args, **kargs)
|
|
print("")
|
|
return result
|
|
except Exception as e:
|
|
print(e)
|
|
print("")
|
|
|
|
if func.__name__ == "do_":
|
|
raise Exception("Invalid func name!")
|
|
name = func.__name__ if len(func.__name__) > 3 and func.__name__[:3] == "do_" else "do_%s" % func.__name__
|
|
setattr(cls, name, wrapcmd)
|
|
|
|
@shell_cmd
|
|
def do_hello(self, arg):
|
|
"""You are welcome"""
|
|
print(arg)
|
|
|
|
def do_quit(self, arg):
|
|
'Stop recording, and exit: BYE'
|
|
print('Thank you for using hikyuu.\n')
|
|
return True
|
|
|
|
# ----- record and playback -----
|
|
def do_record(self, arg):
|
|
'Save future commands to filename: RECORD rose.cmd'
|
|
self.file = open(arg, 'w')
|
|
|
|
def do_playback(self, arg):
|
|
'Playback commands from a file: PLAYBACK rose.cmd'
|
|
self.close()
|
|
with open(arg) as f:
|
|
self.cmdqueue.extend(f.read().splitlines())
|
|
|
|
def precmd(self, line):
|
|
line = line.lower()
|
|
if self.file and 'playback' not in line:
|
|
print(line, file=self.file)
|
|
return line
|
|
|
|
def close(self):
|
|
if self.file:
|
|
self.file.close()
|
|
self.file = None
|