Skip to content

Commit

Permalink
Replace deprecated imp with importlib
Browse files Browse the repository at this point in the history
Fixes matrix-org#9642.

Signed-off-by: Cristina Muñoz <[email protected]>
  • Loading branch information
xmunoz committed Mar 30, 2021
1 parent 4dabcf0 commit dbb748c
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions synapse/storage/prepare_database.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import imp
import importlib.util
import logging
import os
import re
Expand Down Expand Up @@ -454,8 +454,13 @@ def _upgrade_existing_database(
)

module_name = "synapse.storage.v%d_%s" % (v, root_name)
with open(absolute_path) as python_file:
module = imp.load_source(module_name, absolute_path, python_file) # type: ignore

spec = importlib.util.spec_from_file_location(
module_name, absolute_path
)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module) # type: ignore

logger.info("Running script %s", relative_path)
module.run_create(cur, database_engine) # type: ignore
if not is_empty:
Expand Down

0 comments on commit dbb748c

Please sign in to comment.