mirror of
https://github.com/wizarrrr/wizarr.git
synced 2026-08-02 08:17:23 -04:00
7b8f64af7d
- Added __init__ constructors to all SQLAlchemy models - Fixed pyright 'No parameter named' errors across the codebase - Fixed various ruff linting issues (SIM108, E741, E402, SIM102, B904, SIM105) 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
# app/scripts/migrate_libraries.py
|
|
|
|
from app import db
|
|
from app.models import Invitation, Library, Settings
|
|
|
|
|
|
def update_server_verified(app):
|
|
# Function to convert server_verified = "true" to server_verified = True
|
|
with app.app_context():
|
|
# 1) load the Settings row
|
|
row = Settings.query.filter_by(key="server_verified").first()
|
|
if not row or not row.value:
|
|
# nothing to migrate
|
|
return
|
|
|
|
# 2) convert string to boolean
|
|
if row.value == "1":
|
|
row.value = "true"
|
|
|
|
db.session.commit()
|
|
|
|
|
|
def run_library_migration(app):
|
|
"""Idempotently pull old comma-list out of Settings.libraries,
|
|
turn them into Library rows & Invitation<>Library links,
|
|
then delete that Settings entry so future runs skip it."""
|
|
with app.app_context():
|
|
# 1) load the Settings row
|
|
row = Settings.query.filter_by(key="libraries").first()
|
|
if not row or not row.value:
|
|
# nothing to migrate
|
|
return
|
|
|
|
# 2) parse out all the ext-ids
|
|
old_ext_ids = [s.strip() for s in row.value.split(",") if s.strip()]
|
|
if not old_ext_ids:
|
|
# empty list → just delete and quit
|
|
db.session.delete(row)
|
|
db.session.commit()
|
|
return
|
|
|
|
# 3) upsert each global Library record
|
|
for ext in old_ext_ids:
|
|
lib = Library.query.filter_by(external_id=ext).first()
|
|
if not lib:
|
|
db.session.add(Library(external_id=ext, name=ext, enabled=True))
|
|
|
|
# 4) remove that Settings row entirely so we never run again
|
|
db.session.delete(row)
|
|
|
|
db.session.commit()
|
|
|
|
# 5) now wire up per-invite links
|
|
total_links = 0
|
|
for inv in Invitation.query:
|
|
if inv.specific_libraries:
|
|
parts = [
|
|
s.strip() for s in inv.specific_libraries.split(",") if s.strip()
|
|
]
|
|
for ext in parts:
|
|
lib = Library.query.filter_by(external_id=ext).first()
|
|
if lib and lib not in inv.libraries:
|
|
inv.libraries.append(lib)
|
|
total_links += 1
|
|
# clear out the old CSV field
|
|
inv.specific_libraries = None
|
|
|
|
db.session.commit()
|
|
print(
|
|
f"[migrate_libraries] created {len(old_ext_ids)} Library rows and {total_links} invite links."
|
|
)
|