gandc/irc_pc_info_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improvements

Browse Source
This commit is contained in:
GandC Snow 2021-12-08 00:28:29 +03:00
parent 994a5425da
commit 41fe31f477
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bot.py
View File

@ -8,7 +8,7 @@ nickname = argv[2]
server = argv[3] server = argv[3]
if len(argv) !=4: if len(argv) !=4:
print('Usage: {} channel nickname server'.format(argv[0])) print('Usage: {} channel nickname server'.format(argv[0]))
def get_size(bytes, suffix="B"): def get_size(bytes, suffix="B"):
factor = 1024 factor = 1024
for unit in ["", "K", "M", "G", "T", "P"]: for unit in ["", "K", "M", "G", "T", "P"]:
@ -21,6 +21,10 @@ class TestBot(irc.bot.SingleServerIRCBot):
def __init__(self, channel, nickname, server, port=6667): def __init__(self, channel, nickname, server, port=6667):
irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], nickname, nickname) irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], nickname, nickname)
self.channel = channel self.channel = channel
self.connection.add_global_handler('disconnect', self.on_disconnect)
def on_disconnect(self, connection, event):
exit(1)
def on_welcome(self, c, e): def on_welcome(self, c, e):
c.mode(self._nickname, '+B') c.mode(self._nickname, '+B')
@ -38,7 +42,7 @@ class TestBot(irc.bot.SingleServerIRCBot):
return return
def do_command(self, e, cmd): def do_command(self, e, cmd):
#nick = e.source.nick #nick = e.source.nick ## Uncomment when need to send to user not channel
c = self.connection c = self.connection
if cmd.lower() == "uptime": if cmd.lower() == "uptime":
result = subprocess.run(["uptime"], stderr=subprocess.PIPE, stdout=subprocess.PIPE) result = subprocess.run(["uptime"], stderr=subprocess.PIPE, stdout=subprocess.PIPE)
@ -46,8 +50,6 @@ class TestBot(irc.bot.SingleServerIRCBot):
c.privmsg(channel, output) c.privmsg(channel, output)
elif cmd.lower() == "uname": elif cmd.lower() == "uname":
# >>> print(f"Version: {uname.version}")
# Version: #1 SMP Debian 4.19.208-1 (2021-09-29)
uname = platform.uname() uname = platform.uname()
end = 'System name:' + uname.version end = 'System name:' + uname.version
c.privmsg(channel, end) c.privmsg(channel, end)
@ -66,7 +68,6 @@ class TestBot(irc.bot.SingleServerIRCBot):
elif cmd.lower() == "cpu": elif cmd.lower() == "cpu":
percent = psutil.cpu_percent() percent = psutil.cpu_percent()
end = 'CPU load: {}%'.format(percent) end = 'CPU load: {}%'.format(percent)
#c.privmsg(channel, end)
c.privmsg(channel, end) c.privmsg(channel, end)
elif cmd.lower() == "network": elif cmd.lower() == "network":
@ -94,6 +95,8 @@ class TestBot(irc.bot.SingleServerIRCBot):
end = 'Available commands are: help, uptime, ram, cpu, uname, network, temp' end = 'Available commands are: help, uptime, ram, cpu, uname, network, temp'
c.privmsg(channel, end) c.privmsg(channel, end)
elif cmd.lower() == "stop":
exit(1)
else: else:
end = 'Unknow command. Sorry' end = 'Unknow command. Sorry'
c.privmsg(channel, end) c.privmsg(channel, end)