#!/usr/bin/python

"""

Just a hack to convert rst to xml and then docbook . 

May not containt all the required elements of a docbook .

Just done to make it run for the specific rst for our 

sees documentation.

"""

import xml.etree.ElementTree as ET

import os 

import re

import subprocess

import os

import pkg_resources

import glob, os, re, sys

names = glob.glob('ch*.xml')

"""

chapterno=0

def convert2xml(file):

#    print folder,subfolder,file

    global chapterno

    name=file.split('/')[-1]

    name=str(chapterno)+name.split('.')[0]

#    full_file=os.path.join(folder,file)    

#    if file.endswith('.rst'):    

    print file	    

    xml_file=name+'.xml'        

    command="rst2xml.py %s > %s" %(file , xml_file)

    print command        

    a=subprocess.Popen(command , shell=True)

def walk(repo):

    global chapterno

    mainfolder='/home/amit/sttp_latest/'    

    for readline in open('index.config','r').readlines():

        chapterno+=1		

        filename=mainfolder+readline

        convert2xml(filename)

"""

def convert2docbook(xml_string):

#    xml_file_obj=open(file,'r')

#    xml_string=xml_file_obj.read() 

#    xml_file_obj.close()       

    try:

        xml_string=re.sub('<strong>','<emphasis role="strong">', xml_string)   

        xml_string=re.sub('<literal_block','<programlisting>',xml_string)

        xml_string=re.sub('</literal_block>','</programlisting>',xml_string)

        xml_string=re.sub('xml:space="preserve">','',xml_string)

        xml_string=re.sub('<paragraph>' ,'<para>',xml_string)

        xml_string=re.sub('</paragraph>' ,'</para>',xml_string)  

        xml_string=re.sub('</strong>' ,'</emphasis>',xml_string)

        chapter= ET.Element("chapter")

        article=ET.SubElement(chapter,"article")

        articleinfo=ET.SubElement(article,"articleinfo")

        tree = ET.fromstring(xml_string)

        title= ET.SubElement(articleinfo,"title")   

        try:    

            title.text=tree.items()[1][1]

        except:

                pass    

        article.insert(1,tree)

        xml_string=ET.tostring(chapter)

        xml_string=re.sub('<document ids=.*">' ,'',xml_string)

        xml_string=re.sub('</document>' ,'',xml_string)

        return xml_string

#    docbook_file=file.split('.')[0]+'.docbook'

#    f=open(docbook_file,'w')

#    f.write(xml_string)

#    xml_string=None

    except :

        pass

if __name__=='__main__':

#    repo='/home/amit/sphinxvirt/review/'

#    walk(repo)

#    convert(1,2,3)

    for name in names:

        print name        

        xml_string=open(name,'r').read()

        xml_string=convert2docbook(xml_string)

        docbook_file=name.split('.')[0]+'.docbook'

        f=open(docbook_file,'w')

        try:        

            f.write(xml_string)

        except:

                pass