ircbotpoll.py

ircbotpoll.py

@@ -0,0 +1,150 @@
+#!/usr/bin/env python3
+
+################################################################
+# IRC Bot for creating polls                                   #
+#                                                              #
+# Created by q3aql (q3aql@duck.com)                            #
+# Licensed by GPL v2.0                                         #
+# Last update: 27-08-2025                                      #
+#                                                              #
+# Requirements:                                                #
+#    pip install irc                                           #
+################################################################
+
+import sys
+import time
+import signal
+import logging
+from collections import defaultdict
+from threading import Event, Thread
+
+import irc.client
+
+######## CONFIGURATION (Edit with your settings)
+SERVER   = "irc.example.net"
+PORT     = 6667
+USE_TLS  = False
+NICK     = "poll-bot"
+REALNAME = "IRC Poll Bot"
+CHANNELS = ["#support", "#linux"]
+#########
+
+polls = {}
+
+def on_connect(conn, event):
+    logging.info("Connected to %s:%s", SERVER, PORT)
+    for chan in CHANNELS:
+        conn.join(chan)
+        logging.info("Joining %s", chan)
+
+def on_join(conn, event):
+    nick = irc.client.NickMask(event.source).nick
+    channel = event.target
+
+    if nick == NICK:
+        return
+
+    logging.info("%s has joined %s", nick, channel)
+
+def on_pubmsg(conn, event):
+    channel = event.target
+    nick = irc.client.NickMask(event.source).nick
+    message = event.arguments[0]
+
+    if message.startswith("!poll"):
+        create_poll(conn, channel, message)
+    elif message.startswith("!vote"):
+        cast_vote(conn, channel, nick, message)
+    elif message.startswith("!results"):
+        show_results(conn, channel)
+    elif message.startswith("!endpoll"):
+        end_poll(conn, channel)
+
+def create_poll(conn, channel, message):
+    parts = message.split(" ", 1)
+    if len(parts) < 2:
+        conn.privmsg(channel, "Usage: !poll <question>")
+        return
+
+    question = parts[1]
+    if channel in polls:
+        conn.privmsg(channel, "A poll is already active. Please end it before starting a new one.")
+        return
+
+    polls[channel] = {"question": question, "votes": defaultdict(int)}
+    conn.privmsg(channel, f"Poll created: {question} - Use !vote <option> to vote.")
+
+def cast_vote(conn, channel, nick, message):
+    if channel not in polls:
+        conn.privmsg(channel, "No active poll in this channel.")
+        return
+
+    parts = message.split(" ", 1)
+    if len(parts) < 2:
+        conn.privmsg(channel, "Usage: !vote <option>")
+        return
+
+    option = parts[1]
+    polls[channel]["votes"][option] += 1
+    conn.privmsg(channel, f"{nick} voted for: {option}")
+
+def show_results(conn, channel):
+    if channel not in polls:
+        conn.privmsg(channel, "No active poll in this channel.")
+        return
+
+    poll = polls[channel]
+    results = ", ".join(f"{option}: {count}" for option, count in poll["votes"].items())
+    conn.privmsg(channel, f"Results for '{poll['question']}': {results}")
+
+def end_poll(conn, channel):
+    if channel not in polls:
+        conn.privmsg(channel, "No active poll in this channel.")
+        return
+
+    results = show_results(conn, channel)
+    conn.privmsg(channel, f"Poll ended: {polls[channel]['question']} - {results}")
+    del polls[channel]
+
+def on_disconnect(conn, event):
+    logging.warning("Disconnected from the server – reconnecting in 10s...")
+    time.sleep(10)
+    connect_and_start()
+
+def on_error(conn, event):
+    logging.error("ERROR from server: %s", event.arguments)
+
+def connect_and_start():
+    reactor = irc.client.Reactor()
+
+    try:
+        if USE_TLS:
+            conn = reactor.server().connect_ssl(
+                SERVER, PORT, NICK, password=None, ssl_verify=False
+            )
+        else:
+            conn = reactor.server().connect(SERVER, PORT, NICK, password=None)
+    except irc.client.ServerConnectionError as e:
+        logging.error("Unable to connect: %s", e)
+        logging.warning("Retrying connection in 10s...")
+        time.sleep(10)
+        connect_and_start()
+
+    conn.add_global_handler("welcome", on_connect)   # 001
+    conn.add_global_handler("join", on_join)
+    conn.add_global_handler("pubmsg", on_pubmsg)
+    conn.add_global_handler("disconnect", on_disconnect)
+    conn.add_global_handler("error", on_error)
+    reactor.process_forever()
+
+if __name__ == "__main__":
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s %(levelname)-8s %(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+    )
+
+    try:
+        connect_and_start()
+    finally:
+        pass