#!/usr/bin/python3
import sys
import json

from docx import Document
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.enum.style import WD_STYLE_TYPE
from docx.shared import Pt

x=sys.argv[1]
data=json.loads(x)

document = Document(data["doc_path"])

#Add legend only if it not exists
if document.paragraphs[0].text != data["legend"]:

	p = document.paragraphs[0].insert_paragraph_before( "" )

	#Create Style
	styles = document.styles

	if 'Custom Legend' in styles:
		style = styles['Custom Legend']
	else:
		style = styles.add_style('Custom Legend', WD_STYLE_TYPE.PARAGRAPH)

	font = style.font
	font.name = 'Calibri'
	font.size = Pt(14)

	#Set Style
	p.style = document.styles['Custom Legend']
	p.paragraph_format.alignment = WD_ALIGN_PARAGRAPH.CENTER
	p.add_run(data["legend"]).italic = True

	#Insert Break Line
	document.paragraphs[1].insert_paragraph_before("")

	document.save(data["doc_path"]);

print(json.dumps( {'status' : 'success' } ))

