def test_influence(self):
opts = ModelOptions()
opts.vizType = "inf"
mdl = parseModel(
self.readOrgFile("tests/badOrgFiles/simpleInfluence.org"))
conv = ModelToGraphVizConverter(opts)
conv.getModelAsDot(mdl)
def main():
global args
args = getArgumentParser().parse_args()
logging.getLogger().setLevel(args.logging)
logging.basicConfig(format="[%(levelname)s] %(message)s ")
logging.debug("Teams: %s", str(args.teams))
opts = ModelOptions()
opts.skipDrawingLegend = args.skipDrawingLegend
opts.skipDrawingTitle = args.skipDrawingTitle
opts.skipDrawingTeams = args.skipDrawingTeams
opts.vizType = args.vizType
opts.outputType = args.outputType
opts.teams = args.teams
opts.attributeMatches = args.attributeMatches
opts.influence = args.influence
opts.profilePictures = args.profilePictures
opts.profilePictureDirectory = args.profilePictureDirectory
opts.dpi = args.dpi
out = None
try:
mdl = parseModel(getFileContents())
converter = ModelToGraphVizConverter(opts)
out = converter.getModelAsDot(mdl)
except Exception as exception:
logging.error(str(exception))
if args.dotout:
logging.warning("Outputting graph to stdout, not running Dot")
print(out)
elif out is not None:
with tempfile.NamedTemporaryFile(
delete=args.keepDotfile) as temporaryGraphvizFile:
temporaryGraphvizFile.write(bytes(out, "UTF-8"))
temporaryGraphvizFile.flush()
logging.debug("Wrote DOT file to: %s", str(temporaryGraphvizFile.name))
outputImageFilename = args.output + "/orgviz." + args.outputType
runDot(str(temporaryGraphvizFile.name), outputImageFilename,
args.outputType)
temporaryGraphvizFile.close(
) # will delege the org file if !args.keepDotFile