[V3 Wolfram] Add initial commit

wolfram/wolfram.py

@@ -0,0 +1,57 @@
+import os
+import aiohttp
+from redbot.core import Config, commands, checks
+import xml.etree.ElementTree as ET
+
+
+class Wolfram:
+    """Ask Wolfram Alpha a question."""
+    def __init__(self, bot):
+        self.bot = bot
+		
+        default_global = {
+            "WOLFRAM_API_KEY": None,
+        }
+		
+        self.config = Config.get_conf(self, 2788801004)
+        self.config.register_guild(**default_global)
+
+
+    @commands.command(name='wolfram', aliases=['ask'])
+    async def _wolfram(self, ctx, *arguments: str):
+        """
+        Ask Wolfram Alpha any question.
+        """
+        api_key = await self.config.WOLFRAM_API_KEY()
+        if api_key:
+            url = 'http://api.wolframalpha.com/v2/query?'
+            query = ' '.join(arguments)
+            payload = {'input': query, 'appid': api_key}
+            headers = {'user-agent': 'Red-cog/2.0.0'}
+            conn = aiohttp.TCPConnector(verify_ssl=False)
+            session = aiohttp.ClientSession(connector=conn)
+            async with session.get(url, params=payload, headers=headers) as r:
+                result = await r.text()
+            session.close()
+            root = ET.fromstring(result)
+            a = []
+            for pt in root.findall('.//plaintext'):
+                if pt.text:
+                    a.append(pt.text.capitalize())
+            if len(a) < 1:
+                message = 'There is as yet insufficient data for a meaningful answer.'
+            else:
+                message = '\n'.join(a[0:3])
+        else:
+            message = 'No API key set for Wolfram Alpha. Get one at http://products.wolframalpha.com/api/'
+        await ctx.send('```{0}```'.format(message))
+
+    @commands.command(name='setwolframapi', aliases=['setwolfram'])
+    @checks.is_owner()
+    async def _setwolframapi(self, ctx, key: str):
+        """
+        Set the api-key.
+        """
+        if key:
+            await self.config.WOLFRAM_API_KEY.set(key)
+            await ctx.send("Key set.")