Miroirs/slpkg
1
0
Fork 0
mirror of https://gitlab.com/dslackw/slpkg.git synced 2024-12-28 09:58:21 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix suffix errors

Browse source
This commit is contained in:
Dimitris Zlatanidis 2015-07-25 14:00:14 +03:00
parent e2f9006e3e
commit 81d6fa384d
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
slpkg/graph.py
View file

@ -40,10 +40,7 @@ def graph_deps(deps_dict, image):
"svgz", "tif", "tiff", "tk", "vml", "vmlz", "vrml", "wbmp",
"x11", "xdot", "xdot1.2", "xdot1.4", "xlib"
]
if image.split(".")[1] not in file_format:
print("Format: {0} not recognized. Use one of: {1}".format(
image.split(".")[1], " ".join(file_format)))
raise SystemExit()
format_error(image, file_format)
try:
G = pgv.AGraph(deps_dict)
G.layout(prog="fdp")
@ -53,3 +50,16 @@ def graph_deps(deps_dict, image):
if os.path.isfile(image):
print("Graph file '{0}' created".format(image))
raise SystemExit()
def format_error(image, file_format):
"""Check for file format and type
"""
try:
if image.split(".")[1] not in file_format:
print("Format: {0} not recognized. Use one of: {1}".format(
image.split(".")[1], " ".join(file_format)))
raise SystemExit()
except IndexError:
print("slpkg: error: File image suffix missing")
raise SystemExit()