No longer use AbstractContextManager

spinn_front_end_common/utilities/base_database.py

@@ -15,7 +15,6 @@
 import os
 import sqlite3
 import time
-from spinn_utilities.abstract_context_manager import AbstractContextManager
 from spinn_front_end_common.data import FecDataView
 from spinn_front_end_common.utilities.sqlite_db import SQLiteDB
 
@@ -29,7 +28,7 @@ def _timestamp():
     return int(time.time() * _SECONDS_TO_MICRO_SECONDS_CONVERSION)
 
 
-class BaseDatabase(SQLiteDB, AbstractContextManager):
+class BaseDatabase(SQLiteDB):
     """
     Specific implementation of the Database for SQLite 3.
 

spinn_front_end_common/utilities/notification_protocol/notification_protocol.py

@@ -14,7 +14,6 @@
 
 import logging
 from concurrent.futures import ThreadPoolExecutor, wait  # @UnresolvedImport
-from spinn_utilities.abstract_context_manager import AbstractContextManager
 from spinn_utilities.config_holder import get_config_bool, get_config_int
 from spinn_utilities.log import FormatAdapter
 from spinnman.connections.udp_packet_connections import EIEIOConnection
@@ -30,7 +29,7 @@
 logger = FormatAdapter(logging.getLogger(__name__))
 
 
-class NotificationProtocol(AbstractContextManager):
+class NotificationProtocol(object):
     """
     The protocol which hand shakes with external devices about the
     database and starting execution.

spinn_front_end_common/utilities/sqlite_db.py

@@ -18,13 +18,12 @@
 import pathlib
 import sqlite3
 import struct
-from spinn_utilities.abstract_context_manager import AbstractContextManager
 from spinn_front_end_common.utilities.exceptions import DatabaseException
 
 logger = logging.getLogger(__name__)
 
 
-class SQLiteDB(AbstractContextManager):
+class SQLiteDB(object):
     """
     General support class for SQLite databases. This handles a lot of the
     low-level detail of setting up a connection.
@@ -122,6 +121,12 @@ def __init__(self, database_file=None, *, read_only=False, ddl_file=None,
         self.__cursor = None
 
     def _context_entered(self):
+        """
+        Work to do when then context is entered.
+
+        May be extended by super classes
+
+        """
         if self.__db is None:
             raise DatabaseException("database has been closed")
         if self.__cursor is not None:
@@ -130,15 +135,18 @@ def _context_entered(self):
             self.__db.execute("BEGIN")
         self.__cursor = self.__db.cursor()
 
+    def __enter__(self):
+        self._context_entered()
+        return self
+
     def __exit__(self, exc_type, exc_val, exc_tb):
         if self.__db is not None:
             if exc_type is None:
                 self.__db.commit()
             else:
                 self.__db.rollback()
         self.__cursor = None
-        # calls close
-        return super().__exit__(exc_type, exc_val, exc_tb)
+        self.close()
 
     def __del__(self):
         self.close()