update some utils

This commit is contained in:
Kohaku-Blueleaf
2025-10-05 03:06:30 +08:00
parent ef57cd9eb5
commit 84ecb94474
2 changed files with 51 additions and 22 deletions

View File

@@ -8,12 +8,26 @@ region = "us-east-1"
force_path_style = true
[lakefs]
public_endpoint = "http://127.0.0.1:28000"
internal_endpoint = "http://127.0.0.1:28000"
endpoint = "http://127.0.0.1:28000"
access_key = "xxx"
secret_key = "yyy"
repo_namespace = "hf"
[smtp]
enabled = false
host = "smtp.gmail.com"
port = 587
username = "your-email@gmail.com"
password = "your-app-password"
from_email = "noreply@yourdomain.com"
use_tls = true
[auth]
require_email_verification = false
session_secret = "change-me-in-production"
session_expire_hours = 168 # 7 days
token_expire_days = 365
[app]
base_url = "http://127.0.0.1:48888"
api_base = "/api"

View File

@@ -78,37 +78,52 @@ def migrate_sqlite():
def migrate_postgres():
"""Migrate PostgreSQL database - can drop constraint directly."""
"""Migrate PostgreSQL database - drop unique index on full_id."""
print("Migrating PostgreSQL database...")
cursor = db.cursor()
# Check if unique constraint exists
# Check if unique index exists
cursor.execute(
"""
SELECT constraint_name
FROM information_schema.table_constraints
WHERE table_name = 'repository'
AND constraint_type = 'UNIQUE'
AND constraint_name LIKE '%full_id%'
SELECT indexname, indexdef
FROM pg_indexes
WHERE tablename = 'repository'
AND indexname = 'repository_full_id'
"""
)
constraints = cursor.fetchall()
result = cursor.fetchone()
# Drop unique constraint on full_id if it exists
for constraint in constraints:
constraint_name = constraint[0]
print(f"Dropping constraint: {constraint_name}")
cursor.execute(f"ALTER TABLE repository DROP CONSTRAINT {constraint_name}")
if result:
indexdef = result[1] if len(result) > 1 else ""
print(f"Found index: {result[0]}")
print(f"Definition: {indexdef}")
# Ensure we have the regular index on full_id
cursor.execute(
"""
CREATE INDEX IF NOT EXISTS repository_full_id
ON repository (full_id)
"""
)
# Check if it's a unique index
if "UNIQUE" in indexdef.upper():
print("Dropping unique index: repository_full_id")
cursor.execute("DROP INDEX IF EXISTS repository_full_id")
# Create regular (non-unique) index
print("Creating regular index on full_id...")
cursor.execute(
"""
CREATE INDEX repository_full_id
ON repository (full_id)
"""
)
print("✓ Index recreated as non-unique")
else:
print("Index is already non-unique, no migration needed")
else:
print("Index repository_full_id not found, creating regular index...")
cursor.execute(
"""
CREATE INDEX IF NOT EXISTS repository_full_id
ON repository (full_id)
"""
)
db.commit()
print("✓ PostgreSQL migration completed successfully")