|
import docx |
|
|
|
from src.domain.container import Container |
|
from src.domain.paragraph import Paragraph |
|
from src.domain.styles import Styles |
|
import shutil |
|
import os |
|
|
|
|
|
class Doc: |
|
|
|
def __init__(self, path='', id_=None): |
|
|
|
self.xdoc = docx.Document(path) |
|
self.title = path.split('/')[-1] |
|
self.name = self.title.split('.')[0] |
|
self.id_ = id(self) |
|
self.path = path |
|
paragraphs = [Paragraph(xp, self.id_, i) for (i, xp) in enumerate(self.xdoc.paragraphs)] |
|
self.container = Container(paragraphs, father=self) |
|
self.styles = Styles(self.xdoc.styles) |
|
self.tasks = [c.get_fulltask(self.container.one_liner) for c in self.container.containers if c.task] |
|
|
|
def copy(self, new_doc_path): |
|
shutil.copyfile(self.path, new_doc_path) |
|
new_doc = Doc(new_doc_path) |
|
new_doc.save_as_docx(new_doc_path) |
|
return new_doc |
|
|
|
def clear(self): |
|
os.remove(self.path) |
|
|
|
def apply_template(self, template, add_front_pages=True): |
|
if add_front_pages: |
|
self.add_front_pages_from(template) |
|
log = self.styles.apply_from(template.styles) |
|
self.save_as_docx() |
|
return log |
|
|
|
def copy_one_style(self, src_style_name: str, dest_style_name: str, template): |
|
style_dest = template.styles.get_style_from_name(dest_style_name) |
|
src_style = self.styles.get_style_from_name(src_style_name) |
|
log = self.styles.copy_one_style(src_style, style_dest) |
|
return log |
|
|
|
def get_different_styles_with_template(self, template): |
|
different_styles = self.styles.get_different_styles(template.styles) |
|
return different_styles |
|
|
|
def save_as_docx(self, path: str = ''): |
|
path = path if path else self.path |
|
self.path = path |
|
self.xdoc.save(path) |
|
|
|
def add_front_pages_from(self, src_doc): |
|
src_paragraphs = [p for p in src_doc.xdoc.paragraphs] |
|
src_paragraphs.reverse() |
|
for p in src_paragraphs: |
|
self.xdoc.paragraphs[0].insert_paragraph_before(text=p.text, style=p.style) |
|
paragraphs = [Paragraph(xp, self.id_, i) for (i, xp) in enumerate(self.xdoc.paragraphs)] |
|
self.container = Container(paragraphs, father=self) |
|
|
|
def get_blocks(self): |
|
|
|
def from_list_to_str(index_list): |
|
index_str = str(index_list[0]) |
|
for el in index_list[1:]: |
|
index_str += '.' + str(el) |
|
return index_str |
|
|
|
blocks = self.container.blocks |
|
for block in blocks: |
|
block.doc = self.title |
|
if block.level == 0: |
|
blocks.remove(block) |
|
block.index = from_list_to_str(block.index) |
|
return blocks |
|
|
|
|
|
@property |
|
def structure(self): |
|
|
|
return self.container.structure |
|
|
|
def replace_tasks(self, resolutions: [str]): |
|
if len(resolutions) == len(self.tasks): |
|
p_tasks = [p for p in self.get_paragraphs() if p.type == 'task'] |
|
for p, r in zip(p_tasks, resolutions): |
|
p.set_text(r) |
|
else: |
|
print(f"résolutions : {len(resolutions)} != {len(self.tasks)} tasks") |
|
return self |
|
|
|
def get_paragraphs(self): |
|
return self.container.all_paragraphs |
|
|
