|
@@ -58,10 +58,10 @@ if __name__ == "__main__":
|
|
|
" on it."
|
|
|
)
|
|
|
)
|
|
|
- parser.add_argument("-v", action='store_true')
|
|
|
+ parser.add_argument("-v", action="store_true")
|
|
|
parser.add_argument(
|
|
|
"--database-config",
|
|
|
- type=argparse.FileType('r'),
|
|
|
+ type=argparse.FileType("r"),
|
|
|
required=True,
|
|
|
help="A database config file for either a SQLite3 database or a PostgreSQL one.",
|
|
|
)
|
|
@@ -101,10 +101,7 @@ if __name__ == "__main__":
|
|
|
|
|
|
# Instantiate and initialise the homeserver object.
|
|
|
hs = MockHomeserver(
|
|
|
- config,
|
|
|
- database_engine,
|
|
|
- db_conn,
|
|
|
- db_config=config.database_config,
|
|
|
+ config, database_engine, db_conn, db_config=config.database_config,
|
|
|
)
|
|
|
# setup instantiates the store within the homeserver object.
|
|
|
hs.setup()
|
|
@@ -112,13 +109,13 @@ if __name__ == "__main__":
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
|
def run_background_updates():
|
|
|
- yield store.run_background_updates(sleep=False)
|
|
|
+ yield store.db.updates.run_background_updates(sleep=False)
|
|
|
# Stop the reactor to exit the script once every background update is run.
|
|
|
reactor.stop()
|
|
|
|
|
|
# Apply all background updates on the database.
|
|
|
- reactor.callWhenRunning(lambda: run_as_background_process(
|
|
|
- "background_updates", run_background_updates
|
|
|
- ))
|
|
|
+ reactor.callWhenRunning(
|
|
|
+ lambda: run_as_background_process("background_updates", run_background_updates)
|
|
|
+ )
|
|
|
|
|
|
reactor.run()
|