def __init__(self, repo):
DataBase.__init__(self, "drawings", repo)
self.dimdrawings = []
self.condrawings = []
self.conlocations = []
self.dimdrawing_classes = Links()
self.condrawings_classes = BipartiteLinks()
self.conlocations_condrawings = BipartiteLinks()
self.collection_dimdrawings = Links()
self.collection_condrawings = Links()
if not exists(join(self.backend_root)):
e = MalformedRepositoryError("drawings directory does not exist")
e.set_repo_path(path)
raise e
for coll in listdir(self.backend_root):
basefilename = join(self.backend_root, coll, "%s.base" % coll)
if not exists(basefilename):
#skip directory that is no collection
continue
if coll not in repo.collections:
raise MalformedRepositoryError(
"Drawings for unknown collection found: %s " % coll)
base_info = list(yaml.load_all(open(basefilename, "r", "utf8")))
if len(base_info) != 1:
raise MalformedCollectionError(
"Not exactly one YAML document found in file %s" %
basefilename)
base_info = base_info[0]
for drawing_element in base_info:
if drawing_element["type"] == "drawing-dimensions":
draw = DrawingDimensions(drawing_element, coll,
self.backend_root)
if draw.get_svg() is None and draw.get_png() is None:
raise MalformedRepositoryError(
"No drawing files present for %s/%s" %
(coll, draw.filename))
self.dimdrawings.append(draw)
if drawing_element["classids"] == []:
raise MalformedBaseError(
"Drawing with no associated classes found")
for id in drawing_element["classids"]:
self.dimdrawing_classes.add_link(
draw, self.repo.classes[id])
self.collection_dimdrawings.add_link(
repo.collections[coll], draw)
if drawing_element["type"] == "drawing-connector":
draw = DrawingConnectors(drawing_element, coll,
self.backend_root)
if draw.get_svg() is None and draw.get_png() is None:
raise MalformedRepositoryError(
"No drawing files present for %s/%s" %
(coll, draw.filename))
if not draw.location in self.conlocations:
self.conlocations.append(draw.location)
self.conlocations_condrawings.add_link(draw.location, draw)
self.condrawings.append(draw)
for id in drawing_element["classids"]:
self.condrawings_classes.add_link(
draw, self.repo.classes[id])
self.collection_condrawings.add_link(
repo.collections[coll], draw)
