Fix cosmetic issues.

Author: funilrys

PyFunceble/data/alembic/mysql/env.py

@@ -57,8 +57,8 @@
 from sqlalchemy import engine_from_config, pool
 
 import PyFunceble.cli.facility
+import PyFunceble.database.sqlalchemy.all_schemas
 import PyFunceble.facility
-from PyFunceble.database.sqlalchemy.all_schemas import Continue, Inactive, WhoisRecord
 from PyFunceble.database.sqlalchemy.base_schema import SchemaBase
 
 config = context.config

PyFunceble/data/alembic/postgresql/env.py

@@ -57,8 +57,8 @@
 from sqlalchemy import engine_from_config, pool
 
 import PyFunceble.cli.facility
+import PyFunceble.database.sqlalchemy.all_schemas
 import PyFunceble.facility
-from PyFunceble.database.sqlalchemy.all_schemas import Continue, Inactive, WhoisRecord
 from PyFunceble.database.sqlalchemy.base_schema import SchemaBase
 
 config = context.config

PyFunceble/dataset/autocontinue/sql.py

@@ -70,7 +70,7 @@ class SQLDBContinueDataset(SQLDBDatasetBase, ContinueDatasetBase):
 
     @SQLDBDatasetBase.execute_if_authorized(None)
     # pylint: disable=arguments-differ
-    def cleanup(self, *, session_id: str) -> "MariaDBContinueDataset":
+    def cleanup(self, *, session_id: str) -> "SQLDBContinueDataset":
         """
         Cleanups the dataset. Meaning that we delete every entries which are
         needed anymore.

PyFunceble/dataset/sql_base.py

@@ -121,7 +121,7 @@ def get_content(self) -> Generator[dict, None, None]:
             yield row
 
     @DBDatasetBase.execute_if_authorized(None)
-    def update(self, row, *, ignore_if_exist: bool = False) -> "MariaDBDatasetBase":
+    def update(self, row, *, ignore_if_exist: bool = False) -> "SQLDBDatasetBase":
         """
         Adds the given dataset into the database if it does not exists.
         Update otherwise.
@@ -168,7 +168,7 @@ def update(self, row, *, ignore_if_exist: bool = False) -> "MariaDBDatasetBase":
 
     @DBDatasetBase.execute_if_authorized(None)
     @ensure_orm_obj_is_given
-    def remove(self, row) -> "MariaDBDatasetBase":
+    def remove(self, row) -> "SQLDBDatasetBase":
         """
         Removes the given dataset from the database.
 
@@ -276,7 +276,7 @@ def get_existing_row(self, row):
 
     @DBDatasetBase.execute_if_authorized(None)
     @ensure_orm_obj_is_given
-    def add(self, row) -> "MariaDBDatasetBase":
+    def add(self, row) -> "SQLDBDatasetBase":
         """
         Adds the given dataset into the database.
 

PyFunceble/dataset/whois/sql.py

@@ -121,7 +121,7 @@ def get_content(self) -> Generator[dict, None, None]:
     @SQLDBDatasetBase.execute_if_authorized(None)
     def update(
         self, row: Union[dict, WhoisRecord], *, ignore_if_exist: bool = False
-    ) -> "MariaDBWhoisDataset":
+    ) -> "SQLDBWhoisDataset":
         """
         Adds the given dataset into the database if it does not exists.
         Update otherwise.
@@ -155,7 +155,7 @@ def update(
         return self
 
     @SQLDBDatasetBase.execute_if_authorized(None)
-    def cleanup(self) -> "MariaDBWhoisDataset":
+    def cleanup(self) -> "SQLDBWhoisDataset":
         """
         Cleanups the dataset. Meaning that we delete every entries which are
         in the past.

tests/dataset/test_db_base.py

@@ -210,6 +210,7 @@ def test_get_filtered_row(self) -> None:
         """
 
         self.dataset.authorized = True
+        # pylint: disable=invalid-name
         self.dataset.FIELDS = ["hello", "world", "fun", "state"]
         self.dataset.COMPARISON_FIELDS = ["hello"]