# made for those stupid bitches who think that making a screenshot uploader to static.md is difficult
# static md has a great API which makes every so easy
# an interface can be made using kivy
# i recommend to take screenshot after detecting some keyboard interrupt
import pyscreenshot
import requests
import hashlib
import time

address_token = 'http://static.md/api/v2/get-token/'
address_upload = 'http://static.md/api/v2/upload/'


def take_a_screen(file_name):
    file_name += ".png"
    pyscreenshot.grab_to_file(file_name)


def md5(file_name):
    hash_md5 = hashlib.md5()
    with open(file_name, "rb") as f:
        for chunk in iter(lambda: f.read(4096), b""):
            hash_md5.update(chunk)
    return hash_md5.hexdigest()


def upload_file(file_name):
    file_name += ".png"
    md5_value = md5(file_name)
    response = requests.post(address_token, data={'md5': md5_value})
    response_json = response.json()
    token_value = response_json['token']
    files = {'image': open(file_name, 'rb')}
    time.sleep(3)
    upload_response = requests.post(address_upload, files=files, data={'token': token_value})
    upload_json = upload_response.json()
    return upload_json['image']


if __name__ == "__main__":
    some_file_name = 'you_are_really_f_up'
    take_a_screen(some_file_name)
    link_to_image = upload_file(some_file_name)
    print link_to_image