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

Add spack edit -c option to edit commands.

Browse Source
This commit is contained in:
Todd Gamblin 2014-10-07 23:52:32 -07:00
parent ff546358f3
commit 319b37af0e
1 changed files with 32 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,9 +27,10 @@
from contextlib import closing from contextlib import closing
import llnl.util.tty as tty import llnl.util.tty as tty
from llnl.util.filesystem import mkdirp from llnl.util.filesystem import mkdirp, join_path
import spack import spack
import spack.cmd
from spack.util.naming import mod_to_class from spack.util.naming import mod_to_class
description = "Open package files in $EDITOR" description = "Open package files in $EDITOR"
@ -57,6 +58,9 @@ def setup_parser(subparser):
subparser.add_argument( subparser.add_argument(
'-f', '--force', dest='force', action='store_true', '-f', '--force', dest='force', action='store_true',
help="Open a new file in $EDITOR even if package doesn't exist.") help="Open a new file in $EDITOR even if package doesn't exist.")
subparser.add_argument(
'-c', '--command', dest='edit_command', action='store_true',
help="Edit the command with the supplied name instead of a package.")
subparser.add_argument( subparser.add_argument(
'name', nargs='?', default=None, help="name of package to edit") 'name', nargs='?', default=None, help="name of package to edit")
@ -64,25 +68,34 @@ def setup_parser(subparser):
def edit(parser, args): def edit(parser, args):
name = args.name name = args.name
# By default open the directory where packages live. if args.edit_command:
if not name: if not name:
path = spack.packages_path path = spack.cmd.command_path
else:
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 args.force:
tty.die("No package '%s'. Use spack create, or supply -f/--force "
"to edit a new file." % name)
else: else:
mkdirp(os.path.dirname(path)) path = join_path(spack.cmd.command_path, name + ".py")
with closing(open(path, "w")) as pkg_file: if not os.path.exists(path):
pkg_file.write( tty.die("No command named '%s'." % name)
package_template.substitute(name=name, class_name=mod_to_class(name)))
else:
# By default open the directory where packages or commands live.
if not name:
path = spack.packages_path
else:
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 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. # If everything checks out, go ahead and edit.
spack.editor(path) spack.editor(path)