def img(bot, event, *args):
if len(args) > 0 and args[0] == 'list':
aliases = UtilDB.get_list_of_aliases()
segments = []
for alias in aliases:
segments.append(hangups.ChatMessageSegment(alias))
segments.append(hangups.ChatMessageSegment('\n', hangups.SegmentType.LINE_BREAK))
bot.send_message_segments(event.conv, segments)
elif len(args) > 0 and args[0] == 'add':
if len(args) < 3:
bot.send_message(event.conv, "Error: not enough arguments")
return
# alias is all arguments except the first and last
alias = ''.join(args[1:len(args)-1])
# strip spaces and non-alphanumeric characters
alias = ''.join(filter(str.isalnum, alias))
alias = alias.lower()
url = args[len(args)-1]
if UtilDB.get_urls_for_alias(alias) is not None:
bot.send_message(event.conv, "Error: that alias already exists")
return
print(str(is_valid_url(url)))
if not is_valid_url(url):
bot.send_message(event.conv, "Error: invalid URL")
return
UtilDB.set_alias_for_url(url, alias)
bot.send_message(event.conv, "Alias {alias} saved with URL {url}".format(alias=alias,url=url))
#no special arguments
elif len(args) > 0:
url = args[0]
is_alias = False
alias = ''.join(args)
# strip spaces and non-alphanumeric characters
alias = ''.join(filter(str.isalnum, alias))
alias = alias.lower()
alias_url_list = UtilDB.get_urls_for_alias(alias)
if alias_url_list is not None:
random_url = random.choice(alias_url_list)
if random_url is not None:
url = random_url
is_alias = True
image_id_list = UtilDB.get_imageids_for_alias(alias)
image_id = None
if image_id_list is not None:
image_id = random.choice(image_id_list)
is_alias = True
if not is_valid_url(url):
url = 'http://' + url
if not is_valid_url(url) and image_id is None:
bot.send_message(event.conv, "Error: invalid alias or URL.")
return
if image_id is None:
image_id = UtilDB.get_imageid_for_url(url)
desc = None
if not is_alias:
image_info = UtilBot.get_image_info(url)
url, desc = image_info
if desc is None and not is_alias:
desc = ' '.join(args[1:])
print(image_id)
if image_id is None:
filename = UtilBot.download_image(url, 'images', False)
image_id = yield from UtilBot.upload_image(bot, filename)
UtilDB.set_imageid_for_url(url, image_id)
os.remove(filename)
bot.send_message_segments(event.conv,
[hangups.ChatMessageSegment(desc)] if desc else None,
image_id)
