瀏覽代碼

init: restructure for quick overview and PEP8 compliance

git-svn-id: https://svn.osgeo.org/grass/grass/trunk@73350 15284696-431f-4ddb-bdfa-cd5b030d7da7
Vaclav Petras 6 年之前
父節點
當前提交
3f9656137e
共有 1 個文件被更改,包括 32 次插入19 次删除
  1. 32 19
      lib/init/grass.py

+ 32 - 19
lib/init/grass.py

@@ -1737,9 +1737,10 @@ def bash_startup(location, location_name, grass_env_file):
     if not os.getenv('HISTSIZE') and not os.getenv('HISTFILESIZE'):
     if not os.getenv('HISTSIZE') and not os.getenv('HISTFILESIZE'):
         os.environ['HISTSIZE'] = "3000"
         os.environ['HISTSIZE'] = "3000"
 
 
-    # instead of changing $HOME, start bash with: --rcfile "$LOCATION/.bashrc" ?
-    #   if so, must care be taken to explicitly call .grass.bashrc et al for
-    #   non-interactive bash batch jobs?
+    # instead of changing $HOME, start bash with:
+    #   --rcfile "$LOCATION/.bashrc" ?
+    #   if so, must care be taken to explicitly call .grass.bashrc et al
+    #   for non-interactive bash batch jobs?
     userhome = os.getenv('HOME')      # save original home
     userhome = os.getenv('HOME')      # save original home
     home = location                   # save .bashrc in $LOCATION
     home = location                   # save .bashrc in $LOCATION
     os.environ['HOME'] = home
     os.environ['HOME'] = home
@@ -1749,24 +1750,36 @@ def bash_startup(location, location_name, grass_env_file):
 
 
     f = open(bashrc, 'w')
     f = open(bashrc, 'w')
     f.write("test -r ~/.alias && . ~/.alias\n")
     f.write("test -r ~/.alias && . ~/.alias\n")
+
     if os.getenv('ISISROOT'):
     if os.getenv('ISISROOT'):
-        f.write("PS1='ISIS-GRASS %s (%s):\\w > '\n" % (grass_version, location_name))
+        # GRASS GIS and ISIS blend
+        grass_name = "ISIS-GRASS"
     else:
     else:
-        f.write("PS1='GRASS %s (%s):\\w > '\n" % (grass_version, location_name))
-
-    f.write("""grass_prompt() {
-	LOCATION="`g.gisenv get=GISDBASE,LOCATION_NAME,MAPSET separator='/'`"
-	if test -d "$LOCATION/grid3/G3D_MASK" && test -f "$LOCATION/cell/MASK" ; then
-		echo [%s]
-	elif test -f "$LOCATION/cell/MASK" ; then
-		echo [%s]
-	elif test -d "$LOCATION/grid3/G3D_MASK" ; then
-		echo [%s]
-	fi
-}
-PROMPT_COMMAND=grass_prompt\n""" % (_("2D and 3D raster MASKs present"),
-                                    _("Raster MASK present"),
-                                    _("3D raster MASK present")))
+        grass_name = "GRASS"
+    f.write("PS1='{name} {version} ({location}):\\w > '\n".format(
+        name=grass_name, version=grass_version, location=location_name))
+
+    mask2d_test = 'test -f "$LOCATION/cell/MASK"'
+    mask3d_test = 'test -d "$LOCATION/grid3/G3D_MASK"'
+
+    # double curly brackets means single one for format function
+    f.write(
+        """grass_prompt() {{
+    LOCATION="`g.gisenv get=GISDBASE,LOCATION_NAME,MAPSET separator='/'`"
+    if {mask2d_test} && {mask3d_test} ; then
+        echo [{both_masks}]
+    elif {mask2d_test} ; then
+        echo [{mask2d}]
+    elif {mask3d_test} ; then
+        echo [{mask3d}]
+    fi
+}}
+PROMPT_COMMAND=grass_prompt\n""".format(
+            both_masks=_("2D and 3D raster MASKs present"),
+            mask2d=_("Raster MASK present"),
+            mask3d=_("3D raster MASK present"),
+            mask2d_test=mask2d_test, mask3d_test=mask3d_test
+            ))
 
 
     # read other settings (aliases, ...) since environmental variables
     # read other settings (aliases, ...) since environmental variables
     # have been already set by load_env(), see #3462
     # have been already set by load_env(), see #3462