findEmote.py

# coding=UTF-8
#!/usr/bin/python
# -*- coding: UTF-8 -*-
# vim: set fileencoding=UTF-8 :

import json
from pprint import pprint
import os
import random
import urllib.request, urllib.parse, urllib.error
from PIL import Image, ImageFont, ImageDraw
import cacher
import configparser

config = configparser.ConfigParser(inline_comment_prefixes=(';',))
config.readfp(open('config.cfg'))


#
defaultSeed = config.get('Options','default_seed') #RAND0m_XD or something or whatever
MIN_LENGTH = config.getint('Options','emotional_diversity') #minimum number of emotes for a tag to be valid

emotesByName = {}
emotesByPony = {}
BANNED_TAGS = config.get('Ignore','banned_tags').split()
emoteMetadata = {}

for file in os.listdir('tagAssignments'):
	#print 'tagassignments file:'+file
	with open('tagAssignments/'+file) as data_file:
		data = json.load(data_file)
		emoteMetadata.update(data)
for fn in os.listdir('emotes'):
	data = None
	with open('emotes/'+fn) as data_file:
		data = json.load(data_file)
		for key in list(data.keys()):
			value = data[key]
			if 'Emotes' not in value or '' not in value['Emotes']:
				del data[key]
				continue
			if 'Image' in value['Emotes']['']:
				if 'amazonaws' in value['Emotes']['']['Image']:
					del data[key]
					continue
			else:
				del data[key]
			if 'Offset' in value['Emotes']['']:
				offset = value['Emotes']['']['Offset']
				if offset[0] > 0 or offset[1] > 0:
					del data[key]
				else:
					offset[0] = offset[0]*-1
					offset[1] = offset[1]*-1
	emotesByName.update(data)
for fn in os.listdir('tags'):
	data = None
	with open('tags/'+fn) as data_file:
		data = json.load(data_file)
	for key, value in list(data.items()):
		if key in emotesByName and len(value) == 1:
			emotes = [] #set()
			#print 'tag: '+value[0]
			if value[0][1:] not in BANNED_TAGS:
				if value[0] not in emotesByPony:
					emotesByPony[value[0]] = emotes
				else:
					emotes = emotesByPony[value[0]]
				emotes.append(key) #.add
			emotes.sort()
for key in list(emotesByPony.keys()):
	if len(emotesByPony[key]) < MIN_LENGTH:
		del emotesByPony[key]
		continue

#pprint(emotesByPony) # This is the only one of the 3 structures that is anywhere reasonable to output
#pprint(emotesByName)
#pprint(emoteMetadata)

#
def getProceduralPony(seed):
	random.seed(seed)
	pony = random.choice(list(emotesByPony.keys()))
	return pony

#
def getProceduralEmote(seed1, seed2):
	return getRandomEmote(seed2, getProceduralPony(seed1))

#
def getEmote(emoteName):
	data = emotesByName[emoteName]
	url = data['Emotes']['']['Image']
	offset = [0,0]
	if 'Offset' in data['Emotes']['']:
		offset = data['Emotes']['']['Offset']
	#offset[0]=offset[0]*-1
	#offset[1]=offset[1]*-1
	#if offset[0]<0 or offset[1]<1:
	#	return None
	size = data['Emotes']['']['Size']
	print(('emotename: '+emoteName))
	pprint(data)
	print(url)
	if 'http:' not in url: # let's hope no one bugs this out with a URL of gopher://fuckhttp:6969/~yourmom.png
		url = 'http:'+url
	imgloc = cacher.getUrlFile(url)
	fullImage = Image.open(imgloc).convert('RGBA')
	print(offset)
	print(size)
	fullImage = fullImage.crop((offset[0], offset[1], offset[0]+size[0], offset[1]+size[1]))
	if emoteName in emoteMetadata:
		data = emoteMetadata[emoteName]
		if 'right' in data:
			print('findemote flipping image')
			fullImage = fullImage.transpose(Image.FLIP_LEFT_RIGHT)
	return fullImage


#
def getRandomEmoteByPony(pony, seed=None):
	if seed is not None:
		random.seed(seed)
	emoteNames = list(emotesByPony[pony])
	print(('randomly choosing emote from list of '+str(len(emoteNames))+" with seed "+str(seed)))
	random.seed(seed)
	emoteName = random.choice(emoteNames)
	print(('selected emote: '+emoteName))
	return getEmote(emoteName)

#
def getRandomEmote(seed, pony=None):
	# set seed for consistency
	global defaultSeed
	if seed is None or len(seed) < 1:
		print("Using default seed")
		seed = defaultSeed
	random.seed(seed+defaultSeed)

	if pony is None:
		print("Picking from an empty pony")
		pony = random.choice(list(emotesByPony.keys()))
	print(pony)

	return getRandomEmoteByPony(pony, seed+defaultSeed)

# Increment is a parameter used to make sure this does not loop indefinitely
def select_horse(nickname, ponylist, unique=False, increment=1):
	word = ""
	if unique is True:
		word = "unique "
	print(("Picking "+word+"pony for "+nickname+", take "+str(increment)))
	pony = getProceduralPony(nickname*increment)
	print(("Trying "+pony))
	if unique is False or increment > 12 or pony not in ponylist:
		ponylist.append(pony)
		return pony
	else:
		print(('pony '+pony+' is already in use; picking a new horse'))
		return select_horse(nickname, ponylist, unique, increment + 1)

# Other code

#image=getProceduralEmote("pondy")
#image.show()