def main(orig_file, notebook_path):
assert(os.path.isdir(notebook_path))
assert(os.path.isfile(orig_file))
dumper = get_format('wiki').Dumper()
parser = get_format('wiki').Parser()
text = open(orig_file).read()
parsetree = parser.parse(text)
newtree = ParseTree().fromstring("<zim-tree></zim-tree>")
for para in parsetree.findall('p'):
p = Element("p")
for node in flatten_list(para, para):
p.append(node)
newtree.getroot().extend(p)
# new todo list
text = ''.join(dumper.dump(newtree)).encode('utf-8')
tomorrow = date.today() + timedelta(1)
directory = os.path.join(notebook_path, "Calendar", str("%04d" % tomorrow.year), str("%02d" % tomorrow.month))
filename = "%02d.txt" % tomorrow.day
if (not os.path.exists(directory)): os.makedirs(directory)
# write tasks to tomorrow page.
with open(os.path.join(directory, filename), 'a+') as the_file:
the_file.write(text)
# update original wiki page.
text = ''.join(dumper.dump(parsetree) ).encode('utf-8')
with open(orig_file, 'w') as the_file:
the_file.write(text)
