ACIL FM

Pilih semua

Nama	Ukuran	Aksi
fs_scanner	-	Open
rpc	-	Open
__pycache__	-	Open
config.py	1.86 MB	View DL Edit
constants.py	908 B	View DL Edit
custom_integration.py	5.18 MB	View DL Edit
exceptions.py	826 B	View DL Edit
feature.py	3.42 MB	View DL Edit
license.py	5.29 MB	View DL Edit
model.py	2.01 MB	View DL Edit
state.py	921 B	View DL Edit
utils.py	1.41 MB	View DL Edit
__init__.py	0 B	View DL Edit

Edit file: /opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman/model.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>
"""
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

Batal

Isi Zip:

Create