增加生产环境配置文件。

This commit is contained in:
dengqichen 2025-03-25 17:19:52 +08:00
parent 7595f3b933
commit a6e47ed527

View File

@ -18,14 +18,15 @@ function Show-Menu {
Write-Host "1. View All Tasks Status" Write-Host "1. View All Tasks Status"
Write-Host "2. Set Daily Task - DEV Environment (Default: 23:00)" Write-Host "2. Set Daily Task - DEV Environment (Default: 23:00)"
Write-Host "3. Set Daily Task - UAT Environment" Write-Host "3. Set Daily Task - UAT Environment"
Write-Host "4. Run Test Now" Write-Host "4. Set Daily Task - PROD Environment"
Write-Host "5. Delete Tasks" Write-Host "5. Run Test Now"
Write-Host "6. Delete Tasks"
Write-Host "Q. Exit" Write-Host "Q. Exit"
Write-Host "`nEnter your choice (1-5, or Q to exit): " -NoNewline Write-Host "`nEnter your choice (1-6, or Q to exit): " -NoNewline
} }
function Get-TaskStatus { function Get-TaskStatus {
$environments = @("Dev", "UAT") $environments = @("Dev", "UAT", "PROD")
$found = $false $found = $false
Write-Host "`nTasks Status:" Write-Host "`nTasks Status:"
@ -59,7 +60,7 @@ function Get-TaskStatus {
} }
if (-not $found) { if (-not $found) {
Write-Host "`nNo tasks found. Please set up tasks using options 2 or 3." Write-Host "`nNo tasks found. Please set up tasks using options 2, 3 or 4."
} }
Pause-Script Pause-Script
} }
@ -205,12 +206,14 @@ function Start-TestNow {
Write-Host "`nSelect environment:" Write-Host "`nSelect environment:"
Write-Host "1. Development (dev)" Write-Host "1. Development (dev)"
Write-Host "2. UAT" Write-Host "2. UAT"
Write-Host "`nEnter your choice (1-2): " -NoNewline Write-Host "3. PROD"
Write-Host "`nEnter your choice (1-3): " -NoNewline
$envChoice = Read-Host $envChoice = Read-Host
$env = switch ($envChoice) { $env = switch ($envChoice) {
"1" { "dev" } "1" { "dev" }
"2" { "uat" } "2" { "uat" }
"3" { "prod" }
default { default {
Write-Host "`nInvalid choice. Using default (dev)" Write-Host "`nInvalid choice. Using default (dev)"
"dev" "dev"
@ -233,14 +236,16 @@ function Remove-AutomationTask {
Write-Host "`nSelect tasks to delete:" Write-Host "`nSelect tasks to delete:"
Write-Host "1. DEV Environment Task" Write-Host "1. DEV Environment Task"
Write-Host "2. UAT Environment Task" Write-Host "2. UAT Environment Task"
Write-Host "3. Both Tasks" Write-Host "3. PROD Environment Task"
Write-Host "`nEnter your choice (1-3): " -NoNewline Write-Host "4. All Tasks"
Write-Host "`nEnter your choice (1-4): " -NoNewline
$choice = Read-Host $choice = Read-Host
$tasksToDelete = switch ($choice) { $tasksToDelete = switch ($choice) {
"1" { @("Dev") } "1" { @("Dev") }
"2" { @("UAT") } "2" { @("UAT") }
"3" { @("Dev", "UAT") } "3" { @("PROD") }
"4" { @("Dev", "UAT", "PROD") }
default { default {
Write-Host "`nInvalid choice." Write-Host "`nInvalid choice."
Pause-Script Pause-Script
@ -288,8 +293,18 @@ do {
Pause-Script Pause-Script
} }
} }
"4" { Start-TestNow } "4" {
"5" { Remove-AutomationTask } Write-Host "`nEnter time for PROD task (HH:mm, e.g. 02:30): " -NoNewline
$prodTime = Read-Host
if ($prodTime -match "^([01]?[0-9]|2[0-3]):[0-5][0-9]$") {
Set-DailyTask -runTime $prodTime -environment "PROD"
} else {
Write-Host "`nInvalid time format"
Pause-Script
}
}
"5" { Start-TestNow }
"6" { Remove-AutomationTask }
"Q" { exit } "Q" { exit }
default { default {
Write-Host "`nInvalid choice. Please try again." Write-Host "`nInvalid choice. Please try again."