class MikrotikCommandLoop(cmd.Cmd):
m = None
prompt = '> '
def precmd(self, line):
if self.m and line.startswith("/"):
line = "run %s" % line
return cmd.Cmd.precmd(self, line)
def do_greet(self, line):
print("Welcome to use Mikrotik remote command line tool")
def do_EOF(self, line):
print("")
return True
def do_login(self, line):
"""
Connect to device
Usage: address user password
"""
args = line.split()
if len(args) < 1:
print("Usage: login address [user] [password]")
return
if len(args) < 2:
user = input("Username: "******"Password: "******"Password: "******"""
Run command on remote Mikrotik.
"""
if not self.m:
print("login first!")
return
args = line.split()
if len(args) < 1:
print("Usage: run [args]")
return
command = args[0]
arguments = {}
queries = {}
argument_dict = arguments
expect_number = False
for i in args[1:]:
if i.lower() == 'where':
argument_dict = queries
continue
if i.lower() in ['print', 'add']:
command = "%s/%s" % (command, i.lower())
continue
if i.lower() in ['set', 'remove']:
command = "%s/%s" % (command, i.lower())
expect_number = True
continue
if i.isdigit() and expect_number:
argument_dict['.id'] = '*%s' % i
expect_number=False
continue
expect_number = False
try:
(a, b) = i.split("=",1)
argument_dict[a] = b
except ValueError:
print("Invalid argument %s" % i)
return
response = self.m.run(command, attributes=arguments, queries=queries)
for r in response:
if r.status == "done":
continue
attributes = ' '.join(["%s%s%s=%s%s%s" % (colors.BLUE, k, colors.ENDC, colors.GREEN, v, colors.ENDC) for k, v in r.attributes.items()])
if '.id' in r.attributes:
_id = r.attributes['.id'].lstrip("*")
print("!%s%s%s %s%s%s %s %s" % (colors.BOLD, colors.MAGENTA, r.status, colors.RED, _id, colors.ENDC, attributes, ' '.join(r.error)))
else:
print("!%s%s%s%s %s %s" % (colors.BOLD, colors.MAGENTA, r.status, colors.ENDC, attributes, ' '.join(r.error)))
def do_logout(self, line):
if self.m:
self.m.disconnect()
self.prompt = "> "
else:
self.prompt = "> "
