149 lines
5 KiB
Python
149 lines
5 KiB
Python
#!/usr/bin/python3
|
|
from os import mkdir, environ
|
|
from os.path import exists
|
|
import requests
|
|
import slack
|
|
import json
|
|
|
|
# Twitch emote api link for emote metadata
|
|
twitch_emote_api_link = 'https://api.twitchemotes.com/api/v4/channels/0'
|
|
# Twitch emote link for emote images
|
|
twitch_emote_file_tmpl = 'https://static-cdn.jtvnw.net/emoticons/v1/'
|
|
|
|
# BTTV emote api link for emote metadata
|
|
bttv_emote_api_link = 'https://api.betterttv.net/2/emotes'
|
|
# BTTV emote link for emote images
|
|
bttv_emote_file_tmpl = 'https://cdn.betterttv.net/emote/'
|
|
|
|
def import_client_id():
|
|
if(exists('.client_id')):
|
|
try:
|
|
client_id_file = open('.client_id', 'rt')
|
|
output = client_id_file.readline().strip()
|
|
print(output)
|
|
return output
|
|
except Exception as e:
|
|
print(e)
|
|
exit(1)
|
|
|
|
else:
|
|
try:
|
|
client_id = environ['TWITCH_APP_CLIENT_ID']
|
|
return client_id
|
|
except Exception as e:
|
|
print(e)
|
|
exit(1)
|
|
|
|
|
|
def grab_id(n):
|
|
"""Map function to extract the id and code from each dictionary entry
|
|
|
|
Arguments:
|
|
n {int} -- Current dictionary object
|
|
|
|
Returns:
|
|
arr -- Array containing the id and code
|
|
"""
|
|
return [f'{n["id"]}', f'{n["code"]}']
|
|
|
|
|
|
def request_twitch_emote_ids(client_id, service):
|
|
"""Send the request for a json of the Twitch emote metadata
|
|
|
|
Arguments:
|
|
service {str} -- String containing either "Twitch" or "BTTV" to determine which code to use
|
|
|
|
Returns:
|
|
list -- List iterable of the ids and codes for the emotes returned by the request json
|
|
"""
|
|
|
|
req_dict = {}
|
|
|
|
if(service.lower() == 'twitch'):
|
|
headers = {
|
|
'accept': 'application/vnd.twitchtv.v5+json',
|
|
'Client-ID': client_id
|
|
}
|
|
req = requests.get(twitch_emote_api_link, headers=headers)
|
|
req_dict = json.loads(req.content)
|
|
req.close()
|
|
elif(service.lower() == 'bttv'):
|
|
req = requests.get(bttv_emote_api_link)
|
|
req_dict = json.loads(req.content)
|
|
req.close()
|
|
|
|
return list(map(grab_id, req_dict['emotes']))
|
|
|
|
|
|
def save_twitch_emotes_metadata(dirpath, emote_array):
|
|
"""Save the emote_array metadata to a text file to refer to later
|
|
|
|
Arguments:
|
|
dirpath {str} -- String containing the directory path for output
|
|
emote_array {list} -- List iterable of the ids and code for the emotes returned by the original request json
|
|
"""
|
|
|
|
if(not exists(f'{dirpath}/emote_metadata.txt')):
|
|
metadata_file = open(f'{dirpath}/emote_metadata.txt', 'w+')
|
|
else:
|
|
metadata_file = open(f'{dirpath}/emote_metadata.txt', 'a')
|
|
|
|
for x in emote_array:
|
|
metadata_file.write(f'id:{x[0]}\tcode:{x[1]}\n')
|
|
metadata_file.close()
|
|
|
|
|
|
def save_twitch_emotes_images(client_id, service, dirpath, emote_array):
|
|
"""Send a request for the emote images and save them locally to a specified directory
|
|
|
|
Arguments:
|
|
service {str} -- String contianing either "Twitch" or "BTTV" to determine which code to use
|
|
dirpath {str} -- String containing the directory path for output
|
|
emote_array {list} -- List iterable of the ids and codes for the emotes returned by the original request json
|
|
"""
|
|
|
|
if(not exists(dirpath)):
|
|
mkdir(dirpath)
|
|
|
|
if(service.lower() == 'twitch'):
|
|
for emote_index in emote_array:
|
|
if(not exists(f'{dirpath}/{emote_index[1].lower()}.png')):
|
|
headers = {
|
|
'accept': 'application/vnd.twitchtv.v5+json',
|
|
'Client-ID': client_id
|
|
}
|
|
req = requests.get(
|
|
f'{twitch_emote_file_tmpl}{emote_index[0]}/1.0', headers=headers)
|
|
if('/' in emote_index[1] or '\\' in emote_index[1] or ':' in emote_index[1]):
|
|
pass
|
|
else:
|
|
emote_file = open(f'{dirpath}/{emote_index[1].lower()}.png', 'bw+')
|
|
emote_file.write(req.content)
|
|
emote_file.close()
|
|
elif(service.lower() == 'bttv'):
|
|
for emote_index in emote_array:
|
|
if(not exists(f'{dirpath}/{emote_index[1].lower()}.png')):
|
|
req = requests.get(
|
|
f'{bttv_emote_file_tmpl}{emote_index[0]}/1x')
|
|
emote_file = open(f'{dirpath}/{emote_index[1].lower()}.png', 'bw+')
|
|
emote_file.write(req.content)
|
|
emote_file.close()
|
|
|
|
|
|
def main():
|
|
# Attempt import of Client ID
|
|
client_id = import_client_id()
|
|
|
|
# Grab the Twitch official emotes
|
|
emotes_array = request_twitch_emote_ids(client_id, 'twitch')
|
|
save_twitch_emotes_images(client_id, 'twitch', 'out', emotes_array)
|
|
save_twitch_emotes_metadata('out', emotes_array)
|
|
|
|
# Grab the BTTV emotes
|
|
emotes_array = request_twitch_emote_ids(client_id, 'bttv')
|
|
save_twitch_emotes_images(client_id, 'bttv', 'out', emotes_array)
|
|
save_twitch_emotes_metadata('out', emotes_array)
|
|
|
|
|
|
if(__name__ == "__main__"):
|
|
main()
|