Add initial Pelit plugin

Pelit/README.md

@@ -0,0 +1 @@
+Find text from Wiki and highlight people

Pelit/__init__.py

@@ -0,0 +1,68 @@
+###
+# Copyright (c) 2016, Toni Fadjukoff
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions, and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions, and the following disclaimer in the
+#     documentation and/or other materials provided with the distribution.
+#   * Neither the name of the author of this software nor the name of
+#     contributors to this software may be used to endorse or promote products
+#     derived from this software without specific prior written consent.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+# POSSIBILITY OF SUCH DAMAGE.
+
+###
+
+"""
+Pelit: Find text from Wiki and highlight people
+"""
+
+import supybot
+import supybot.world as world
+
+# Use this for the version of this plugin.  You may wish to put a CVS keyword
+# in here if you're keeping the plugin in CVS or some similar system.
+__version__ = ""
+
+# XXX Replace this with an appropriate author or supybot.Author instance.
+__author__ = supybot.authors.unknown
+
+# This is a dictionary mapping supybot.Author instances to lists of
+# contributions.
+__contributors__ = {}
+
+# This is a url where the most recent plugin package can be downloaded.
+__url__ = ''
+
+from . import config
+from . import plugin
+from imp import reload
+# In case we're being reloaded.
+reload(config)
+reload(plugin)
+# Add more reloads here if you add third-party modules and want them to be
+# reloaded when this plugin is reloaded.  Don't forget to import them as well!
+
+if world.testing:
+    from . import test
+
+Class = plugin.Class
+configure = config.configure
+
+
+# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:

Pelit/config.py

@@ -0,0 +1,69 @@
+###
+# Copyright (c) 2016, Toni Fadjukoff
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions, and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions, and the following disclaimer in the
+#     documentation and/or other materials provided with the distribution.
+#   * Neither the name of the author of this software nor the name of
+#     contributors to this software may be used to endorse or promote products
+#     derived from this software without specific prior written consent.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+# POSSIBILITY OF SUCH DAMAGE.
+
+###
+
+import supybot.conf as conf
+import supybot.registry as registry
+try:
+    from supybot.i18n import PluginInternationalization
+    _ = PluginInternationalization('Pelit')
+except:
+    # Placeholder that allows to run the plugin on a bot
+    # without the i18n module
+    _ = lambda x: x
+
+
+def configure(advanced):
+    # This will be called by supybot to configure this module.  advanced is
+    # a bool that specifies whether the user identified themself as an advanced
+    # user or not.  You should effect your configuration by manipulating the
+    # registry as appropriate.
+    from supybot.questions import expect, anything, something, yn
+    conf.registerPlugin('Pelit', True)
+
+
+Pelit = conf.registerPlugin('Pelit')
+# This is where your configuration variables (if any) should go.  For example:
+# conf.registerGlobalValue(Pelit, 'someConfigVariableName',
+#     registry.Boolean(False, _("""Help for someConfigVariableName.""")))
+conf.registerGlobalValue(Pelit, 'wikiLogin',
+    registry.String('', _("The username for wiki")))
+conf.registerGlobalValue(Pelit, 'wikiPassword',
+    registry.String('', _("The password for wiki")))
+conf.registerGlobalValue(Pelit, 'wikiApi',
+    registry.String('', _("The URL for mediawiki api.php")))
+conf.registerGlobalValue(Pelit, 'wikiPage',
+    registry.String('', _("The path to mediawiki page to scrap")))
+conf.registerGlobalValue(Pelit, 'httpBasicAuthUser',
+    registry.String('', _("HTTP Basic Authentication username")))
+conf.registerGlobalValue(Pelit, 'httpBasicAuthPass',
+    registry.String('', _("HTTP Basic Authentication password")))
+
+
+# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:

Pelit/local/__init__.py

@@ -0,0 +1 @@
+# Stub so local is a module, used for third-party modules

Pelit/plugin.py

@@ -0,0 +1,136 @@
+###
+# Copyright (c) 2016, Toni Fadjukoff
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions, and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions, and the following disclaimer in the
+#     documentation and/or other materials provided with the distribution.
+#   * Neither the name of the author of this software nor the name of
+#     contributors to this software may be used to endorse or promote products
+#     derived from this software without specific prior written consent.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+# POSSIBILITY OF SUCH DAMAGE.
+
+###
+
+import socket
+socket.setdefaulttimeout(15)
+
+import requests
+import json
+
+import supybot.utils as utils
+from supybot.commands import *
+import supybot.plugins as plugins
+import supybot.ircutils as ircutils
+import supybot.callbacks as callbacks
+try:
+    from supybot.i18n import PluginInternationalization
+    _ = PluginInternationalization('Pelit')
+except ImportError:
+    # Placeholder that allows to run the plugin on a bot
+    # without the i18n module
+    _ = lambda x: x
+
+
+class Pelit(callbacks.Plugin):
+    """Find text from Wiki and highlight people"""
+    pass
+
+    def doLogin(self):
+        wikiApi = self.registryValue('wikiApi')
+        httpAuth = self.createHttpAuth()
+
+        # Fetch token
+        tokenParams = self.createTokenParams()
+        resp = requests.post(wikiApi, auth=httpAuth, params=tokenParams)
+        data = json.loads(resp.text)
+        
+        # Do login
+        token = data["query"]["tokens"]["logintoken"]
+        loginParams = self.createLoginParams(token)
+        resp = requests.post(wikiApi, auth=httpAuth, params=loginParams, cookies=resp.cookies)
+        return resp.cookies
+
+    def fetchPageText(self, cookies):
+        wikiApi = self.registryValue('wikiApi')
+        pageParams = self.createPageParams()
+        resp = requests.get(wikiApi, auth=self.createHttpAuth(), params=pageParams, cookies=cookies)
+        data = json.loads(resp.text)
+        text = data["parse"]["text"]["*"]
+        return text
+
+    def createHttpAuth(self):
+        httpBasicAuthUser = self.registryValue('httpBasicAuthUser')
+        httpBasicAuthPass = self.registryValue('httpBasicAuthPass')
+        httpAuth = (httpBasicAuthUser, httpBasicAuthPass)
+        return httpAuth
+
+    def createTokenParams(self):
+        tokenParams = {"action":"query","meta":"tokens","type":"login","format":"json"}
+        return tokenParams
+
+    def createLoginParams(self, token):
+        wikiLogin = self.registryValue('wikiLogin')
+        wikiPassword = self.registryValue('wikiPassword')
+        loginParams = {"action":"login","lgname":wikiLogin,"lgpassword":wikiPassword,"lgtoken":token,"format":"json"}
+        return loginParams
+
+    def createPageParams(self):
+        wikiPage = self.registryValue('wikiPage')
+        params = {"action": "parse", "page": wikiPage, "format": "json"}
+        return params 
+
+    def processText(self, game, text):
+        state = False
+        nicks = []
+        for line in text.splitlines():
+            if "mw-headline" in line:
+                state = game.lower() in line.lower()
+            elif state:
+                if line.startswith("</li></ul>"):
+                    break
+                nick = utils.web.htmlToText(line).strip()
+                nicks.append(nick)
+        return nicks
+
+    def getPeopleForGame(self, game):
+
+        cookies = self.doLogin()
+        text = self.fetchPageText(cookies)
+        people = self.processText(game, text)
+
+        return people
+
+    def peli(self, irc, msg, args, game):
+        """<peli>
+
+        Etsii wikistä pelin ja huutaa kaikille että pelit alkamassa.
+        """
+        people = self.getPeopleForGame(game)
+        if not people:
+            return irc.reply("Eipä ole sellaista peliä wikissä")
+        else:
+            irc.reply(format("pelataanpa peliä %s! %s", game, " ".join(people)), prefixNick=False)
+    peli = wrap(peli, ['text'])
+
+
+Class = Pelit
+
+
+# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: