AbstractGenerator

Complexity

Total Complexity

40

Size/Duplication

Total Lines	261
Duplicated Lines	0 %

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	8	1
B	write_docx()	0	28	5
A	_insert_image()	0	12	3
A	read_xlsx()	0	4	1
F	_write_doc_aini2016()	0	155	18
B	_toArray()	0	5	5
B	_write_doc_jscpb2016()	0	28	3
A	_removeParentheses()	0	12	4

# -*- coding: utf-8 -*-

"""

Generate abstract document (docx) file from table (xlsx)

by nebula


Dependency: pandas, xlrd, python-docx

"""

import pandas as pd

import docx

import math

import re

import os



class AbstractGenerator:

    def __init__(self, image_dir='', template_type='aini2016'):

        self.records = None

        self.image_dir = image_dir

        self.template_type = template_type

        self.exreg4author = re.compile(r'^([^\)]+)((?:\(.+\))*)$')

        self.exreg4affiliation = re.compile(r'^((?:\(.+\))*)(.+)$')

        self.exreg4super = re.compile(r'(\(\w+\))')

        self.exreg4italic = re.compile(r'(\<i\>\w+\</i\>)')


    def _insert_image(self, filename, image_filename):

        doc = docx.Document(filename)


        for paragraph in doc.paragraphs:

            if '[[FIGURE]]' in paragraph.text:

                #paragraph.text = ''

                run = paragraph.add_run()

                run.add_paragraph()

                inline_shape = run.add_picture(image_filename, width=docx.shared.Pt(300))

                run.add_paragraph()


        doc.save(filename)


    def _toArray(self, text, delim):

        if isinstance(text, float) and math.isnan(text):

            return []

        items = text.split(delim)

        return [item for item in items if item.strip()]


    def _removeParentheses(self, text):

        exreg = re.compile(r'\((\w+)\)')

        nums = exreg.split(text)

        num = ''

        for n in nums:

             n = n.strip()

             if n == '':

                 continue

             if num != '':

                 num += ', '

             num += n

        return num


    def read_xlsx(self, filename):

        print('Reading: %s' % filename)

        exls = pd.ExcelFile(filename)

        self.records = exls.parse()


    def write_docx(self, filename, template=None):

        print('Writing: %s' % filename)


        if template is not None:

            doc = docx.Document(template)

        else:

            doc = docx.Document()


        first = True

        for i in self.records.index:

            if first == True:

                section = doc.sections[0]

            else:

                section = doc.add_section(docx.enum.section.WD_SECTION.NEW_PAGE)

            section.orientation = docx.enum.section.WD_ORIENT.PORTRAIT

            section.page_height = docx.shared.Mm(297)

            section.page_width = docx.shared.Mm(210)

            section.top_margin = docx.shared.Mm(20)

            section.right_margin = docx.shared.Mm(20)

            section.left_margin = docx.shared.Mm(20)

            section.bottom_margin = docx.shared.Mm(15)

            if self.template_type == 'aini2016':

                self._write_doc_aini2016(doc, self.records.loc[i])

            else:

                self._write_doc_jscpb2016(doc, self.records.loc[i])

            first = False


        doc.save(filename)


    def _write_doc_jscpb2016(self, doc, record):

        print('"%s"' % record['title'])


        # Title

        p = doc.add_paragraph(record.title)

        p.runs[0].font.size = docx.shared.Pt(12)

        p.runs[0].bold = True


        # Authors

        p = doc.add_paragraph()

        author_list = self.exreg4super.split(record.authors)

        for j in range(len(author_list)):

            if j & 1:

                p.add_run(author_list[j]).font.superscript = True

            else:

                p.add_run(author_list[j])


        # Affiliations

        p = doc.add_paragraph(record.affiliations)

        p.runs[0].font.size = docx.shared.Pt(9)

        p.runs[0].italic = True


        # Abstract Body

        p = doc.add_paragraph(record.abstract)


        # keywords

        p = doc.add_paragraph('Keywords: ')

        p.add_run(record.keywords).italic = True



    def _write_doc_aini2016(self, doc, record):

        print('"%s"' % record['Title'])

        exreg4num = re.compile(r'\((\w+)\)')


        font = doc.styles['Normal'].font

        font.size = docx.shared.Pt(10)

        font.name = 'Lucida Grande'


        # Program Number

        #p = doc.add_paragraph()

        #p.paragraph_format.line_spacing = docx.shared.Pt(12)

        #p.paragraph_format.space_after = docx.shared.Pt(5)

        #r = p.add_run(record['Program No.'].strip())


        # Title

        p = doc.add_paragraph()

        p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER

        p.paragraph_format.space_before = docx.shared.Pt(25)

        p.paragraph_format.space_after = docx.shared.Pt(14)

        r = p.add_run(record['Title'].strip())

        r.font.size = docx.shared.Pt(12)

        r.bold = True

        r.italic = True


        # Authors

        p = doc.add_paragraph()

        p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER

        p.paragraph_format.line_spacing = docx.shared.Pt(12)

        p.paragraph_format.space_after = docx.shared.Pt(12)

        authors = self._toArray(record['Name'], '\n')

        first = True

        for author in authors:

            m = self.exreg4author.match(author)

            if first == False:

                p.add_run(', ').bold = True

            name = m.group(1).strip().replace(' ', '\u00A0')

            num = self._removeParentheses(m.group(2).strip())

            p.add_run(name).bold = True

            if num != '':

                r = p.add_run(num)

                r.bold = True

                r.font.superscript = True

            first = False

        p.add_run('\n')


        # Affiliation

        affiliations = self._toArray(record['Affiliation'], '\n')

        first = True

        for affiliation in affiliations:

            m = self.exreg4affiliation.match(affiliation)

            if first == False:

                p.add_run(', ')

            num = self._removeParentheses(m.group(1).strip())

            name = m.group(2).strip()

            if num != '':

                r = p.add_run(num + ' ')

                r.font.superscript = True

            p.add_run(name)

            first = False

        p.add_run('\n' + record['e-mail'])


        # DOI

        p = doc.add_paragraph('DOI:' + record['DOI'].strip())

        p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER

        p.paragraph_format.space_after = docx.shared.Pt(12)


        # Abstract Body

        items = self._toArray(record['Abstract'], '\n')

        for item in items:

            p = doc.add_paragraph(item)

            p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.JUSTIFY

            p.paragraph_format.line_spacing = docx.shared.Pt(10)


        p.paragraph_format.space_after = docx.shared.Pt(12)


        # Figure

        doc.add_picture(os.path.join(self.image_dir, record['Figure file Name']))

        p = doc.paragraphs[-1]

        p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER


        # Figure Comment

        items = self._toArray(record['Figure comment'], '\n')

        first = True

        for item in items:

            p = doc.add_paragraph()

            p.paragraph_format.line_spacing = docx.shared.Pt(10)

            p.paragraph_format.space_after = docx.shared.Pt(0)

            p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.JUSTIFY

            if first:

                p.add_run('Figure: ').bold = True

                first = False

            p.add_run(item)

        p.paragraph_format.space_after = docx.shared.Pt(14)


        # References

        items = self._toArray(record['References'], '\n')

        first = True

        for item in items:

            if first:

                p = doc.add_paragraph()

                p.paragraph_format.line_spacing = docx.shared.Pt(11)

                p.paragraph_format.space_after = docx.shared.Pt(0)

                p.add_run('References:').bold = True

                first = False

            p = doc.add_paragraph()

            p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.JUSTIFY

            p.paragraph_format.line_spacing = docx.shared.Pt(10)

            p.paragraph_format.space_after = docx.shared.Pt(0)

            p.add_run(item)

        p.paragraph_format.space_after = docx.shared.Pt(10)


        # Acknowledgement

        items = self._toArray(record['Acknowledgement'], '\n')

        first = True

        for item in items:

            p = doc.add_paragraph()

            p.paragraph_format.line_spacing = docx.shared.Pt(10)

            p.paragraph_format.space_after = docx.shared.Pt(0)

            p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.JUSTIFY

            if first:

                p.add_run('Ackknowledgement: ').bold = True

                first = False

            p.add_run(item)

        p.paragraph_format.space_after = docx.shared.Pt(10)


        # Funding

        items = self._toArray(record['Funding'], '\n')

        first = True

        for item in items:

            p = doc.add_paragraph()

            p.paragraph_format.line_spacing = docx.shared.Pt(10)

            p.paragraph_format.space_after = docx.shared.Pt(0)

            p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.JUSTIFY

            if first:

                p.add_run('Funding: ').bold = True

                first = False

            p.add_run(item)

        p.paragraph_format.space_after = docx.shared.Pt(10)


        # Citation

        p = doc.add_paragraph()

        p.paragraph_format.line_spacing = docx.shared.Pt(10)

        p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.JUSTIFY

        p.add_run('Citation: ').bold = True

        author_tmp = ''

        first = True

        for author in authors:

            m = self.exreg4author.match(author)

            if first == False:

                author_tmp += ', '

            author_tmp += m.group(1).strip()

            first = False

        p.add_run(author_tmp + ' (2016). ' + record['Title'] + '. ')

        p.add_run('Advances in Neuroinformatics IV. ').italic = True

        p.add_run('AINI 2016 and INCF Nodes Workshop Abstract: ' + record['Program No. Long'].strip() + '. DOI:' + record['DOI'].strip())



if __name__ == '__main__':

    img_dir = './image'

    input_xlsx = 'input.xlsx'

    output_docx = 'output.docx'

    template_docx = './template/aini2016.docx'