# HG changeset patch
# User Aurélien Campéas <aurelien.campeas@pythonian.fr>
# Date 1701451054 -3600
#      Fri Dec 01 18:17:34 2023 +0100
# Node ID df82cbe88d793c5c589b19afc58be4b21f45e0b2
# Parent  f7e02027e98a564d16f0e1b1e1b269852ab7baa7
replace imp.load_source with a new thing based on importlib

The builtin imp module has been deprecated for a long time
and is gone in Python 3.12.

diff --git a/rework/cli.py b/rework/cli.py
--- a/rework/cli.py
+++ b/rework/cli.py
@@ -1,4 +1,3 @@
-import imp
 from datetime import timedelta
 from time import sleep
 import tzlocal
@@ -19,6 +18,7 @@
     cleanup_tasks,
     cleanup_workers,
     find_dburi,
+    load_source,
     schedule_plan,
     utcnow,
     setuplogger
@@ -60,10 +60,9 @@
     The registered operations will be relative to the current host.
     """
     # load the module
-    imp.load_source('operations', module)
+    load_source('operations', module)
     engine = create_engine(find_dburi(dburi))
     ok, ko = api.freeze_operations(engine, domain)
-
     print(f'registered {len(ok)} new operation ({len(ko)} already known)')
 
 
diff --git a/rework/helper.py b/rework/helper.py
--- a/rework/helper.py
+++ b/rework/helper.py
@@ -11,6 +11,8 @@
 import json
 import struct
 import tzlocal
+import importlib
+import sys
 
 from icron import croniter_range
 import pyzstd as zstd
@@ -37,6 +39,19 @@
     return logger
 
 
+def load_source(modname, filename):
+    loader = importlib.machinery.SourceFileLoader(
+        modname, filename
+    )
+    spec = importlib.util.spec_from_file_location(
+        modname, filename, loader=loader
+    )
+    module = importlib.util.module_from_spec(spec)
+    sys.modules[modname] = module
+    loader.exec_module(module)
+    return module
+
+
 def utcnow():
     return datetime.utcnow().replace(tzinfo=pytz.utc)
 
diff --git a/rework/task.py b/rework/task.py
--- a/rework/task.py
+++ b/rework/task.py
@@ -1,5 +1,4 @@
 import sys
-import imp
 import time
 from pickle import dumps, loads
 import traceback as tb
@@ -9,6 +8,7 @@
 from sqlhelp import select, update
 
 from rework.helper import (
+    load_source,
     pack_io,
     PGLogHandler,
     PGLogWriter,
@@ -313,15 +313,15 @@
             ).values(
                 started=utcnow()
             ).do(cn)
+            name, path = self.engine.execute(
+                'select name, path '
+                'from rework.operation '
+                'where rework.operation.id = %(operation)s',
+                operation=self.operation
+            ).fetchone()
+        mod = load_source('module', path)
+        func = getattr(mod, name)
         try:
-            name, path = self.engine.execute("""
-                select name, path
-                from rework.operation
-                where rework.operation.id = %(operation)s
-            """, {'operation': self.operation}
-            ).fetchone()
-            mod = imp.load_source('module', path)
-            func = getattr(mod, name)
             func(self)
         except:
             with self.engine.begin() as cn: