noDRM_DeDRM_tools/Other_Tools/eReader_PDB_Tools/eReaderPDB2PMLZ.pyw

#!/usr/bin/env python
# vim:ts=4:sw=4:softtabstop=4:smarttab:expandtab

import sys
sys.path.append('lib')

import os, os.path, urllib
os.environ['PYTHONIOENCODING'] = "utf-8"
import subprocess
from subprocess import Popen, PIPE, STDOUT
import Tkinter
import Tkconstants
import tkFileDialog
import tkMessageBox
import subasyncio
from subasyncio import Process
from scrolltextwidget import ScrolledText

class MainDialog(Tkinter.Frame):
    def __init__(self, root):
        Tkinter.Frame.__init__(self, root, border=5)
        self.root = root
        self.interval = 2000
        self.p2 = None
        self.status = Tkinter.Label(self, text='eReader eBook Conversion to PMLZ')
        self.status.pack(fill=Tkconstants.X, expand=1)
        body = Tkinter.Frame(self)
        body.pack(fill=Tkconstants.X, expand=1)
        sticky = Tkconstants.E + Tkconstants.W
        body.grid_columnconfigure(1, weight=2)

        Tkinter.Label(body, text='eBook PDB input file').grid(row=0, sticky=Tkconstants.E)
        self.pdbpath = Tkinter.Entry(body, width=50)
        self.pdbpath.grid(row=0, column=1, sticky=sticky)
        cwd = os.getcwdu()
        cwd = cwd.encode('utf-8')
        self.pdbpath.insert(0, cwd)
        button = Tkinter.Button(body, text="...", command=self.get_pdbpath)
        button.grid(row=0, column=2)

        Tkinter.Label(body, text='Name on CC').grid(row=1, sticky=Tkconstants.E)
        self.name = Tkinter.StringVar()
        self.nameinfo = Tkinter.Entry(body, width=40, textvariable=self.name)
        self.nameinfo.grid(row=1, column=1, sticky=sticky)

        Tkinter.Label(body, text='Last 8 digits of CC Number').grid(row=2, sticky=Tkconstants.E)
        self.ccnum = Tkinter.StringVar()
        self.ccinfo = Tkinter.Entry(body, width=10, textvariable=self.ccnum)
        self.ccinfo.grid(row=2, column=1, sticky=sticky)

        msg1 = 'Conversion Log \n\n'
        self.stext = ScrolledText(body, bd=5, relief=Tkconstants.RIDGE, height=15, width=60, wrap=Tkconstants.WORD)
        self.stext.grid(row=3, column=0, columnspan=2,sticky=sticky)
        self.stext.insert(Tkconstants.END,msg1)

        buttons = Tkinter.Frame(self)
        buttons.pack()
        self.sbotton = Tkinter.Button(
            buttons, text="Start", width=10, command=self.convertit)
        self.sbotton.pack(side=Tkconstants.LEFT)

        Tkinter.Frame(buttons, width=10).pack(side=Tkconstants.LEFT)
        self.qbutton = Tkinter.Button(
            buttons, text="Quit", width=10, command=self.quitting)
        self.qbutton.pack(side=Tkconstants.RIGHT)

    # read from subprocess pipe without blocking
    # invoked every interval via the widget "after"
    # option being used, so need to reset it for the next time
    def processPipe(self):
        poll = self.p2.wait('nowait')
        if poll != None: 
            text = self.p2.readerr()
            text += self.p2.read()
            msg = text + '\n\n' + 'File successfully converted\n'
            if poll != 0:
                msg = text + '\n\n' + 'Error: Conversion Failed\n'
            self.showCmdOutput(msg)
            self.p2 = None
            self.sbotton.configure(state='normal')
            return
        text = self.p2.readerr()
        text += self.p2.read()
        self.showCmdOutput(text)
        # make sure we get invoked again by event loop after interval 
        self.stext.after(self.interval,self.processPipe)
        return

    # post output from subprocess in scrolled text widget
    def showCmdOutput(self, msg):
        if msg and msg !='':
            if sys.platform.startswith('win'):
                msg = msg.replace('\r\n','\n')
            self.stext.insert(Tkconstants.END,msg)
            self.stext.yview_pickplace(Tkconstants.END)
        return

    # run erdr2pml.py as a subprocess via pipes and collect stdout
    def erdr(self, infile, name, ccnum):
        # os.putenv('PYTHONUNBUFFERED', '1')
        pengine = sys.executable
        if pengine is None or pengine == '':
            pengine = "python"
        pengine = os.path.normpath(pengine)
        cmdline = pengine + ' ./lib/erdr2pml.py --make-pmlz "' + infile + '" "' + name + '" ' + ccnum 
        if sys.platform[0:3] == 'win':
            # search_path = os.environ['PATH']
            # search_path = search_path.lower()
            # if search_path.find('python') >= 0: 
            #     cmdline = 'python lib\erdr2pml.py --make-pmlz "' + infile + '" "' + name + '" ' + ccnum
            # else :
            #     cmdline = 'lib\erdr2pml.py --make-pmlz "' + infile + '" "' + name + '" ' + ccnum
            cmdline = pengine + ' lib\\erdr2pml.py --make-pmlz "' + infile + '" "' + name + '" ' + ccnum
        cmdline = cmdline.encode(sys.getfilesystemencoding())
        p2 = Process(cmdline, shell=True, bufsize=1, stdin=None, stdout=PIPE, stderr=PIPE, close_fds=False)
        return p2


    def get_pdbpath(self):
        pdbpath = tkFileDialog.askopenfilename(
            parent=None, title='Select eReader PDB File',
            defaultextension='.pdb', filetypes=[('eReader eBooks', '.pdb'),
                                                ('All Files', '.*')])
        if pdbpath:
            pdbpath = os.path.normpath(pdbpath)
            self.pdbpath.delete(0, Tkconstants.END)
            self.pdbpath.insert(0, pdbpath)
        return

    def quitting(self):
        # kill any still running subprocess
        if self.p2 != None:
            if (self.p2.wait('nowait') == None):
                self.p2.terminate()
        self.root.destroy()

    # actually ready to run the subprocess and get its output
    def convertit(self):
        # now disable the button to prevent multiple launches
        self.sbotton.configure(state='disabled')
        pdbpath = self.pdbpath.get()
        if not pdbpath or not os.path.exists(pdbpath):
            self.status['text'] = 'Specified eBook file does not exist'
            self.sbotton.configure(state='normal')
            return
        name = self.name.get()
        if not name or name == '':
            self.status['text'] = 'Your forgot to enter the Name on the CC'
            self.sbotton.configure(state='normal')
            return
        ccnum = self.ccnum.get()
        if not ccnum or ccnum == '':
            self.status['text'] = 'Your forgot to enter the last 8 digits on the CC'
            self.sbotton.configure(state='normal')
            return

        log = 'Command = "python erdr2pml.py --make-pmlz "\n'
        log += 'PDB Path = "'+ pdbpath + '"\n'
        log += 'Name = "' + name + '"\n'
        log += 'Last 8 of CC = "' + ccnum + '"\n'
        log += '\n\n'
        log += 'Please Wait ...\n'
        self.stext.insert(Tkconstants.END,log)
        self.p2 = self.erdr(pdbpath, name, ccnum)

        # python does not seem to allow you to create
        # your own eventloop which every other gui does - strange 
        # so need to use the widget "after" command to force
        # event loop to run non-gui events every interval
        self.stext.after(self.interval,self.processPipe)
        return


def main(argv=None):
    root = Tkinter.Tk()
    root.title('eReader PDB to PMLZ Conversion')
    root.resizable(True, False)
    root.minsize(300, 0)
    MainDialog(root).pack(fill=Tkconstants.X, expand=1)
    root.mainloop()
    return 0
    

if __name__ == "__main__":
    sys.exit(main())