"""Algorithms to generate files.
"""
-from os.path import isfile
+import re
+
+from os.path import join
from collections import OrderedDict
import logging
u"\n .. raw:: latex\n\n"
u" \\csvautolongtable{{{file_latex}}}\n\n")
+REGEX_NIC_SHORT = re.compile(r'(\d*ge\dp\d)(\D*\d*[a-z]*)-')
+
def generate_files(spec, data):
"""Generate all files specified in the specification file.
logging.error(
f" No tables to include in {table_set}. Skipping."
)
- return
+ continue
logging.info(u" Creating the test data set...")
tests = input_data.filter_data(
continue_on_error=True
)
if tests.empty:
- return
+ continue
tests = input_data.merge_data(tests)
+ tests.sort_index(inplace=True)
logging.info(u" Creating the suite data set...")
suites = input_data.filter_data(
data_set=u"suites"
)
if suites.empty:
- return
+ continue
suites = input_data.merge_data(suites)
suites.sort_index(inplace=True)
logging.info(u" Generating files...")
- file_name = u""
- sub_chapter = u"-".join(table_set.split(u"_")[-2:])
+ chapter_l1 = u""
+ chapter_l2 = u"-".join(table_set.split(u"_")[-2:])
for suite_longname, suite in suites.items():
suite_lvl = len(suite_longname.split(u"."))
if suite_lvl == start_lvl:
# Our top-level suite
- chapter = suite_longname.split(u'.')[-1]
- file_name = f"{table_set}/{chapter}.rst"
- logging.info(f" Writing file {file_name}")
- with open(file_name, u"a") as file_handler:
- file_handler.write(rst_header)
- if chapters.get(chapter, None) is None:
- chapters[chapter] = OrderedDict()
- chapters[chapter][sub_chapter] = file_name
-
- title_line = get_rst_title_char(suite[u"level"] - start_lvl + 2) * \
- len(sub_chapter)
- with open(file_name, u"a") as file_handler:
- if not (u"-ndrpdr" in suite[u"name"] or
- u"-mrr" in suite[u"name"] or
- u"-dev" in suite[u"name"]):
- file_handler.write(f"\n{sub_chapter}\n{title_line}\n")
-
- if _tests_in_suite(suite[u"name"], tests):
- for tbl_file in table_lst:
- if suite[u"name"] in tbl_file:
- file_handler.write(
- f"\n{suite[u'name']}\n{title_line}\n"
- )
- file_handler.write(
- f"\n{suite[u'doc']}\n".
- replace(u'|br|', u'\n\n -')
+ chapter_l1 = suite_longname.split(u'.')[-1]
+ if chapters.get(chapter_l1, None) is None:
+ chapters[chapter_l1] = OrderedDict()
+ if chapters[chapter_l1].get(chapter_l2, None) is None:
+ chapters[chapter_l1][chapter_l2] = OrderedDict()
+ continue
+
+ if _tests_in_suite(suite[u"name"], tests):
+ groups = re.search(REGEX_NIC_SHORT, suite[u"name"])
+ nic = groups.group(2) if groups else None
+ if nic is None:
+ continue
+ if chapters[chapter_l1][chapter_l2].get(nic, None) is None:
+ chapters[chapter_l1][chapter_l2][nic] = dict(
+ rst_file=f"{join(table_set, chapter_l1)}_{nic}.rst".
+ replace(u"2n1l-", u""),
+ tables=list()
+ )
+ for idx, tbl_file in enumerate(table_lst):
+ if suite[u"name"] in tbl_file:
+ chapters[chapter_l1][chapter_l2][nic][u"tables"].append(
+ (
+ table_lst.pop(idx),
+ suite[u"doc"].replace(u'|br|', u'\n\n -')
)
- if frmt == u"html":
- file_handler.write(
- f"\n.. include:: {tbl_file.split(u'/')[-1]}"
- f"\n"
- )
- elif frmt == u"rst":
- file_handler.write(
- RST_INCLUDE_TABLE.format(
- file_latex=tbl_file,
- file_html=tbl_file.split(u"/")[-1])
- )
- break
+ )
+ break
titles = {
# VPP Perf, MRR
u"container_memif": u"LXC/DRC Container Memif",
u"ip6": u"IPv6 Routing",
u"ip6_tunnels": u"IPv6 Tunnels",
u"l2": u"L2 Ethernet Switching",
- u"lb": u"Link Bonding",
+ u"lb": u"LoadBalancer",
u"nfv_density": u"NFV Service Density",
u"srv6": u"SRv6 Routing",
u"vm_vhost": u"KVM VMs vhost-user",
}
order_chapters = file_spec.get(u"order-chapters", None)
- if not order_chapters:
- order_chapters = chapters.keys()
- order_sub_chapters = file_spec.get(u"order-sub-chapters", None)
+ if order_chapters:
+ order_1 = order_chapters.get(u"level-1", None)
+ order_2 = order_chapters.get(u"level-2", None)
+ order_3 = order_chapters.get(u"level-3", None)
+ if not order_1:
+ order_1 = chapters.keys()
+ else:
+ order_1 = None
+ order_2 = None
+ order_3 = None
- for chapter in order_chapters:
- sub_chapters = chapters.get(chapter, None)
- if not sub_chapters:
+ for chapter_l1 in order_1:
+ content_l1 = chapters.get(chapter_l1, None)
+ if not content_l1:
continue
with open(f"{fileset_file_name}/index.rst", u"a") as file_handler:
- file_handler.write(f" {chapter}\n")
- chapter_file_name = f"{fileset_file_name}/{chapter}.rst"
- if not isfile(chapter_file_name):
- with open(chapter_file_name, u"a") as file_handler:
- title = titles.get(chapter, chapter)
- file_handler.write(
- f"{title}\n"
- f"{get_rst_title_char(2) * len(title)}\n\n"
- f".. toctree::\n\n"
- )
-
- if not order_sub_chapters:
- order_sub_chapters = sub_chapters.keys()
- for sub_chapter in order_sub_chapters:
- testbed = sub_chapters.get(sub_chapter, None)
- if not testbed:
+ file_handler.write(f" {chapter_l1}\n")
+ l1_file_name = f"{join(fileset_file_name, chapter_l1)}.rst"
+ title = titles.get(chapter_l1, chapter_l1)
+ logging.info(f" Generating {title} ...")
+ with open(l1_file_name, u"w") as file_handler:
+ file_handler.write(
+ f"{title}\n"
+ f"{get_rst_title_char(1) * len(title)}\n\n"
+ f".. toctree::\n\n"
+ )
+
+ if not order_2:
+ order_2 = chapters[chapter_l1].keys()
+ for chapter_l2 in order_2:
+ content_l2 = content_l1.get(chapter_l2, None)
+ if not content_l2:
continue
- with open(chapter_file_name, u"a") as file_handler:
- file_handler.write(
- f" ../{u'/'.join(testbed.split(u'/')[-2:])}\n"
- )
+ if not order_3:
+ order_3 = chapters[chapter_l1][chapter_l2].keys()
+ for chapter_l3 in order_3:
+ content_l3 = content_l2.get(chapter_l3, None)
+ if not content_l3:
+ continue
+ with open(l1_file_name, u"a") as file_handler:
+ item = u"/".join(content_l3[u'rst_file'].split(u'/')[-2:])
+ file_handler.write(f" ../{item}\n")
+ logging.info(f" Writing the file {content_l3[u'rst_file']}")
+ with open(content_l3[u'rst_file'], u"w+") as file_handler:
+ title = f"{chapter_l2}-{chapter_l3}"
+ file_handler.write(
+ f"{rst_header}\n"
+ f"{title}\n"
+ f"{get_rst_title_char(2) * len(title)}\n"
+ )
+ for table in content_l3[u'tables']:
+ title = table[0].split(u"/")[-1].split(u".")[0]
+ file_handler.write(
+ f"\n{title}\n"
+ f"{get_rst_title_char(3) * len(title)}\n"
+ )
+ file_handler.write(f"\n{table[1]}\n")
+ if frmt == u"html":
+ file_handler.write(
+ f"\n.. include:: {table[0].split(u'/')[-1]}"
+ f"\n"
+ )
+ elif frmt == u"rst":
+ file_handler.write(
+ RST_INCLUDE_TABLE.format(
+ file_latex=table[0],
+ file_html=table[0].split(u"/")[-1])
+ )
def file_details_split_html(file_spec, input_data):
Code Review / csit.git / blobdiff