zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Factor edit_package function out of edit command.

Browse Source
This commit is contained in:
Todd Gamblin 2015-04-08 00:22:07 -07:00
parent c4a6d04131
commit ab40d0b1a4
1 changed files with 23 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
lib/spack/spack/cmd/edit.py
View File

@ -24,7 +24,6 @@
############################################################################## ##############################################################################
import os import os
import string import string
from contextlib import closing
import llnl.util.tty as tty import llnl.util.tty as tty
from llnl.util.filesystem import mkdirp, join_path from llnl.util.filesystem import mkdirp, join_path
@ -54,6 +53,27 @@ def install(self, spec, prefix):
""") """)
def edit_package(name, force=False):
path = spack.db.filename_for_package_name(name)
if os.path.exists(path):
if not os.path.isfile(path):
tty.die("Something's wrong. '%s' is not a file!" % path)
if not os.access(path, os.R_OK|os.W_OK):
tty.die("Insufficient permissions on '%s'!" % path)
elif not force:
tty.die("No package '%s'. Use spack create, or supply -f/--force "
"to edit a new file." % name)
else:
mkdirp(os.path.dirname(path))
with open(path, "w") as pkg_file:
pkg_file.write(
package_template.substitute(
name=name, class_name=mod_to_class(name)))
spack.editor(path)
def setup_parser(subparser): def setup_parser(subparser):
subparser.add_argument( subparser.add_argument(
'-f', '--force', dest='force', action='store_true', '-f', '--force', dest='force', action='store_true',
@ -80,22 +100,7 @@ def edit(parser, args):
# By default open the directory where packages or commands live. # By default open the directory where packages or commands live.
if not name: if not name:
path = spack.packages_path path = spack.packages_path
spack.editor(path)
else: else:
path = spack.db.filename_for_package_name(name) edit_package(name, args.force)
if os.path.exists(path):
if not os.path.isfile(path):
tty.die("Something's wrong. '%s' is not a file!" % path)
if not os.access(path, os.R_OK|os.W_OK):
tty.die("Insufficient permissions on '%s'!" % path)
elif not args.force:
tty.die("No package '%s'. Use spack create, or supply -f/--force "
"to edit a new file." % name)
else:
mkdirp(os.path.dirname(path))
with closing(open(path, "w")) as pkg_file:
pkg_file.write(
package_template.substitute(name=name, class_name=mod_to_class(name)))
# If everything checks out, go ahead and edit.
spack.editor(path)