Idempotent-ise schema update script

If any ASes don't have an ID, the schema will fail, and then it will
error when trying to add the column again.
This commit is contained in:
Daniel Wagner-Hall 2016-03-08 17:32:29 +00:00
parent 95481e7ba7
commit edca2d9891

View file

@ -20,7 +20,11 @@ logger = logging.getLogger(__name__)
def run_upgrade(cur, database_engine, config, *args, **kwargs):
# NULL indicates user was not registered by an appservice.
cur.execute("ALTER TABLE users ADD COLUMN appservice_id TEXT")
try:
cur.execute("ALTER TABLE users ADD COLUMN appservice_id TEXT")
except:
# Maybe we already added the column? Hope so...
pass
cur.execute("SELECT name FROM users")
rows = cur.fetchall()