DeDRM_tools/Calibre_Plugins/K4MobiDeDRM_plugin/__init__.py

#!/usr/bin/env python

from __future__ import with_statement

from calibre.customize import FileTypePlugin
from calibre.gui2 import is_ok_to_use_qt
# from calibre.ptempfile import PersistentTemporaryDirectory

from calibre_plugins.k4mobidedrm import kgenpids
from calibre_plugins.k4mobidedrm import topazextract
from calibre_plugins.k4mobidedrm import mobidedrm

import sys
import os
import re

class K4DeDRM(FileTypePlugin):
    name                = 'K4PC, K4Mac, Kindle Mobi and Topaz DeDRM' # Name of the plugin
    description         = 'Removes DRM from K4PC and Mac, Kindle Mobi and Topaz files.  Provided by the work of many including DiapDealer, SomeUpdates, IHeartCabbages, CMBDTC, Skindle, DarkReverser, ApprenticeAlf, etc.'
    supported_platforms = ['osx', 'windows', 'linux'] # Platforms this plugin will run on
    author              = 'DiapDealer, SomeUpdates' # The author of this plugin
    version             = (0, 3, 1)   # The version number of this plugin
    file_types          = set(['prc','mobi','azw','azw1','tpz']) # The file types that this plugin will be applied to
    on_import           = True # Run this plugin during the import
    priority            = 210  # run this plugin before mobidedrm, k4pcdedrm, k4dedrm
    minimum_calibre_version = (0, 7, 55)

    def run(self, path_to_ebook):

        k4 = True
        if sys.platform.startswith('linux'):
            k4 = False
        pids = []
        serials = []
        kInfoFiles = []

        # Get supplied list of PIDs to try from plugin customization.
        customvalues = self.site_customization.split(',')
        for customvalue in customvalues:
            customvalue = str(customvalue)
            customvalue = customvalue.strip()
            if len(customvalue) == 10 or len(customvalue) == 8:
                pids.append(customvalue)
            else :
                if len(customvalue) == 16 and customvalue[0] == 'B':
                    serials.append(customvalue)
                else:
                    print "%s is not a valid Kindle serial number or PID." % str(customvalue)
            		
        # Load any kindle info files (*.info) included Calibre's config directory.
        try:
            # Find Calibre's configuration directory.
            confpath = os.path.split(os.path.split(self.plugin_path)[0])[0]
            print 'K4MobiDeDRM: Calibre configuration directory = %s' % confpath
            files = os.listdir(confpath)
            filefilter = re.compile("\.info$|\.kinf$", re.IGNORECASE)
            files = filter(filefilter.search, files)
            if files:
                for filename in files:
                    fpath = os.path.join(confpath, filename)
                    kInfoFiles.append(fpath)
                print 'K4MobiDeDRM: Kindle info/kinf file %s found in config folder.' % filename
        except IOError:
            print 'K4MobiDeDRM: Error reading kindle info/kinf files from config directory.'
            pass

        mobi = True
        magic3 = file(path_to_ebook,'rb').read(3)
        if magic3 == 'TPZ':
            mobi = False

        bookname = os.path.splitext(os.path.basename(path_to_ebook))[0]

        if mobi:
            mb = mobidedrm.MobiBook(path_to_ebook)
        else:
            mb = topazextract.TopazBook(path_to_ebook)

        title = mb.getBookTitle()
        md1, md2 = mb.getPIDMetaInfo()
        pidlst = kgenpids.getPidList(md1, md2, k4, pids, serials, kInfoFiles) 

        try:
            mb.processBook(pidlst)

        except mobidedrm.DrmException:
            #if you reached here then no luck raise and exception
            if is_ok_to_use_qt():
                from PyQt4.Qt import QMessageBox
                d = QMessageBox(QMessageBox.Warning, "K4MobiDeDRM Plugin", "Error decoding: %s\n" % path_to_ebook)
                d.show()
                d.raise_()
                d.exec_()
            raise Exception("K4MobiDeDRM plugin could not decode the file")
        except topazextract.TpzDRMError:
            #if you reached here then no luck raise and exception
            if is_ok_to_use_qt():
                    from PyQt4.Qt import QMessageBox
                    d = QMessageBox(QMessageBox.Warning, "K4MobiDeDRM Plugin", "Error decoding: %s\n" % path_to_ebook)
                    d.show()
                    d.raise_()
                    d.exec_()
            raise Exception("K4MobiDeDRM plugin could not decode the file")

        print "Success!"
        if mobi:
            of = self.temporary_file(bookname+'.mobi')
            mb.getMobiFile(of.name)
        else :
            of = self.temporary_file(bookname+'.htmlz')
            mb.getHTMLZip(of.name)
            mb.cleanup()
        return of.name

    def customization_help(self, gui=False):
        return 'Enter 10 character PIDs and/or Kindle serial numbers, use a comma (no spaces) to separate each PID or SerialNumber from the next.'