4grab/sorter.py

######################################################################
# Copyright 2009, 2010 ryuslash
#
# This file is part of 4grab.
#
# 4grab is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# 4grab is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with 4grab.  If not, see <http://www.gnu.org/licenses/>.
######################################################################

import config
import Image
import shutil
import os
import datetime
import backend

import util

def dummy_option_creator(value1, value2): pass
config._optioncreator = dummy_option_creator

class Sorter:

    def __init__(self):
        self.conf = config.Configuration()
        self.resolutions = self.conf.get_resolutions()
    
    def act(self, filename):
        util.log(util.LogType.Msg, "sorter is acting", filename)

        download_base = self.conf.get_download_location()
        retval = True
        
        if self.check_filename(filename):
            image = None
            try:
                image = Image.open(os.path.join(download_base,
                                            filename))
            except IOError:
                retval = False

            if not image == None and self.archive_check(filename):
                util.log(util.LogType.Msg, "Checking resolution", {"filename":filename, "resolution":image.size})

                for resolution in self.resolutions:
                    resolution = resolution.split('x')
                    foldername = "%s-%s" % (resolution[0],
                                            resolution[1])
                    folderpath = os.path.join(download_base,
                                              foldername)
                    
                    if str(image.size[0]) == resolution[0] and \
                            str(image.size[1]) == resolution[1]:
                        if not os.path.exists(folderpath):
                            os.makedirs(folderpath)
                            
                        self.copy(filename, folderpath)
                        break
                self.archive(filename)
        self.remove(filename)
        return retval

    def copy(self, filename, destpath):
        download_base = self.conf.get_download_location()
        source = os.path.join(download_base,
                              os.path.basename(filename))
        
        today = datetime.date.today()
        dest = os.path.join(destpath,
                            "%d-%d-%d" % (today.year, today.month, today.day))
        util.log(util.LogType.Msg, "going to copy %s to %s" % (source, dest), None)

        if not os.path.exists(dest):
            os.makedirs(dest)

        dest = os.path.join(dest,
                            os.path.basename(filename))
                            
        if source != dest:
            shutil.copy(source, dest)
        else:
            print "\nHow can this even happen?! Copying", source, "to", dest

    def archive(self, filename):
        be = backend.Backend()
        be.add(os.path.basename(filename))

    def archive_check(self, filename):
        be = backend.Backend()
        return be.check(os.path.basename(filename))

    def check_filename(self, filename):
        ext = os.path.splitext(filename)[1]
        return ext == ".jpg" or \
            ext == ".png" or \
            ext == ".gif"

    def remove(self, filename):
        download_base = self.conf.get_download_location()
        source = os.path.join(download_base, filename)
        os.remove(source)
                                          
if __name__ == "__main__":
    conf = config.Configuration()
    download_base = conf.get_download_location()
    sorter = Sorter()
    for item in os.listdir(download_base):
        sorter.act(item)