Merge pull request #1 from cccs-rs/compatibility/v4.5.0.stable66

Update function signature to be compatible with service-base changes
kam193 · Dec 13, 2024 · 9560429 · 9560429
2 parents 3929ebc + 78df993
commit 9560429
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/ASARExtractor/service/updater.py b/ASARExtractor/service/updater.py
@@ -20,7 +20,7 @@ def __init__(self, *args, **kwargs):
     # def is_valid(self, file_path) -> bool:
     #     return True
 
-    def import_update(self, files_sha256, source, default_classification) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         output_dir = os.path.join(self.latest_updates_dir, source)
         os.makedirs(os.path.join(self.latest_updates_dir, source), exist_ok=True)
         for file, _ in files_sha256:

diff --git a/ASTGrep/service/updater.py b/ASTGrep/service/updater.py
@@ -47,7 +47,7 @@ def _preprocess_rule(self, rule: dict) -> dict:
             del rule["paths"]
         return rule
 
-    def import_update(self, files_sha256, source, default_classification) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         signatures: list[Signature] = []
         for file, _ in files_sha256:
             with open(file, "r") as f:

diff --git a/clamav-service/service/updater.py b/clamav-service/service/updater.py
@@ -171,7 +171,7 @@ def _test_database_file(self, file: str) -> bool:
 
         raise ValueError(result.stderr)
 
-    def import_update(self, files_sha256, source, default_classification=None) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         output_dir = os.path.join(self.latest_updates_dir, source)
         os.makedirs(os.path.join(self.latest_updates_dir, source), exist_ok=True)
         for file, _ in files_sha256:

diff --git a/file-similarity/service/updater.py b/file-similarity/service/updater.py
@@ -158,7 +158,7 @@ def do_source_update(self, service: Service) -> None:
         super().do_source_update(service)
         self._clean_up_old_sources(service, self.latest_updates_dir)
 
-    def import_update(self, files_sha256, source, default_classification=None) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         # TODO: preprocess updates
         output_dir = os.path.join(self.latest_updates_dir, source)
         os.makedirs(os.path.join(self.latest_updates_dir, source), exist_ok=True)

diff --git a/network-information/service/updater.py b/network-information/service/updater.py
@@ -20,7 +20,7 @@ def __init__(self, *args, **kwargs):
     # def is_valid(self, file_path) -> bool:
     #     return True
 
-    def import_update(self, files_sha256, source, default_classification) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         output_dir = os.path.join(self.latest_updates_dir, source)
         os.makedirs(os.path.join(self.latest_updates_dir, source), exist_ok=True)
         for file, _ in files_sha256:

diff --git a/pcap-extractor/service/updater.py b/pcap-extractor/service/updater.py
@@ -17,7 +17,7 @@ def __init__(self, *args, **kwargs):
     # ) -> None:
     #     pass
 
-    def import_update(self, files_sha256, client, source, default_classification) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         pass
 
     def prepare_output_directory(self) -> str:

diff --git a/python-magic/service/updater.py b/python-magic/service/updater.py
@@ -20,7 +20,7 @@ def __init__(self, *args, **kwargs):
     # def is_valid(self, file_path) -> bool:
     #     return True
 
-    def import_update(self, files_sha256, source, default_classification) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         output_dir = os.path.join(self.latest_updates_dir, source)
         os.makedirs(os.path.join(self.latest_updates_dir, source), exist_ok=True)
         for file, _ in files_sha256:

diff --git a/semgrep/service/updater.py b/semgrep/service/updater.py
@@ -45,7 +45,7 @@ def _preprocess_rule(self, rule: dict) -> dict:
             del rule["paths"]
         return rule
 
-    def import_update(self, files_sha256, source, default_classification) -> None:
+    def import_update(self, files_sha256, source, default_classification=None, *args, **kwargs) -> None:
         signatures: list[Signature] = []
         for file, _ in files_sha256:
             with open(file, "r") as f: