Merge pull request #30 from wardellchandler/image_system

Image system

Author: wardellbagby

Core/Bot.py

@@ -2,11 +2,15 @@
 # coding=utf-8
 from datetime import date, datetime
 import os
+import random
 import sys
 import asyncio
+import tempfile
 import time
 import signal
 import traceback
+from urllib import request
+from urllib.request import FancyURLopener
 
 import hangups
 from hangups.ui.utils import get_conv_name
@@ -16,6 +20,12 @@
 from Core import Handlers
 
 
+class HangoutsBotOpener(FancyURLopener):
+    version = 'Mozilla/5.0 (Windows NT 6.3; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/37.0.2049.0 Safari/537.36'
+
+
+request.urlretrieve = HangoutsBotOpener().retrieve
+
 __version__ = '1.1'
 LOG_FORMAT = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
 
@@ -217,17 +227,32 @@ def send_message(self, conversation, text):
         """"Send simple chat message"""
         self.send_message_segments(conversation, [hangups.ChatMessageSegment(text)])
 
-    def send_message_segments(self, conversation, segments):
+
+    def send_message_segments(self, conversation, segments, image_id=None):
         """Send chat message segments"""
         # Ignore if the user hasn't typed a message.
         if len(segments) == 0:
             return
         # XXX: Exception handling here is still a bit broken. Uncaught
         # exceptions in _on_message_sent will only be logged.
         asyncio.async(
-            conversation.send_message(segments)
+            conversation.send_message(segments, image_id=image_id)
         ).add_done_callback(self._on_message_sent)
 
+
+    @asyncio.coroutine
+    def upload_image(self, url, filename=None, delete=False):
+        if not filename:
+            tempdir = tempfile.gettempdir()
+            filename = tempdir + os.sep + '{}.png'.format(random.randint(0, 9999999999))
+        request.urlretrieve(url, filename)
+        file = open(filename, "rb")
+        image_id = yield from self._client.upload_image(file)
+        if delete:
+            file.close()
+            os.remove(filename)
+        return image_id
+
     def list_conversations(self):
         """List all active conversations"""
         convs = sorted(self._conv_list.get_all(),

Core/Commands/DefaultCommands.py

@@ -1,4 +1,9 @@
+import asyncio
+import html
 import json
+import os
+import random
+import threading
 from urllib import parse
 from urllib import request
 import re
@@ -715,9 +720,48 @@ def karma(bot, event, name=None, *args):
                 segments.append(hangups.ChatMessageSegment("{}: {}".format(karma_list[i][0], karma_list[i][1])))
                 segments.append(hangups.ChatMessageSegment("\n", segment_type=hangups.SegmentType.LINE_BREAK))
 
-
         segments.append(hangups.ChatMessageSegment("\n", segment_type=hangups.SegmentType.LINE_BREAK))
         segments.append(hangups.ChatMessageSegment("Average Karma:", is_italic=True))
         segments.append(hangups.ChatMessageSegment("\n", segment_type=hangups.SegmentType.LINE_BREAK))
         segments.append(hangups.ChatMessageSegment('{}'.format((sum([i[1] for i in karma_list]) / len(karma_list)))))
         bot.send_message_segments(event.conv, segments)
+
+
+@DispatcherSingleton.register_aliases(["img"])
+def image(bot, event, *args):
+    query = ' '.join(args)
+    url = 'http://ajax.googleapis.com/ajax/services/search/images?v=1.0&rsz=8&safe=active&' \
+          + parse.urlencode({'q': query})
+
+    resp = request.urlopen(url)
+    image_json = json.loads(resp.read().decode())
+    url = image_json['responseData']['results'][0]['unescapedUrl']
+
+    @asyncio.coroutine
+    def send_image(bot, event, url):
+        image_id = yield from bot.upload_image(url)
+        bot.send_message_segments(event.conv, [
+            hangups.ChatMessageSegment("Picture Message", segment_type=hangups.SegmentType.LINE_BREAK)],
+                                  image_id=image_id)
+
+    yield from send_image(bot, event, url)
+
+
+@DispatcherSingleton.register
+def gif(bot, event, *args):
+    query = ' '.join(args)
+    url = 'http://ajax.googleapis.com/ajax/services/search/images?v=1.0&rsz=8&safe=active&imgtype=animated&' \
+          + parse.urlencode({'q': query})
+    resp = request.urlopen(url)
+    image_json = json.loads(resp.read().decode())
+    url = image_json['responseData']['results'][0]['unescapedUrl']
+
+    @asyncio.coroutine
+    def send_image(bot, event, url):
+        image_id = yield from bot.upload_image(url)
+        bot.send_message_segments(event.conv, [
+            hangups.ChatMessageSegment("Picture Message", segment_type=hangups.SegmentType.LINE_BREAK)],
+                                  image_id=image_id)
+
+    yield from send_image(bot, event, url)
+

Core/Commands/Dispatcher.py

@@ -1,5 +1,6 @@
 import asyncio
 from datetime import datetime
+from functools import wraps
 from Core.Util import UtilBot
 import traceback
 
@@ -70,12 +71,45 @@ def run(self, bot, event, bot_command_char, *args, **kwds):
             log.close()
             print(traceback.format_exc())
 
+    def register_aliases(self, aliases=None):
+        """Registers a command under the function name & any names specified in aliases.
+        """
+
+        def func_wrapper(func):
+            self.commands[func.__name__] = func
+            for alias in aliases:
+                self.commands[alias] = func
+            return func
+
+        return func_wrapper
+
+    def register_extras(self, is_hidden=False, aliases=None):
+        """Registers a function as hidden with aliases, or any combination of that."""
+
+        def func_wrapper(func):
+            if is_hidden and aliases:
+                self.hidden_commands[func.__name__] = func
+                for alias in aliases:
+                    self.hidden_commands[alias] = func
+            elif aliases:
+                self.commands[func.__name__] = func
+                for alias in aliases:
+                    self.commands[alias] = func
+            elif is_hidden:
+                self.hidden_commands[func.__name__] = func
+            else:
+                self.commands[func.__name__] = func
+            return func
+
+        return func_wrapper
+
     def register(self, func):
         """Decorator for registering command"""
         self.commands[func.__name__] = func
         return func
 
     def register_hidden(self, func):
+        """Registers a command as hidden (This makes it only runnable by the Bot and it won't appear in the help menu)"""
         self.hidden_commands[func.__name__] = func
         return func
 

Core/Commands/ExtraCommands.py

@@ -12,6 +12,7 @@
 import hangups
 import re
 import requests
+import parsedatetime
 from Core.Commands.Dispatcher import DispatcherSingleton
 from Core.Util import UtilBot
 from Libraries import Genius
@@ -140,43 +141,20 @@ def send_reminder(bot, conv, reminder_time, reminder_text, loop):
 
     # Set a new reminder
     args = list(args)
-    date = str(datetime.now().today().date())
-    time = str((datetime.now() + timedelta(hours=1)).time())
-    set_date = False
-    set_time = False
-    index = 0
-    while index < len(args):
-        item = args[index]
-        if item[0].isnumeric():
-            if '/' in item or '-' in item:
-                date = item
-                args.remove(date)
-                set_date = True
-                index -= 1
-            else:
-                time = item
-                args.remove(time)
-                set_time = True
-                index -= 1
-        if set_date and set_time:
-            break
-        index += 1
-
-    reminder_time = date + ' ' + time
-    if len(args) > 0:
-        reminder_text = ' '.join(args)
-    else:
+    reminder_text = ' '.join(args)
+    result = parsedatetime.nlp(reminder_text)
+    reminder_time = result[0][0]
+    reminder_text.replace(result[0][-1], '')
+    if reminder_text.strip() == '':
         bot.send_message(event.conv, 'No reminder text set.')
         return
+
     current_time = datetime.now()
-    try:
-        reminder_time = parser.parse(reminder_time)
-    except (ValueError, TypeError):
-        bot.send_message(event.conv, "Couldn't parse " + reminder_time + " as a valid date.")
-        return
     if reminder_time < current_time:
-        reminder_time = current_time + timedelta(hours=1)
+        bot.send_message("Invalid Date: {}".format(reminder_time.strftime('%B %d, %Y %I:%M%p')))
+
     reminder_interval = (reminder_time - current_time).seconds
+
     reminder_timer = threading.Timer(reminder_interval, send_reminder,
                                      [bot, event.conv, reminder_interval, reminder_text, asyncio.get_event_loop()])
     reminders.append((reminder_timer, reminder_text, current_time))