def __init__(self, data, source):
self.data = data
self.source = source
if not Event.query.filter(
Event.name == "{} scraping".format(self.source)).first():
# add source
obj = Source()
obj.name = self.source
db.session.add(obj)
db.session.commit()
# create event
self.add_event({
"name":
"{} scraping".format(self.source),
"details":
"Observables mined from {}".format(self.source),
"confidence":
10,
"source":
source,
"tlp":
"Green",
"impact":
"Low",
"likelihood":
"Low"
})
def setUp(self):
# create source
source = Source.find_source_by_id(1)
if source is None:
source = Source(id=1, name="source1", url="url1")
db.session.add(source)
db.session.commit()
else:
source.name = "source1"
source.url = "url1"
db.session.flush()
db.session.commit()
def source_create():
form = SourceCreateForm()
if form.validate_on_submit():
source = Source()
source.sid = form.sid.data
source.name = form.name.data
source.url = form.url.data
db.session.add(source)
db.session.commit()
return redirect(url_for('sources'))
return render_template('source_create.html', form=form)
