ComicInfoEditor α

ComicInfoEditor.py

import os
import wx
import InfoEditor


class ComicInfoGetter(wx.App):

    def OnInit(self):
        frm = GuiWindow("ComicInfoEditor")
        frm.Show()
        return 1


class GuiWindow(wx.Frame):

    def __init__(self, title):

        wx.Frame.__init__(self, None, -1, title, size=(480, 600), pos=(400, 448))
        panel = wx.Panel(self, wx.ID_ANY)

        self.lexicon_url_input = wx.TextCtrl(panel, wx.ID_ANY, 'Info URL')
        self.filepath = wx.TextCtrl(panel, wx.ID_ANY , 'Archive Path', style=wx.TE_READONLY)
        self.openfilebutton = wx.Button(panel, wx.ID_ANY, '...')

        self.savebutton = wx.Button(panel, wx.ID_ANY, 'Save XML to Archive')
        self.resetbutton = wx.Button(panel, wx.ID_ANY, 'Reset Editing')

        self.editorfield_series = wx.TextCtrl(panel, wx.ID_ANY, 'Series')
        self.editorfield_writer = wx.TextCtrl(panel, wx.ID_ANY, '')
        self.editorfield_penciller = wx.TextCtrl(panel, wx.ID_ANY, '')
        self.editorfield_genre = wx.TextCtrl(panel, wx.ID_ANY, '')
        self.editorfield_year = wx.TextCtrl(panel, wx.ID_ANY, '')
        self.editorfield_month = wx.TextCtrl(panel, wx.ID_ANY, '')
        self.editorfield_day = wx.TextCtrl(panel, wx.ID_ANY, '')

        self.openfilebutton.Bind(wx.EVT_BUTTON, self.open_archive)
        self.filepath.Bind(wx.EVT_TEXT, self.load_comicinfo)
        self.savebutton.Bind(wx.EVT_BUTTON, self.save_comicinfo)
        self.resetbutton.Bind(wx.EVT_BUTTON, self.load_comicinfo)

        dt = Droptarget(self)
        self.SetDropTarget(dt)

        line_filepath = wx.BoxSizer(wx.HORIZONTAL)
        line_lexiconurl = wx.BoxSizer(wx.HORIZONTAL)
        line_series = wx.BoxSizer(wx.HORIZONTAL)
        line_artist = wx.BoxSizer(wx.HORIZONTAL)
        line_others = wx.BoxSizer(wx.HORIZONTAL)
        line_saveloadbuttons = wx.BoxSizer(wx.HORIZONTAL)

        layout = wx.BoxSizer(wx.VERTICAL)

        line_filepath.Add(self.filepath, proportion=1, flag=wx.GROW)
        line_filepath.Add(self.openfilebutton, proportion=0)

        line_lexiconurl.Add(self.lexicon_url_input, proportion=1, flag=wx.GROW)

        line_series.Add(self.editorfield_series, proportion=1, flag=wx.GROW)

        line_artist.Add(self.editorfield_writer, proportion=1, flag=wx.GROW)
        line_artist.Add(self.editorfield_penciller, proportion=1, flag=wx.GROW)

        line_others.Add(self.editorfield_genre, proportion=1, flag=wx.GROW)
        line_others.Add(self.editorfield_year, proportion=0, flag=wx.GROW)
        line_others.Add(self.editorfield_month, proportion=0, flag=wx.GROW)
        line_others.Add(self.editorfield_day, proportion=0, flag=wx.GROW)

        line_saveloadbuttons.Add(self.savebutton, proportion=0)
        line_saveloadbuttons.Add(self.resetbutton, proportion=0)

        layout.Add(line_filepath, proportion=0, flag=wx.ALIGN_RIGHT | wx.EXPAND | wx.ALL, border=5)
        layout.Add(line_lexiconurl, proportion=0, flag=wx.ALIGN_RIGHT | wx.EXPAND | wx.ALL, border=5)
        layout.Add(line_series, proportion=0, flag=wx.ALIGN_RIGHT | wx.EXPAND | wx.ALL, border=5)
        layout.Add(line_artist, proportion=0, flag=wx.ALIGN_RIGHT | wx.EXPAND | wx.ALL, border=5)
        layout.Add(line_others, proportion=0, flag=wx.ALIGN_RIGHT | wx.EXPAND | wx.ALL, border=5)
        layout.Add(line_saveloadbuttons, proportion=0, flag=wx.ALIGN_CENTER | wx.ALL, border=5)

        panel.SetSizer(layout)

    def open_archive(self, e):
        path = ''
        dialog = wx.FileDialog(self, "Choose a File", path, "*.*")
        if dialog.ShowModal() == wx.ID_OK:
            path = os.path.join(dialog.GetDirectory(), dialog.GetFilename())
            self.filepath.SetValue(path)
        dialog.Destroy()

    def set_text_fields(self, metadata):
        self.editorfield_series.SetValue('')
        if metadata['Series']:
            self.editorfield_series.SetValue(metadata['Series'])

        self.editorfield_writer.SetValue('')
        if metadata['Writer']:
            self.editorfield_writer.SetValue(metadata['Writer'])

        self.editorfield_penciller.SetValue('')
        if metadata['Penciller']:
            self.editorfield_penciller.SetValue(metadata['Penciller'])

        self.editorfield_genre.SetValue('')
        if metadata['Genre']:
            self.editorfield_genre.SetValue(metadata['Genre'])

        self.editorfield_year.SetValue('')
        if metadata['Year']:
            self.editorfield_year.SetValue(metadata['Year'])

        self.editorfield_month.SetValue('')
        if metadata['Month']:
            self.editorfield_month.SetValue(metadata['Month'])

        self.editorfield_day.SetValue('')
        if metadata['Day']:
            self.editorfield_day.SetValue(metadata['Day'])


    def load_comicinfo(self, e):
        filepath = self.filepath.GetValue()
        metadata = InfoEditor.get_metadata(filepath)
        self.set_text_fields(metadata)

    def save_comicinfo(self, e):
        filepath = self.filepath.GetValue()
        metadata = self.get_metadata_from_fields()
        InfoEditor.write_metadata(filepath, metadata)

    def get_metadata_from_fields(self):

        metadata = {}

        metadata['Series'] = self.editorfield_series.GetValue()
        metadata['Writer'] = self.editorfield_writer.GetValue()
        metadata['Penciller'] = self.editorfield_penciller.GetValue()
        metadata['Genre'] = self.editorfield_genre.GetValue()
        metadata['Year'] = self.editorfield_year.GetValue()
        metadata['Month'] = self.editorfield_month.GetValue()
        metadata['Day'] = self.editorfield_day.GetValue()

        return metadata

class Droptarget(wx.FileDropTarget):

    def __init__(self, window):
        wx.FileDropTarget.__init__(self)
        self.window = window

    def OnDropFiles(self, x, y, files):
        if len(files) == 1:
            self.window.filepath.SetValue(files[0])
        else:
            wx.MessageBox("I can't do with multiple files!", 'Error')

if __name__ == '__main__':
    app = ComicInfoGetter()
    app.MainLoop()

InfoEditor.py

import zipfile
import tempfile
import xml.etree.ElementTree as etree
import io
import shutil
import os


def get_metadata(filepath):
    """Return Comics Metadata as Dictionary"""
    def get_full_series():
        series = root.findtext('Series')
        number = root.findtext('Number')
        title = root.findtext('Title')

        if series and number and title:
            series = series + ' ' + number + ' ' + title
        elif series and number and not title:
            series = series + ' ' + number
        elif series and not number and title:
            series = series + ' ' + title
        else:
            series = series
        return series

    def get_series():
        series = root.findtext('Series')
        return series

    def get_writer():
        writer = root.findtext('Writer')
        return writer

    def get_penciller():
        penciller = root.findtext('Penciller')
        return penciller

    def get_genre():
        genre = root.findtext('Genre')
        return genre

    def get_year():
        year = root.findtext('Year')
        return year

    def get_month():
        month = root.findtext('Month')
        return month

    def get_day():
        day = root.findtext('Day')
        return day

    metadata = {}

    with zipfile.ZipFile(filepath, mode='r',) as z:
        try:
            comicinfoxml = z.open('ComicInfo.xml', mode='r')
            infotree = etree.parse(comicinfoxml)
            root = infotree.getroot()
        except:
            print('There is No ComicInfo.xml or Some Error')

    metadata['Series'] = get_full_series()
    metadata['Writer'] = get_writer()
    metadata['Penciller'] = get_penciller()
    metadata['Genre'] = get_genre()
    metadata['Year'] = get_year()
    metadata['Month'] = get_month()
    metadata['Day'] = get_day()

    print(metadata)
    return metadata


def write_metadata(filepath, metadata):
    """Write metadata to Comic Archive"""

    def generate_xml():
        """Generate ComicInfo.xml to Write"""
        g = io.StringIO()

        with zipfile.ZipFile(filepath, mode='r',) as z:
            comicinfoxml = z.open('ComicInfo.xml', mode='r')
            infotree = etree.parse(comicinfoxml)
            root = infotree.getroot()

            try:
                root.find('Number').remove
            except:
                pass
            try:
                root.find('Title').remove
            except:
                pass
            for k, v in metadata.items():
                if (v != '') & (root.find(k) is not None):
                    root.find(k).text = v
                elif (v != '') & (root.find(k) is None):
                    newelement = etree.Element(k)
                    newelement.text = v
                    root.append(newelement)
                elif v == '':
                    try:
                        root.find(k).remove
                    except:
                        pass
            infotree.write(g, encoding='unicode')

        return g

    gx = generate_xml()

    remove_comicinfo_from_zip(filepath)

    with zipfile.ZipFile(filepath, mode='a',) as z:
        z.writestr('ComicInfo.xml', gx.getvalue())

def remove_comicinfo_from_zip(filepath):
    tempdir = tempfile.mkdtemp()
    try:
        tempname = os.path.join(tempdir, 'new.zip')
        with zipfile.ZipFile(filepath, 'r') as zipread:
            with zipfile.ZipFile(tempname, 'w') as zipwrite:
                for item in zipread.infolist():
                    if item.filename != 'ComicInfo.xml':
                        data = zipread.read(item.filename)
                        zipwrite.writestr(item, data)
        shutil.move(tempname, filepath)
    finally:
        shutil.rmtree(tempdir)