|
| 1 | +import logging |
| 2 | +from .exc import DatabaseUpgradeRequired |
| 3 | + |
| 4 | + |
| 5 | +def migrate(conn, *, check_only): |
| 6 | + from .. import db |
| 7 | + |
| 8 | + # Room info_updates triggers for global mods didn't fire for invisible global mods/admins, but |
| 9 | + # should (so that other mods/admins notice the change). |
| 10 | + has_bad_trigger = db.query( |
| 11 | + """ |
| 12 | + SELECT COUNT(*) FROM sqlite_master |
| 13 | + WHERE type = 'trigger' AND name = :trigger |
| 14 | + AND LOWER(sql) LIKE :bad |
| 15 | + """ |
| 16 | + if db.engine.name == "sqlite" |
| 17 | + else """ |
| 18 | + SELECT COUNT(*) FROM information_schema.triggers |
| 19 | + WHERE trigger_name = :trigger |
| 20 | + AND LOWER(action_condition) LIKE :bad |
| 21 | + """, |
| 22 | + trigger='room_metadata_global_mods_insert', |
| 23 | + bad='% new.visible_mod%', |
| 24 | + dbconn=conn, |
| 25 | + ).first()[0] |
| 26 | + |
| 27 | + if not has_bad_trigger: |
| 28 | + return False |
| 29 | + |
| 30 | + logging.warning("DB migration: fixing global hidden mod room triggers") |
| 31 | + if check_only: |
| 32 | + raise DatabaseUpgradeRequired("global hidden mod room triggers need to be recreated") |
| 33 | + |
| 34 | + if db.engine.name == "sqlite": |
| 35 | + conn.execute("DROP TRIGGER IF EXISTS room_metadata_global_mods_insert") |
| 36 | + conn.execute("DROP TRIGGER IF EXISTS room_metadata_global_mods_update") |
| 37 | + conn.execute("DROP TRIGGER IF EXISTS room_metadata_global_mods_delete") |
| 38 | + conn.execute( |
| 39 | + """ |
| 40 | +CREATE TRIGGER room_metadata_global_mods_insert AFTER INSERT ON users |
| 41 | +FOR EACH ROW WHEN (NEW.admin OR NEW.moderator) |
| 42 | +BEGIN |
| 43 | + UPDATE rooms SET info_updates = info_updates + 1; -- WHERE everything! |
| 44 | +END |
| 45 | +""" |
| 46 | + ) |
| 47 | + conn.execute( |
| 48 | + """ |
| 49 | +CREATE TRIGGER room_metadata_global_mods_update AFTER UPDATE ON users |
| 50 | +FOR EACH ROW WHEN (NEW.moderator != OLD.moderator OR NEW.admin != OLD.admin OR NEW.visible_mod != OLD.visible_mod) |
| 51 | +BEGIN |
| 52 | + UPDATE rooms SET info_updates = info_updates + 1; -- WHERE everything! |
| 53 | +END |
| 54 | +""" # noqa: E501 |
| 55 | + ) |
| 56 | + conn.execute( |
| 57 | + """ |
| 58 | +CREATE TRIGGER room_metadata_global_mods_delete AFTER DELETE ON users |
| 59 | +FOR EACH ROW WHEN (OLD.moderator OR OLD.admin) |
| 60 | +BEGIN |
| 61 | + UPDATE rooms SET info_updates = info_updates + 1; -- WHERE everything! |
| 62 | +END |
| 63 | +""" |
| 64 | + ) |
| 65 | + |
| 66 | + else: # postgresql |
| 67 | + conn.execute( |
| 68 | + """ |
| 69 | +DROP TRIGGER IF EXISTS room_metadata_global_mods_insert ON users; |
| 70 | +DROP TRIGGER IF EXISTS room_metadata_global_mods_update ON users; |
| 71 | +DROP TRIGGER IF EXISTS room_metadata_global_mods_delete ON users; |
| 72 | +
|
| 73 | +CREATE TRIGGER room_metadata_global_mods_insert AFTER INSERT ON users |
| 74 | +FOR EACH ROW WHEN (NEW.admin OR NEW.moderator) |
| 75 | +EXECUTE PROCEDURE trigger_room_metadata_info_update_all(); |
| 76 | +
|
| 77 | +CREATE TRIGGER room_metadata_global_mods_update AFTER UPDATE OF moderator, admin, visible_mod ON users |
| 78 | +FOR EACH ROW WHEN (NEW.moderator != OLD.moderator OR NEW.admin != OLD.admin OR NEW.visible_mod != OLD.visible_mod) |
| 79 | +EXECUTE PROCEDURE trigger_room_metadata_info_update_all(); |
| 80 | +
|
| 81 | +CREATE TRIGGER room_metadata_global_mods_delete AFTER DELETE ON users |
| 82 | +FOR EACH ROW WHEN (OLD.moderator OR OLD.admin) |
| 83 | +EXECUTE PROCEDURE trigger_room_metadata_info_update_all(); |
| 84 | +""" # noqa: E501 |
| 85 | + ) |
| 86 | + |
| 87 | + return True |
0 commit comments