mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
59 lines
1.6 KiB
PowerShell
59 lines
1.6 KiB
PowerShell
|
#!/usr/bin/env pwsh
|
||
|
|
# Create a new Alembic migration with autogenerate (Windows)
|
||
|
|
|
||
|
|
Param(
|
||
|
|
[string]$MigrationName,
|
||
|
|
[switch]$DryRun
|
||
|
|
)
|
||
|
|
|
||
|
|
$ErrorActionPreference = 'Stop'
|
||
|
|
|
||
|
|
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
|
||
|
|
$BaseDir = Split-Path -Parent $ScriptDir
|
||
|
|
Set-Location $BaseDir
|
||
|
|
|
||
|
|
# Ensure repository root is importable for Alembic env/module resolution.
|
||
|
|
if ($env:PYTHONPATH) {
|
||
|
|
$env:PYTHONPATH = "$BaseDir;$($env:PYTHONPATH)"
|
||
|
|
} else {
|
||
|
|
$env:PYTHONPATH = $BaseDir
|
||
|
|
}
|
||
|
|
|
||
|
|
$EnvFile = Join-Path $BaseDir 'api/.env'
|
||
|
|
|
||
|
|
# Load environment variables
|
||
|
|
if (Test-Path $EnvFile) {
|
||
|
|
Get-Content $EnvFile | ForEach-Object {
|
||
|
|
$line = $_.Trim()
|
||
|
|
if ($line -and -not $line.StartsWith('#')) {
|
||
|
|
$parts = $line -split '=', 2
|
||
|
|
if ($parts.Count -eq 2) {
|
||
|
|
[Environment]::SetEnvironmentVariable($parts[0].Trim(), $parts[1].Trim().Trim('"'), 'Process')
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
Write-Host "Error: Environment file $EnvFile not found." -ForegroundColor Red
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
|
||
|
|
# Prompt for migration name when not provided via parameter
|
||
|
|
if (-not $MigrationName) {
|
||
|
|
$MigrationName = Read-Host "Enter the migration name (minimum 5 characters)"
|
||
|
|
}
|
||
|
|
|
||
|
|
if (-not $MigrationName -or $MigrationName.Length -lt 5) {
|
||
|
|
Write-Host "Error: Migration name must be at least 5 characters long." -ForegroundColor Red
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
|
||
|
|
# Generate the Alembic revision
|
||
|
|
$cmd = "alembic -c api/alembic.ini revision --autogenerate -m `"$MigrationName`""
|
||
|
|
|
||
|
|
if ($DryRun) {
|
||
|
|
Write-Host "Dry run: $cmd"
|
||
|
|
exit 0
|
||
|
|
}
|
||
|
|
|
||
|
|
alembic -c api/alembic.ini revision --autogenerate -m $MigrationName
|