Skip to content

Commit

Permalink
Merge pull request #60 from CybercentreCanada/improvements/update_sou…
Browse files Browse the repository at this point in the history
…rces

Update relative to changes in service base
  • Loading branch information
cccs-rs authored Dec 12, 2024
2 parents 6eea9bd + 268ff93 commit 59b5f22
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions avclass_updater.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,7 @@ class AVClassUpdateServer(ServiceUpdater):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)

def import_update(
self,
files_sha256: str,
source: str,
default_classification: str = None,
):
def import_update(self, files_sha256: str, source: str, *args, **kwargs):
assert len(files_sha256) == 1
shutil.move(files_sha256[0][0], os.path.join(self.latest_updates_dir, f"{source}.json"))
self.log.info(f"Finished moving {source}.json to {self.latest_updates_dir}")
Expand Down

0 comments on commit 59b5f22

Please sign in to comment.