Adrian Vladu 954753b6b8 replace imp module with importlib
When trying to run cloudbase-init using Python 3.12, it errors out
ModuleNotFoundError: No module named 'imp'.

The 'imp' module was replaced with similar functionality from module
importlib.

These two implementations are equivalent:

```python
import imp

import os
import site

wmi_path = None
for packages_path in site.getsitepackages():
    path = os.path.join(packages_path, "wmi.py")
    if os.path.isfile(path):
        wmi_path = path
        break

wmi_module_name = "wmi"
wmi_module = imp.load_source(wmi_module_name, wmi_path)

```

```python
import importlib.util

import os
import site

wmi_path = None
for packages_path in site.getsitepackages():
    path = os.path.join(packages_path, "wmi.py")
    if os.path.isfile(path):
        wmi_path = path
        break

wmi_module_name = "wmi"
wmi_module_spec = importlib.util.spec_from_file_location(wmi_module_name, wmi_path)
wmi_module = importlib.util.module_from_spec(wmi_module_spec)
wmi_module_spec.loader.exec_module(wmi_module)
```

Fixes: https://github.com/cloudbase/cloudbase-init/issues/139

Change-Id: I6490c6d9922efea26ab8d167a0d6e41ce34d6c2c
Signed-off-by: Adrian Vladu <avladu@cloudbasesolutions.com>
2024-05-30 14:02:14 +00:00

47 lines
1.4 KiB
Python

# Copyright 2012 Cloudbase Solutions Srl
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, 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 importlib
import os
from oslo_log import log as oslo_logging
LOG = oslo_logging.getLogger(__name__)
def load_module_from_path(module_name, path):
spec = importlib.util.spec_from_file_location(module_name, path)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
return module
class ClassLoader(object):
def load_class(self, class_path):
LOG.debug('Loading class \'%s\'' % class_path)
parts = class_path.rsplit('.', 1)
module = __import__(parts[0], fromlist=parts[1])
return getattr(module, parts[1])
def load_module(self, path):
module_name, file_ext = os.path.splitext(os.path.split(path)[-1])
if file_ext.lower() in ['.py', '.pyc']:
module = load_module_from_path(module_name, path)
return module