# -*- coding: utf-8 -*-

import os
import sys
import datetime
import base64

import requests


def main():
    input_filename = sys.argv[1]

    try:
        output_filename = sys.argv[2]
    except:
        now = datetime.datetime.now().isoformat()
        output_filename = 'page_{0}.png'.format(now)
    file_type = output_filename.split(".")[-1]

    payload = {'file_type': file_type,
    'filename': "{0}/{1}".format("/".join(os.path.realpath(__file__).split('/')[:-1]), input_filename)}
    response = requests.post('http://localhost:8080/', data=payload)
    png_bytes = base64.b64decode(response.content)

    with open(output_filename, 'wb') as f:
        f.write(png_bytes)

    print 'Writing to ' + output_filename


if __name__ == '__main__':
    main()