FM / patchman

/opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman/

/opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman

Name	Size	Mode	Actions
fs_scanner/	-	0755	rm
rpc/	-	0755	rm
__pycache__/	-	0755	rm
config.py	1908	0644	edit dl rm
constants.py	908	0644	edit dl rm
custom_integration.py	6216	0644	edit dl rm
exceptions.py	826	0644	edit dl rm
feature.py	3742	0644	edit dl rm
license.py	5417	0644	edit dl rm
model.py	2055	0644	edit dl rm
state.py	921	0644	edit dl rm
utils.py	1443	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman/model.py (2055B)

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>
"""
from contextlib import ExitStack, contextmanager
from tempfile import NamedTemporaryFile
from typing import List

from peewee import (
    BooleanField,
    CharField,
    ForeignKeyField,
    IntegerField,
    Model,
    SqliteDatabase,
)

class User(Model):
    class Meta:
        db_table = "users"

name = CharField()
    parent = CharField(default="")
    email = CharField(default="")
    language = CharField(default="")
    suspended = BooleanField(default=False)
    level = IntegerField(default=0)

class Domain(Model):
    class Meta:
        db_table = "domains"

name = CharField()
    user = ForeignKeyField(User, backref="domains")

class Path(Model):
    class Meta:
        db_table = "paths"

name = CharField()
    domain = ForeignKeyField(Domain, backref="paths")

@contextmanager
def patchman_db(models: List[Model], *, target_dir: str, prefix: str = None):
    tmp_file = NamedTemporaryFile(
        suffix=".db",
        prefix="patchman_" + f"{prefix}_" if prefix else "",
        dir=target_dir,
        delete=False,
    )
    db = SqliteDatabase(tmp_file.name)

stack = ExitStack()
    stack.enter_context(db)
    stack.enter_context(db.bind_ctx(models))

with stack:
        db.create_tables(models)
        yield db

""" 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 . Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see """ from contextlib import ExitStack, contextmanager from tempfile import NamedTemporaryFile from typing import List from peewee import ( BooleanField, CharField, ForeignKeyField, IntegerField, Model, SqliteDatabase, ) class User(Model): class Meta: db_table = "users" name = CharField() parent = CharField(default="") email = CharField(default="") language = CharField(default="") suspended = BooleanField(default=False) level = IntegerField(default=0) class Domain(Model): class Meta: db_table = "domains" name = CharField() user = ForeignKeyField(User, backref="domains") class Path(Model): class Meta: db_table = "paths" name = CharField() domain = ForeignKeyField(Domain, backref="paths") @contextmanager def patchman_db(models: List[Model], *, target_dir: str, prefix: str = None): tmp_file = NamedTemporaryFile( suffix=".db", prefix="patchman_" + f"{prefix}_" if prefix else "", dir=target_dir, delete=False, ) db = SqliteDatabase(tmp_file.name) stack = ExitStack() stack.enter_context(db) stack.enter_context(db.bind_ctx(models)) with stack: db.create_tables(models) yield db