A Python robot that edits Wikipedia and interacts with people over IRC https://en.wikipedia.org/wiki/User:EarwigBot
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

144 lines
5.8 KiB

  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2009-2012 Ben Kurtovic <ben.kurtovic@verizon.net>
  4. #
  5. # Permission is hereby granted, free of charge, to any person obtaining a copy
  6. # of this software and associated documentation files (the "Software"), to deal
  7. # in the Software without restriction, including without limitation the rights
  8. # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. # copies of the Software, and to permit persons to whom the Software is
  10. # furnished to do so, subject to the following conditions:
  11. #
  12. # The above copyright notice and this permission notice shall be included in
  13. # all copies or substantial portions of the Software.
  14. #
  15. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21. # SOFTWARE.
  22. import threading
  23. import re
  24. from earwigbot.commands import Command
  25. class Threads(Command):
  26. """Manage wiki tasks from IRC, and check on thread status."""
  27. name = "threads"
  28. commands = ["tasks", "task", "threads", "tasklist"]
  29. def process(self, data):
  30. self.data = data
  31. if not self.config.irc["permissions"].is_owner(data):
  32. msg = "You must be a bot owner to use this command."
  33. self.reply(data, msg)
  34. return
  35. if not data.args:
  36. if data.command == "tasklist":
  37. self.do_list()
  38. else:
  39. msg = "No arguments provided. Maybe you wanted '!{0} list', '!{0} start', or '!{0} listall'?"
  40. self.reply(data, msg.format(data.command))
  41. return
  42. if data.args[0] == "list":
  43. self.do_list()
  44. elif data.args[0] == "start":
  45. self.do_start()
  46. elif data.args[0] in ["listall", "all"]:
  47. self.do_listall()
  48. else: # They asked us to do something we don't know
  49. msg = "Unknown argument: \x0303{0}\x0F.".format(data.args[0])
  50. self.reply(data, msg)
  51. def do_list(self):
  52. """With !tasks list (or abbreviation !tasklist), list all running
  53. threads. This includes the main threads, like the irc frontend and the
  54. watcher, and task threads."""
  55. threads = threading.enumerate()
  56. normal_threads = []
  57. daemon_threads = []
  58. for thread in threads:
  59. tname = thread.name
  60. if tname == "MainThread":
  61. t = "\x0302MainThread\x0F (id {0})"
  62. normal_threads.append(t.format(thread.ident))
  63. elif tname in self.config.components:
  64. t = "\x0302{0}\x0F (id {1})"
  65. normal_threads.append(t.format(tname, thread.ident))
  66. elif tname.startswith("reminder"):
  67. tname = tname.replace("reminder ", "")
  68. t = "\x0302reminder\x0F (until {0})"
  69. normal_threads.append(t.format(tname))
  70. else:
  71. tname, start_time = re.findall("^(.*?) \((.*?)\)$", tname)[0]
  72. t = "\x0302{0}\x0F (id {1}, since {2})"
  73. daemon_threads.append(t.format(tname, thread.ident,
  74. start_time))
  75. if daemon_threads:
  76. if len(daemon_threads) > 1:
  77. msg = "\x02{0}\x0F threads active: {1}, and \x02{2}\x0F command/task threads: {3}."
  78. else:
  79. msg = "\x02{0}\x0F threads active: {1}, and \x02{2}\x0F command/task thread: {3}."
  80. msg = msg.format(len(threads), ', '.join(normal_threads),
  81. len(daemon_threads), ', '.join(daemon_threads))
  82. else:
  83. msg = "\x02{0}\x0F threads active: {1}, and \x020\x0F command/task threads."
  84. msg = msg.format(len(threads), ', '.join(normal_threads))
  85. self.reply(self.data, msg)
  86. def do_listall(self):
  87. """With !tasks listall or !tasks all, list all loaded tasks, and report
  88. whether they are currently running or idle."""
  89. threads = threading.enumerate()
  90. tasklist = []
  91. for task in sorted([task.name for task in self.bot.tasks]):
  92. threadlist = [t for t in threads if t.name.startswith(task)]
  93. ids = [str(t.ident) for t in threadlist]
  94. if not ids:
  95. tasklist.append("\x0302{0}\x0F (idle)".format(task))
  96. elif len(ids) == 1:
  97. t = "\x0302{0}\x0F (\x02active\x0F as id {1})"
  98. tasklist.append(t.format(task, ids[0]))
  99. else:
  100. t = "\x0302{0}\x0F (\x02active\x0F as ids {1})"
  101. tasklist.append(t.format(task, ', '.join(ids)))
  102. tasks = ", ".join(tasklist)
  103. msg = "\x02{0}\x0F tasks loaded: {1}.".format(len(tasklist), tasks)
  104. self.reply(self.data, msg)
  105. def do_start(self):
  106. """With !tasks start, start any loaded task by name with or without
  107. kwargs."""
  108. data = self.data
  109. try:
  110. task_name = data.args[1]
  111. except IndexError: # No task name given
  112. self.reply(data, "What task do you want me to start?")
  113. return
  114. if task_name not in [task.name for task in self.bot.tasks]:
  115. # This task does not exist or hasn't been loaded:
  116. msg = "Task could not be found; either it doesn't exist, or it wasn't loaded correctly."
  117. self.reply(data, msg.format(task_name))
  118. return
  119. data.kwargs["fromIRC"] = True
  120. self.bot.tasks.start(task_name, **data.kwargs)
  121. msg = "Task \x0302{0}\x0F started.".format(task_name)
  122. self.reply(data, msg)