bitbot-3.11-fork/modules/stats.py

158 lines
5.6 KiB
Python
Raw Normal View History

#--depends-on commands
2016-05-09 15:14:12 +00:00
import time
from src import ModuleManager, utils
2016-05-09 15:14:12 +00:00
HIDDEN_MODES = set(["s", "p"])
class Module(ModuleManager.BaseModule):
def _uptime(self):
return utils.to_pretty_time(int(time.time()-self.bot.start_time))
@utils.hook("received.command.uptime")
2016-05-09 15:14:12 +00:00
def uptime(self, event):
"""
:help: Show my uptime
"""
event["stdout"].write("Uptime: %s" % self._uptime())
@utils.hook("api.get.uptime")
def uptime_api(self, event):
return self._uptime()
2016-05-09 15:14:12 +00:00
def _stats(self):
2019-09-07 13:13:44 +00:00
networks = {}
2016-05-09 15:14:12 +00:00
for server in self.bot.servers.values():
2019-09-07 13:13:44 +00:00
networks[server.alias.lower()] = [len(server.channels),
len(server.users)]
return networks
def _plural(self, s, count):
return "%s%s" % (s, "" if count == 1 else "s")
2016-05-09 15:14:12 +00:00
@utils.hook("received.command.stats")
2019-09-07 13:13:44 +00:00
@utils.kwarg("help", "Show my network/channel/user stats")
@utils.kwarg("usage", "[network]")
def stats(self, event):
2019-09-07 13:13:44 +00:00
networks = self._stats()
if event["args"]:
alias = event["args_split"][0].lower()
if not alias in networks:
raise utils.EventError("Unknown alias '%s'" % alias)
2016-05-09 15:14:12 +00:00
2019-09-07 13:13:44 +00:00
channels, users = networks[alias]
2019-09-07 13:16:27 +00:00
event["stdout"].write("on %s I have %d %s and %d visible %s" %
2019-09-07 13:13:44 +00:00
(alias, channels, self._plural("channel", channels), users,
self._plural("user", users)))
else:
total_channels = 0
total_users = 0
for channels, users in networks.values():
total_channels += channels
total_users += users
2016-05-09 15:14:12 +00:00
2019-09-07 13:13:44 +00:00
network_count = len(networks.keys())
network_plural = self._plural("network", network_count)
channel_plural = self._plural("channel", total_channels)
user_plural = self._plural("user", total_users)
event["stdout"].write(
"I currently have %d %s, %d %s and %d visible %s" %
(network_count, network_plural, total_channels, channel_plural,
total_users, user_plural))
@utils.hook("api.get.stats")
def stats_api(self, event):
networks, channels, users = self._stats()
return {"networks": networks, "channels": channels, "users": users}
2018-10-04 16:10:15 +00:00
def _server_stats(self, server):
return {
"hostname": server.connection_params.hostname,
"port": server.connection_params.port,
"tls": server.connection_params.tls,
"alias": server.connection_params.alias,
"hostmask": "%s!%s@%s" % (
server.nickname, server.username, server.hostname),
"users": len(server.users),
"bytes-written": server.socket.bytes_written,
"bytes-read": server.socket.bytes_read,
"connected-since": server.socket.connect_time,
"channels": {
c.name: self._channel_stats(c) for c in server.channels
},
"capabilities": list(server.agreed_capabilities),
"version": server.version
}
@utils.hook("api.get.servers")
2018-10-04 16:18:40 +00:00
def servers_api(self, event):
if event["path"]:
server_id = event["path"][0]
if not server_id.isdigit():
return None
server_id = int(server_id)
server = self.bot.get_server_by_id(server_id)
if not server:
return None
return self._server_stats(server)
else:
servers = {}
for server in self.bot.servers.values():
servers[server.id] = self._server_stats(server)
return servers
2018-10-04 16:18:40 +00:00
def _channel_stats(self, channel):
return {
"users": sorted([user.nickname for user in channel.users],
key=lambda nickname: nickname.lower()),
"topic": channel.topic,
"topic-set-at": channel.topic_time,
"topic-set-by": channel.topic_setter.nickname,
"modes": channel.mode_str()
}
@utils.hook("api.get.channels")
2018-10-04 16:10:15 +00:00
def channels_api(self, event):
2018-10-04 16:21:40 +00:00
if event["path"]:
server_id = event["path"][0]
2018-10-04 16:10:15 +00:00
if not server_id.isdigit():
return None
server_id = int(server_id)
server = self.bot.get_server_by_id(server_id)
2018-10-04 16:10:15 +00:00
if not server:
return None
channels = {}
for channel in server.channels.values():
channels[channel.name] = self._channel_stats(channel)
2018-10-04 16:10:15 +00:00
return channels
else:
channels = {}
for server in self.bot.servers.values():
channels[server.id] = {}
for channel in server.channels.values():
channels[server.id][str(channel)] = self._channel_stats(
channel)
2018-10-04 16:10:15 +00:00
return channels
2018-10-04 16:37:47 +00:00
2019-09-16 15:05:36 +00:00
@utils.hook("received.command.channels")
@utils.kwarg("help", "List all the channel I'm in on this network")
@utils.kwarg("permission", "listchannels")
def channels_command(self, event):
channels = []
for channel in event["server"].channels.values():
hidden = bool(HIDDEN_MODES&set(channel.modes.keys()))
if hidden and (
event["is_channel"] and not channel == event["target"]):
continue
channels.append(channel.name)
2019-09-16 15:05:36 +00:00
event["stdout"].write("Current channels: %s" %
" ".join(sorted(channels)))
2019-09-16 15:05:36 +00:00
@utils.hook("api.get.modules")
2018-10-04 16:37:47 +00:00
def modules_api(self, event):
return list(self.bot.modules.modules.keys())