관리-도구

편집 파일: 051_cleanup_vd_license.py

import json import logging from contextlib import suppress logger = logging.getLogger(__name__) class VirusdieLicense: CONFIG_FILE = "/usr/local/vdserver/config.json" def unregister(self): self._write_key("") def _write_key(self, key): with open(self.CONFIG_FILE) as read_file: content = json.load(read_file) content["vdbApiKey"] = key with open(self.CONFIG_FILE, "w") as write_file: json.dump( content, write_file, sort_keys=True, indent=2, separators=(",", ": "), ) def migrate(migrator, database, fake=False, **kwargs): """Write your migrations here.""" with suppress(FileNotFoundError): VirusdieLicense().unregister() def rollback(migrator, database, fake=False, **kwargs): """Write your rollback migrations here.""" pass