Admin

name	type	size	perms	modified	actions
[parent directory]
__pycache__/	dir	-	0755	2026-08-19 08:39:37
000_noop_migration.py	file	947 B	0644	2026-08-05 00:00:00	edit
001_fix_scan_unserialization.py	file	2,923 B	0644	2026-08-05 00:00:00	edit
002_convert_serialized_scans.py	file	2,544 B	0644	2026-08-05 00:00:00	edit
003_add_new_fields_to_malware_history.py	file	1,323 B	0644	2026-08-05 00:00:00	edit
003_whmapi1_set_importance_imav.py	file	1,681 B	0644	2026-08-05 00:00:00	edit
004_add_malwarehit_owner.py	file	1,127 B	0644	2026-08-05 00:00:00	edit
004_patchman_client.py	file	1,747 B	0644	2026-08-05 00:00:00	edit
004_plesk_configs.py	file	6,451 B	0644	2026-08-05 00:00:00	edit
004_plesk_eula.py	file	2,386 B	0644	2026-08-05 00:00:00	edit
005_plesk_cleanup_storage.py	file	4,111 B	0644	2026-08-05 00:00:00	edit
005_populate_malwarehit_owner.py	file	1,142 B	0644	2026-08-05 00:00:00	edit
006_add_malwarehit_snippet.py	file	1,137 B	0644	2026-08-05 00:00:00	edit
007_add_malwarehistory_fileuser.py	file	1,157 B	0644	2026-08-05 00:00:00	edit
007_revisium_ignore_list.py	file	2,660 B	0644	2026-08-05 00:00:00	edit
008_populate_malwarehistory_foleuser.py	file	1,185 B	0644	2026-08-05 00:00:00	edit
008_subscription_permissions.py	file	3,371 B	0644	2026-08-05 00:00:00	edit
009_revisium_scan_history.py	file	9,208 B	0644	2026-08-05 00:00:00	edit
010_add_malwarehistory_scan_id.py	file	1,159 B	0644	2026-08-05 00:00:00	edit
011_add_malwarehistory_index.py	file	1,527 B	0644	2026-08-05 00:00:00	edit
012_add_rescan_outdated_scan_type.py	file	1,432 B	0644	2026-08-05 00:00:00	edit
013_add_malware_ctime.py	file	1,224 B	0644	2026-08-05 00:00:00	edit
014_add_imunify_security.py	file	986 B	0644	2026-08-05 00:00:00	edit
015_manually_deleted_at_field_added.py	file	989 B	0644	2026-08-05 00:00:00	edit
016_add_vulnerability_hit_table.py	file	2,539 B	0644	2026-08-05 00:00:00	edit
017_add_imunify_patch_subscriptions_table.py	file	1,325 B	0644	2026-08-05 00:00:00	edit
017_add_version_field.py	file	988 B	0644	2026-08-05 00:00:00	edit
018_delete_suspicious_hits.py	file	977 B	0644	2026-08-05 00:00:00	edit
019_add_malware_history_signature_id_field.py	file	1,169 B	0644	2026-08-05 00:00:00	edit
020_add_default_plesk_settings_av.py	file	1,784 B	0644	2026-08-05 00:00:00	edit
021_remove_cron_scan_file.py	file	1,091 B	0644	2026-08-05 00:00:00	edit
022_scan_queue_to_json.py	file	1,612 B	0644	2026-08-05 00:00:00	edit
023_create_mrs_upload_queue.py	file	1,832 B	0644	2026-08-05 00:00:00	edit
024_create_mrs_queue_counters_and_meta.py	file	1,332 B	0644	2026-08-05 00:00:00	edit
__init__.py	file	0 B	0644	2026-08-05 00:00:00	edit

editing: 004_plesk_eula.py

"""
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import logging
import time
from peewee import SqliteDatabase, TextField
from imav.migration_utils.revisium import find_revisium_db
from imav.migration_utils.settings import SettingsExtractor

logger = logging.getLogger(__name__)

db = SqliteDatabase(None)

class KeyValue(db.Model):
    """ex-Revisium settings database model"""

key = TextField()
    value = TextField()

class Meta:
        primary_key = False

class Revisium(SettingsExtractor):
    """ex-Revisium database settings extractor"""

def _get(self):
        return KeyValue.get(KeyValue.key == self.key).value

def migrate_eula(migrator, database=None):
    db.init(database)
    eula_value = Revisium("ra_eula", 0).get()
    last_updated_eula_date = "2021-06-29"
    logger.info("EULA value=%s from Revisium database", str(eula_value))
    if eula_value >= 2:
        logger.info("Accepting eula for date=%s", last_updated_eula_date)
        Eula = migrator.orm["eula"]
        Eula.insert(
            updated=last_updated_eula_date, accepted=time.time()
        ).execute()

def migrate(migrator, database, fake=False, **kwargs):
    if fake:
        return

revisium_db_path = find_revisium_db()
    if revisium_db_path is None:
        logger.info("No legacy ImunifyAV database found. Skipping...")
        return

try:
        migrate_eula(migrator, revisium_db_path)
    except Exception as e:
        logger.warning("Failed to migrate eula for ImunifyAV: %r", e)

def rollback(migrator, database, fake=False, **kwargs):
    pass

if __name__ == "__main__":
    print(migrate_eula(find_revisium_db()))

cancel