import requests
import sys
import tempfile
import warnings
from PIL import Image 
import multiprocessing as mp
import os

temp_dir = tempfile.mkdtemp()

warnings.catch_warnings()
warnings.simplefilter("ignore")

def download(emoji):
    emoji_shortcode = emoji['shortcode']
    emoji_url = emoji['static_url']
    file_type = emoji_url.split('.')[-1]
    file_name = emoji_shortcode + '.' + file_type
    file_path = temp_dir + '/' + file_name
    
    image = requests.get(emoji_url, verify=False).content
    file = open(file_path, 'wb')
    file.write(image)
    file.close()
    print('downloaded', file_path)

def main():
    argv = sys.argv
    if len(argv) != 2:
        print('usage: export.py <url>')
        return 

    url = 'https://' + argv[1] + '/api/v1/custom_emojis'
    res = requests.get(url, verify=False)

    emojies = res.json()

    threads = mp.cpu_count()

    with mp.Pool(threads) as p:
        p.map(download, emojies)

    os.system('tar czvf export.tar.gz -C ' + temp_dir + ' . ' + temp_dir)
    os.system('rm -fr ' + temp_dir)

if __name__ == '__main__':
    main()