1616
1717@upgrade_table .register (description = "Latest revision" , upgrades_to = 4 )
1818async def upgrade_blank_to_v4 (conn : Connection , scheme : Scheme ) -> None :
19- await conn .execute (
20- """ CREATE TABLE mx_room_state (
19+ await conn .execute ("""
20+ CREATE TABLE mx_room_state (
2121 room_id TEXT PRIMARY KEY,
2222 is_encrypted BOOLEAN,
2323 has_full_member_list BOOLEAN,
2424 encryption TEXT,
2525 power_levels TEXT,
2626 create_event TEXT
27- )"""
28- )
27+ )
28+ """ )
2929 membership_check = ""
3030 if scheme != Scheme .SQLITE :
3131 await conn .execute (
3232 "CREATE TYPE membership AS ENUM ('join', 'leave', 'invite', 'ban', 'knock')"
3333 )
3434 else :
3535 membership_check = "CHECK (membership IN ('join', 'leave', 'invite', 'ban', 'knock'))"
36- await conn .execute (
37- f""" CREATE TABLE mx_user_profile (
36+ await conn .execute (f"""
37+ CREATE TABLE mx_user_profile (
3838 room_id TEXT,
3939 user_id TEXT,
4040 membership membership NOT NULL { membership_check } ,
4141 displayname TEXT,
4242 avatar_url TEXT,
4343 PRIMARY KEY (room_id, user_id)
44- )"""
45- )
44+ )
45+ """ )
4646
4747
4848@upgrade_table .register (description = "Stop using size-limited string fields" )
@@ -60,16 +60,14 @@ async def upgrade_v2(conn: Connection, scheme: Scheme) -> None:
6060@upgrade_table .register (description = "Mark rooms that need crypto state event resynced" )
6161async def upgrade_v3 (conn : Connection ) -> None :
6262 if await conn .table_exists ("portal" ):
63- await conn .execute (
64- """
63+ await conn .execute ("""
6564 INSERT INTO mx_room_state (room_id, encryption)
6665 SELECT portal.mxid, '{"resync":true}' FROM portal
6766 WHERE portal.encrypted=true AND portal.mxid IS NOT NULL
6867 ON CONFLICT (room_id) DO UPDATE
6968 SET encryption=excluded.encryption
7069 WHERE mx_room_state.encryption IS NULL
71- """
72- )
70+ """ )
7371
7472
7573@upgrade_table .register (description = "Add create event to room state cache" )
0 commit comments