快捷導(dǎo)航

自定義PowerShell控制臺(tái)提示符風(fēng)格的方法

更新時(shí)間：2014年04月17日 13:36:12 作者：

這篇文章主要介紹了自定義PowerShell控制臺(tái)提示符風(fēng)格的方法,需要的朋友可以參考下

PowerShell控制臺(tái)的默認(rèn)命令行提示為PS+（當(dāng)前路徑）+箭頭，形如：

PS C:\Users\非苔>

前面加PS是PowerShell的標(biāo)志，為了和傳統(tǒng)的CMD控制臺(tái)區(qū)別開(kāi)來(lái)。后面加當(dāng)前路徑和箭頭是為了和一般的命令行保持一致。這樣的設(shè)計(jì)本來(lái)無(wú)可厚非，但是如果我在寫(xiě)博客時(shí)，需要執(zhí)行某個(gè)命令然后連同結(jié)果一起復(fù)制出來(lái)，麻煩來(lái)了。比如我要演示Get-Date，打開(kāi)控制臺(tái)，敲入命令：

復(fù)制代碼代碼如下:

PS C:\Users\非苔> Get-Date

2014年4月16日 14:48:47

結(jié)果正確，但是看起來(lái)不爽，這個(gè)路徑是多余的，它會(huì)分散讀者的注意力。有人說(shuō)，你可以定位到C盤根目錄下：

復(fù)制代碼代碼如下:

PS C:\> Get-Date

2014年4月16日 14:52:56

這樣把路徑弱化了，好多了，但是仍舊不爽。“C:\”還是多余的，我是不是有點(diǎn)疑似強(qiáng)迫癥啊，如果改成這樣呢？

復(fù)制代碼代碼如下:

PS> Get-Date

2014年4月16日 14:57:07

這樣好，既能體現(xiàn)PowerShell的風(fēng)格，又無(wú)冗余信息。甚至還能繼續(xù)簡(jiǎn)化，改成：

復(fù)制代碼代碼如下:

>Get-Date

2014年4月16日 14:59:21

Get-Date

2014年4月16日 14:59:48

以上討論的關(guān)鍵其實(shí)就是控制臺(tái)提示風(fēng)格，下面討論如何自定義。

知識(shí)點(diǎn)

控制臺(tái)的提示風(fēng)格由自動(dòng)化函數(shù)prompt控制。函數(shù)的定義默認(rèn)為：

復(fù)制代碼代碼如下:

"PS $($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) "
# .Link
# http://go.microsoft.com/fwlink/?LinkID=225750
# .ExternalHelp System.Management.Automation.dll-help.xml

我們所要做的就是覆蓋這個(gè)函數(shù)。

1.最好的更改應(yīng)當(dāng)是：提示支持多種風(fēng)格，可以來(lái)回切換。既然要切換，就得有個(gè)變量來(lái)存儲(chǔ)狀態(tài)。這個(gè)變量最好是全局的，并且是只讀的。
2.對(duì)于PowerShell內(nèi)置的自動(dòng)化函數(shù)和自動(dòng)化變量的更改只有在當(dāng)前控制臺(tái)會(huì)話有效，所以最好將更改存放在Profile配置文件中。

效果圖：

實(shí)現(xiàn)腳本：

復(fù)制代碼代碼如下:

Function Prompt()
{
    switch ($PSPromptMode)
    {
        'Cmd'
        {
          "$($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) "
        }

        'Arrow'
        {
          '> '
        }

        'None'
        {
          ' '
        }
        'Simple'
        {
         'PS> '
        }
        Default
        {
         "PS $($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) "
        }
    }
}

<#
.Synopsis
   設(shè)置控制臺(tái)提示風(fēng)格
.DESCRIPTION
   設(shè)置控制臺(tái)提示風(fēng)格，支持五種風(fēng)格：Normal，Cmd，Arrow，Simple，None
#>
Function Set-Prompt
{
    param(
    [Parameter(Mandatory=$true)]
    [ValidateSet('Normal','Cmd','Arrow','Simple', 'None',IgnoreCase = $true)]
    $Mode
    )
    $varPSPromptMode = (Get-Variable 'PSPromptMode'  -ea SilentlyContinue)
    #配置變量不存在則新建
    if ( $varPSPromptMode -eq $null)
    {
        New-Variable -Name 'PSPromptMode' -Value $Mode -Scope 'Global'
        $varPSPromptMode = Get-Variable -Name 'PSPromptMode'
        $varPSPromptMode.Description = '提示函數(shù)配置變量'

        #限制配置變量的取值集合
        $varPSPromptModeAtt = New-Object System.Management.Automation.ValidateSetAttribute('Normal','Cmd','Arrow','Simple','None')
        $varPSPromptMode.Attributes.Add($varPSPromptModeAtt)

        #限制配置變量為只讀并且可以貫穿所有作用域ls
        $varPSPromptMode.Options = 'ReadOnly, AllScope'

    }
    #更新配置
    #只讀變量可以通過(guò)-force選項(xiàng)更新值
    Set-Variable -Name PSPromptMode -Value $Mode -Force
}