#!/usr/bin/python
import pymysql
import os
import time
import shutil
import urllib2
from PIL import Image
import hashlib
from urlparse import urlparse
from random import *


class Process(object):
    "Object to process images"

    def __init__(self):

        self.image_folder = '/opt/www/memecdn.net/www/images'
        self.thumbnails = '/opt/www/memecdn.net/www/images/thumbnails'
        self.list = '/opt/www/memecdn.net/www/url_list.txt'
        self.db = 'memecdn'
        self.user = 'memecdn'
        self.sql_pass = 'bz7GY8Yp8bYwrMsX'
        self.sql_host = 'db.zswap.net'
        self.port = 3306

    def sql_connect(self):
        '''
        Connects to a mysql object
        '''
        conn = pymysql.connect(host=self.sql_host,
                               port=self.port,
                               user=self.user,
                               passwd=self.sql_pass,
                               db=self.db)
        return conn

    def remove_broken_images(self):
        images = next(os.walk(self.image_folder))[2]
        for image in images:
            full_path = self.image_folder + image
            if os.path.getsize(full_path) == 0:
                os.remove(full_path)
                print('Removed broken file: %s' % full_path)

    def build_image_list(self):
        url_list = []
        with open(self.list) as f:
            for line in f.readlines():
                url_list.append(line)
            return url_list

    def empty_table(table, self):
        conn = sql_connect()
        cur = conn.cursor()
        sql = "TRUNCATE TABLE %s" % (table)
        try:
            cur.execute(sql)
        except Exception as e:
            print e
        conn.commit()

    def check_size(image, self):
        try:
            im = Image.open(image)
            width, height = im.size
            resolution = [width, height]
            return resolution
        except:
            return None

    def create_thumbnail(image, out):
        size = 256, 256
        width = 0
        height = 0
        try:
            resolution = check_size(image)
            width = resolution[0]
            height = resolution[1]
        except:
            pass
        if width > 256 and height > 256:
            im = Image.open(image)
            im.thumbnail(size)
            im.save(out, "PNG")

    def get_image(url):
        url = url.strip()
        image_id = int(time.time) + randint(1, 99999)
        image_dict = {}
        o = urlparse(url)
        # get some initial information
        image_dict.update({'netloc': o[1]})
        image_dict.update({'path': o[2]})
        image_dict.update({'local_file': image_id})
        image_dict.update({'file_type': url.split(".")[-1]})
        # actually get the image
        header = {'User-Agent': ("Mozilla/5.0 (Windows NT 6.1; "
                                 "WOW64) AppleWebKit/537.36 "
                                 "(KHTML, like Gecko) "
                                 "Chrome/43.0.2357.134 Safari/537.36")}
        try:
            req = urllib2.Request(url, headers={'User-Agent': header})
            raw_img = urllib2.urlopen(req).read()
            f = open(os.path.join('images/',
                                  str(image_dict.get('local_file')) +\
                                  "." +\
                                  image_dict.get('file_type'),
                                  'wb'))
            f.write(raw_img)
            f.close()
        except Exception as e:
            print "failed to get %s" % (url)
            print e
        return image_dict

    def clean_thumbnails(self):
        path = self.thumbnails
        shutil.rmtree(path)
        if not os.path.exists(path):
            os.makedirs(path)


def main():
    '''
    Entry point for class Process to process images
    '''
    process_images = Process()
    process_images.empty_table(images)
    process_images.remove_broken_images()
    for image in process_images.build_image_list():
        print process_images.get_image(image)

if __name__ == '__main__':
    main()