Expose error more nicely

synapse/app/homeserver.py

@@ -35,9 +35,8 @@ if __name__ == '__main__':
 
 
 from synapse.storage.engines import create_engine, IncorrectDatabaseSetup
-from synapse.storage import (
-    are_all_users_on_domain, UpgradeDatabaseException,
-)
+from synapse.storage import are_all_users_on_domain
+from synapse.storage.schema_prepare import UpgradeDatabaseException
 
 from synapse.server import HomeServer
 

synapse/storage/__init__.py

@@ -41,9 +41,6 @@ from .end_to_end_keys import EndToEndKeyStore
 
 from .receipts import ReceiptsStore
 
-from ._schema_prepare import UpgradeDatabaseException
-
-__all__ = [UpgradeDatabaseException]
 
 import logging
 

synapse/storage/engines/postgres.py

@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from synapse.storage._schema_prepare import prepare_database
+from synapse.storage.schema_prepare import prepare_database
 
 from ._base import IncorrectDatabaseSetup
 

synapse/storage/engines/sqlite3.py

@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from synapse.storage._schema_prepare import (
+from synapse.storage.schema_prepare import (
     prepare_database, prepare_sqlite3_database
 )
 

tests/utils.py

@@ -16,7 +16,7 @@
 from synapse.http.server import HttpServer
 from synapse.api.errors import cs_error, CodeMessageException, StoreError
 from synapse.api.constants import EventTypes
-from synapse.storage import prepare_database
+from synapse.storage.schema_prepare import prepare_database
 from synapse.storage.engines import create_engine
 from synapse.server import HomeServer