@@ -24,7 +24,7 @@ def db_command(ctx):
2424 print ("ERROR connection string is missing." )
2525 exit (1 )
2626 provider .connect (connection_str , check_version = False )
27- ctx .call_on_close (provider .disconnect )
27+ ctx .call_on_close (provider .close )
2828 query = select (SchemaVersion ).order_by (SchemaVersion .version .desc ())
2929 schema_version , = provider .session .execute (query ).first ()
3030 print ("Schema version: {} - '{}'" .format (schema_version .version , schema_version .comment ))
@@ -89,7 +89,7 @@ def _print_sqlite_table_sizes(engine):
8989@pass_rcdb_context
9090def update (context ):
9191 provider = RCDBProvider (context .connection_str , check_version = False )
92- click .get_current_context ().call_on_close (provider .disconnect )
92+ click .get_current_context ().call_on_close (provider .close )
9393
9494 # Check something exists
9595 if not sqlalchemy .inspect (provider .engine ).has_table (SchemaVersion .__tablename__ ):
@@ -119,7 +119,7 @@ def update(context):
119119 return
120120
121121 provider = RCDBProvider (context .connection_str , check_version = False )
122- click .get_current_context ().call_on_close (provider .disconnect )
122+ click .get_current_context ().call_on_close (provider .close )
123123
124124 # Create alias table
125125 Alias .__table__ .create (provider .engine )
@@ -184,7 +184,7 @@ def init(context, drop_all, no_defaults, confirm, add_tests, add_cpp_tests):
184184 # That we will need for DB
185185 metadata = rcdb .model .Base .metadata
186186 provider = RCDBProvider (context .connection_str , check_version = False )
187- click .get_current_context ().call_on_close (provider .disconnect )
187+ click .get_current_context ().call_on_close (provider .close )
188188
189189 # Drop all if needed
190190 if drop_all :
0 commit comments