返回列表 发帖

[问题求助] 磁盘序号和分区表格式的vbs代码有点小问题,能否合并成一个正常的vbs?

2个类似功能的vbs代码,各自都有点小问题,能否合并成一个正常的vbs?
总体要求:磁盘序号顺序显示。并且分区表格式显示正确。
diskinfo.vbs,这个代码问题是,分区表GPT的会错误的识别成MBR。但是磁盘序列号显示正常,比如磁盘0、磁盘1、磁盘2....
Main
Sub Main()
  Dim objWMI, colDisks
  Set objWMI = GetObject("winmgmts:\\.\root\cimv2")
  Set colDisks = objWMI.ExecQuery("select * from Win32_DiskDrive where MediaType like 'fixed%'")
  ' 获取硬盘最大序号
  nIndexMax = 0
  For Each objDisk in colDisks
    If Cint(objDisk.Index) > nIndexMax Then nIndexMax = Cint(objDisk.Index)
  Next
  ' 按序号取信息
  For i = 0 To nIndexMax
    For Each objDisk in colDisks
      ' 只取该序号信息
      If objDisk.Index = i Then
        s = s & "硬盘" & objDisk.Index & ":" & vbTab
        s = s & "大小:" & GetSize(objDisk.Size) & vbTab
        's = s & "型号:" & objDisk.Caption &  vbCrLf
        s = s & "型号:" & objDisk.Caption & vbTab
        s = s & "分区表:" & IsGPT(objDisk.Index) & vbCrLf
        strDiskID = Replace(objDisk.DeviceID, "\", "\\")
        Set colPartitions = objWMI.ExecQuery _
            ("ASSOCIATORS OF {Win32_DiskDrive.DeviceID=""" & strDiskID & """}" _
            & " where AssocClass=Win32_DiskDriveToDiskPartition")
        For Each objPartition in colPartitions
          strPartId = objPartition.DeviceID
          Set colLogicalDisks = objWMI.ExecQuery _
              ("ASSOCIATORS OF {Win32_DiskPartition.DeviceID=""" & strPartId _
              & """}  where AssocClass=Win32_LogicalDiskToPartition")
          For Each objLogicalDisk in colLogicalDisks
            size = objLogicalDisk.Size
            free = objLogicalDisk.Freespace
            used = FormatNumber((size - free) / size * 100, 2, true) & "%"
            s = s & "分区" & objLogicalDisk.DeviceID & vbTab
            s = s & "大小:" & GetSize(size) & vbTab
            s = s & "剩余:" & GetSize(free) & vbTab
            s = s & "使用率:" & used & vbCrLf
          Next
        Next
        s = s & vbCrLf
      End If
    Next
  Next
  WScript.Echo s
End Sub
' 格式化
Function GetSize(intSize)
  If intSize/1024/1024 > 1024 Then
    GetSize = FormatNumber(intSize/1024/1024/1024, 2, true) & "GB"
  Else
    GetSize = FormatNumber(intSize/1024/1024, 2, true) & "MB"
  End If
End Function
' 获取指定硬盘的分区表类型(GPT/MBR)
Function IsGPT(ByVal nDiskIndex)
  IsGPT = "MBR"
  Dim wso, sLogFile, sText
  Set wso = CreateObject("WScript.Shell")
  sLogFile = wso.ExpandenVironmentStrings("%temp%\diskpart.log")
  wso.Run "cmd /c ""chcp 437 & cls & (echo list disk | diskpart | find /i ""Disk " & nDiskIndex & """) >""" & sLogFile & """ "" ", 0, False
  Call TxtFile(sLogFile, 1, -2, sText)
  If Trim(Right(sText,3)) = "*" Then IsGPT = "GPT"
End Function
' 对文本指定编码进行读写操作2
'nRW: 1只读, 2只写, 8追加 'nCharset: -2(系统), -1(Unicode), 0(ASCII)
Sub TxtFile(ByVal FileName, ByVal nRW, ByVal nCharset, ByRef sText)
  Dim fso :  Set fso = CreateObject("Scripting.filesystemobject")
  If sText <> "" And (nRW = 2 Or nRW = 8) Then
    fso.OpenTextFile(FileName, nRW, True, nCharset).Write sText
  ElseIf fso.FileExists(FileName) And nRW = 1 Then
    If fso.GetFile(FileName).Size > 0 Then _
      sText = fso.OpenTextFile(FileName, nRW, False, nCharset).ReadAll
  End If
End Sub
' 以管理员身份运行
Sub RunAsAdmin()
  Dim objItems, objItem, strVer, nVer
  Set objItems = GetObject("winmgmts:").InstancesOf("Win32_OperatingSystem")
  For Each objItem In objItems
    strVer = objItem.Version
  Next
  nVer = Split(strVer, ".")(0) & Split(strVer, ".")(1)
  If nVer >= 60 Then
    Dim oShell, oArg, strArgs
    Set oShell = CreateObject("Shell.Application")
    If Not WScript.Arguments.Named.Exists("ADMIN") Then
      For Each oArg In WScript.Arguments
        strArgs = strArgs & " """ & oArg & """"
      Next
      strArgs = strArgs & " /ADMIN:1"
      Call oShell.ShellExecute("WScript.exe", """" & WScript.ScriptFullName & """" & strArgs, "", "runas", 1)
      Set oShell = Nothing
      WScript.Quit(0)
    End If
    Set oShell = Nothing
  End If
End SubCOPY

引导方式-硬盘格式.vbs,这个代码问题是:磁盘序列号排列会乱序。但是分区表格式显示正常。
Set ws = CreateObject("Wscript.Shell")
Set fso = CreateObject("Scripting.FileSystemObject")
If LCase(Right(WSH.FullName, 11)) = "wscript.exe" Then
    ws.run "cscript.exe -nologo """ & WSH.ScriptFullName & """", 0
    WSH.Quit
End If
Dim arr()
Set oExec = ws.Exec("cmd /c ""(echo;list disk|diskpart)|findstr /irc:""[0-9][0-9]*  *[tgmk]b"" "" ")
Do While oExec.StdOut.AtEndOfStream <> true
    str = Trim(oExec.StdOut.ReadLine)
    num = CInt(Split(str, " ")(1))
    ReDim Preserve arr(num)
    If InStrRev(str, "*") >0 Then
        arr(num) = "GPT"
    Else
        arr(num) = "MBR"
    End If
Loop
Dim objWMI, colDisks
Set objWMI = GetObject("winmgmts:\\.\root\cimv2")
Set colDisks = objWMI.ExecQuery("select * from Win32_DiskDrive where MediaType like 'fixed%'")
For Each objDisk in colDisks
    s = s & "硬盘" & objDisk.Index & ":" & vbTab
    s = s & "大小:" & GetSize(objDisk.Size) & vbTab
    s = s & "型号:" & objDisk.Caption & vbTab
    s = s & "分区表:" & arr(CInt(objDisk.Index)) & vbCrLf
    strDiskID = Replace(objDisk.DeviceID, "\", "\\")
    Set colPartitions = objWMI.ExecQuery _
        ("ASSOCIATORS OF {Win32_DiskDrive.DeviceID=""" & strDiskID & """}" _
        & " where AssocClass=Win32_DiskDriveToDiskPartition")
    For Each objPartition in colPartitions
      strPartId = objPartition.DeviceID
      Set colLogicalDisks = objWMI.ExecQuery _
          ("ASSOCIATORS OF {Win32_DiskPartition.DeviceID=""" & strPartId & """}  where AssocClass=Win32_LogicalDiskToPartition")
      For Each objLogicalDisk in colLogicalDisks
        size = objLogicalDisk.Size
        free = objLogicalDisk.Freespace
        used = FormatNumber((size - free) / size * 100, 2, true) & "%"
        s = s & "分区" & objLogicalDisk.DeviceID & vbTab
        s = s & "大小:" & GetSize(size) & vbTab
        s = s & "剩余:" & GetSize(free) & vbTab
        s = s & "使用率:" & used & vbCrLf
      Next
    Next
    s = s & vbCrLf
Next
'Msgbox s
WScript.Echo s
Function GetSize(intSize)
  If intSize/1024/1024 > 1024 Then
    GetSize = FormatNumber(intSize/1024/1024/1024, 2, true) & "GB"
  Else
    GetSize = FormatNumber(intSize/1024/1024, 2, true) & "MB"
  End If
End FunctionCOPY

TOP

回复 2# ygqiang


    第26行代码替换为第一段代码的6—15行,第49行代码后添加第一段代码的41—42行

TOP

本帖最后由 ygqiang 于 2025-3-25 22:40 编辑

回复 3# buyiyang


    谢谢,初步测试成功。

TOP

本帖最后由 ygqiang 于 2025-3-25 23:00 编辑

回复 3# buyiyang


    修改后的代码。分区表类型2.vbs,测试结果如下。
能否进一步修改vbs代码,实现:
首先显示当前系统启动盘。然后按照顺序显示其它盘。
比如,总共4个磁盘。
如果当前启动盘是磁盘1,磁盘显示顺序就是1,0,2,3
如果当前启动盘是磁盘2,磁盘显示顺序就是2,0,1,3
谢谢

TOP

返回列表