mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-09-18 07:47:42 +02:00
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
from django import forms
|
|
from scanEngine.models import EngineType
|
|
from django_ace import AceWidget
|
|
|
|
class AddEngineForm(forms.ModelForm):
|
|
class Meta:
|
|
model = EngineType
|
|
fields = '__all__'
|
|
engine_name = forms.CharField(
|
|
required=True,
|
|
widget=forms.TextInput(
|
|
attrs={
|
|
"class": "form-control",
|
|
"id": "scan_engine_name",
|
|
"placeholder":"Custom Engine"
|
|
}
|
|
))
|
|
subdomain_discovery = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={"checked": ""}))
|
|
dir_file_search = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={}))
|
|
subdomain_takeover = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={}))
|
|
port_scan = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={"checked":""}))
|
|
fetch_url = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={"checked":""}))
|
|
yaml_configuration = forms.CharField(widget=AceWidget(
|
|
mode="yaml",
|
|
theme="monokai",
|
|
width="100%",
|
|
height="450px",
|
|
tabsize=4,
|
|
fontsize=13,
|
|
toolbar=True,))
|
|
|
|
class UpdateEngineForm(forms.ModelForm):
|
|
class Meta:
|
|
model = EngineType
|
|
fields = '__all__'
|
|
engine_name = forms.CharField(
|
|
required=True,
|
|
widget=forms.TextInput(
|
|
attrs={
|
|
"class": "form-control",
|
|
"id": "scan_engine_name",
|
|
"placeholder":"Custom Engine"
|
|
}
|
|
))
|
|
subdomain_discovery = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={"checked": ""}))
|
|
dir_file_search = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={}))
|
|
subdomain_takeover = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={}))
|
|
port_scan = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={"checked":""}))
|
|
fetch_url = forms.BooleanField(
|
|
required=False,
|
|
widget=forms.CheckboxInput(attrs={"checked":""}))
|
|
yaml_configuration = forms.CharField(widget=AceWidget(
|
|
mode="yaml",
|
|
theme="monokai",
|
|
width="100%",
|
|
height="450px",
|
|
tabsize=4,
|
|
fontsize=13,
|
|
toolbar=True,))
|
|
|
|
def set_value(self, engine):
|
|
self.initial['engine_name'] = engine.engine_name
|
|
self.initial['subdomain_discovery'] = engine.subdomain_discovery
|
|
self.initial['dir_file_search'] = engine.dir_file_search
|
|
self.initial['subdomain_takeover'] = engine.subdomain_takeover
|
|
self.initial['port_scan'] = engine.port_scan
|
|
self.initial['fetch_url'] = engine.fetch_url
|
|
self.initial['yaml_configuration'] = engine.yaml_configuration
|