1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
import os
import sys
from docutils import nodes
from docutils.parsers.rst import Directive
from sphinx import addnodes
from sphinx.errors import SphinxError
from sphinx.util.nodes import split_explicit_title, process_index_entry, set_role_source_info
from htc_helpers import *
ATTRIBUTE_FILES = {}
AD_TYPE_FILES = {
"ACCOUNTING" : "accounting-classad-attributes.html",
"ADDED_COLLECTOR" : "classad-attributes-added-by-collector.html",
"COLLECTOR" : "collector-classad-attributes.html",
"DAEMONCORE" : "daemon-core-statistics-attributes.html",
"DEFRAG" : "defrag-classad-attributes.html",
"GRID" : "grid-classad-attributes.html",
"JOB" : "job-classad-attributes.html",
"MACHINE" : "machine-classad-attributes.html",
"MASTER" : "daemon-master-classad-attributes.html",
"NEGOTIATOR" : "negotiator-classad-attributes.html",
"SCHEDD" : "scheduler-classad-attributes.html",
"SUBMITTER" : "submitter-classad-attributes.html",
}
def map_attrs(dir: str):
attrs = {}
files_dir = os.path.join(dir, "classad-attributes")
for ad_file in os.listdir(files_dir):
if ad_file[-4:] != ".rst":
continue
path = os.path.join(files_dir, ad_file)
with open(path, "r", encoding="utf-8") as f:
for line in f:
line = line.strip()
for attr in get_all_defined_role("classad-attribute-def", line):
html_file = ad_file.replace(".rst", ".html")
if attr in attrs:
attrs[attr].append(html_file)
else:
attrs.update({attr : [html_file]})
return attrs
def dump(obj):
for attr in dir(obj):
print("obj.%s = %r" % (attr, getattr(obj, attr)))
def classad_attr_role(name, rawtext, text, lineno, inliner, options={}, content=[]):
docname = inliner.document.settings.env.docname
root_dir = get_rel_path_to_root_dir(inliner)[:-1]
attr_name, attr_info = custom_ext_parser(text)
details = extra_info_parser(attr_info) if attr_info != "" else None
ad_type = details.get("TYPE").upper() if details is not None else None
attr_index = details.get("INDEX", "") if details is not None else ""
filename = AD_TYPE_FILES.get(ad_type) if ad_type in AD_TYPE_FILES else None
if attr_name not in ATTRIBUTE_FILES:
warn(f"{docname} @ {lineno} | ClassAd Attribute '{attr_name}' not defined in any ClassAd Documentation files")
filename = "classad-types.html"
elif filename is None:
filename = ATTRIBUTE_FILES[attr_name][0]
if len(ATTRIBUTE_FILES[attr_name]) > 1:
warn(f"{docname} @ {lineno} | ClassAd Attribute '{attr_name}' is defined in multiple files. Defaulting to {filename}")
ref_link = f"href=\"{root_dir}/classad-attributes/{filename}#{attr_name}\""
return make_ref_and_index_nodes(name, attr_name, attr_index,
ref_link, rawtext, inliner, lineno, options)
def setup(app):
global ATTRIBUTE_FILES
ATTRIBUTE_FILES = map_attrs(app.srcdir)
app.add_role("ad-attr", classad_attr_role)
|
