From 27fcbb63cbb9b3f06bd436bb1344a467093dfe90 Mon Sep 17 00:00:00 2001 From: femalves Date: Mon, 11 Aug 2025 15:21:49 -0400 Subject: [PATCH] fixing falsy values --- ...05b19_adding_library_id_foreign_key_to_user.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/alembic/versions/af63c0205b19_adding_library_id_foreign_key_to_user.py b/alembic/versions/af63c0205b19_adding_library_id_foreign_key_to_user.py index b253b1a..755fac8 100644 --- a/alembic/versions/af63c0205b19_adding_library_id_foreign_key_to_user.py +++ b/alembic/versions/af63c0205b19_adding_library_id_foreign_key_to_user.py @@ -36,13 +36,14 @@ def upgrade(): for user_id, user_data in users: if user_data and isinstance(user_data, dict): link_server = user_data.get('link_server') - if link_server: - library_result = bind.execute( - "SELECT id FROM library WHERE libserver = %(libserver)s", - {"libserver": link_server} - ).fetchone() - - library_id = library_result[0] if library_result else None + if link_server is not None: # Process both empty strings and non-empty values + library_id = None + if link_server: # Only look up library if link_server is not empty + library_result = bind.execute( + "SELECT id FROM library WHERE libserver = %(libserver)s", + {"libserver": link_server} + ).fetchone() + library_id = library_result[0] if library_result else None new_user_data = user_data.copy() new_user_data.pop('link_server', None)